import { Entities } from '@fasstech/oip-core';
import { getCollectionNeedId } from '@fasstech/oip-core/entities/CollectionNeed';
import { getPensionNeedCode, getPensionNeedExternalCode, getPensionNeedExternalId, getPensionNeedId, getPensionNeedKnowledgeLevel, getPensionNeedLabel, getPensionNeedMainExpectation, getPensionNeedPensionNeedId, getPensionNeedPlanAvailabilityOption, getPensionNeedPlanExpectations, setPensionNeedCode, setPensionNeedExternalCode, setPensionNeedExternalId, setPensionNeedKnowledgeLevel, setPensionNeedLabel, setPensionNeedMainExpectation, setPensionNeedPensionNeedId, setPensionNeedPlanAvailabilityOption, setPensionNeedPlanExpectations } from '@fasstech/oip-core/entities/PensionNeed';
import { PensionNeedKnowledgeLevelEnum, PensionNeedMainExpectationEnum, PensionNeedPlanAvailabilityOptionEnum, PensionNeedPlanExpectationsEnum } from '@fasstech/oip-core/entities/PensionNeed/enums';
import { getProjectId } from '@fasstech/oip-core/entities/Project';

(async () => {
  let project = await Entities('Project').create();
  let collectionNeed = await Entities('CollectionNeed').create();
  const pensionNeed = await Entities('PensionNeed').create();
  
  setPensionNeedKnowledgeLevel(PensionNeedKnowledgeLevelEnum.MEDIUM, pensionNeed);
  setPensionNeedMainExpectation([PensionNeedMainExpectationEnum.RETIRED_CAPITAL_PROFIT], pensionNeed);
  setPensionNeedPlanAvailabilityOption(PensionNeedPlanAvailabilityOptionEnum.INVESTMENT_AVAILABLE, pensionNeed);
  setPensionNeedPlanExpectations([PensionNeedPlanExpectationsEnum.PARTNER_DEATH_PROTECTION], pensionNeed);
  setPensionNeedCode('code', pensionNeed);
  setPensionNeedExternalId('externalId', pensionNeed);
  setPensionNeedExternalCode('externalCode', pensionNeed);
  setPensionNeedPensionNeedId('pensionNeedId', pensionNeed);
  setPensionNeedLabel('label', pensionNeed);
  
  collectionNeed = await Entities('PensionNeed').save(collectionNeed, pensionNeed);
  project = await Entities('CollectionNeed').save(project, collectionNeed);
  await Entities('Project').save(project);
  
  const projectToTest = await Entities('Project').get({ id: getProjectId(project) });
  const collectionNeedToTest = await Entities('CollectionNeed').get(projectToTest, { filter: ({ id }) => id === getCollectionNeedId(collectionNeed) });
  const pensionNeedToTest = await Entities('PensionNeed').get(collectionNeedToTest, { filter: ({ id }) => id === getPensionNeedId(pensionNeed) });
  
  getPensionNeedKnowledgeLevel(pensionNeedToTest); // PensionNeedKnowledgeLevelEnum.MEDIUM
  getPensionNeedMainExpectation(pensionNeedToTest); // [PensionNeedMainExpectationEnum.RETIRED_CAPITAL_PROFIT]
  getPensionNeedPlanAvailabilityOption(pensionNeedToTest); // PensionNeedPlanAvailabilityOptionEnum.INVESTMENT_AVAILABLE
  getPensionNeedPlanExpectations(pensionNeedToTest); // [PensionNeedPlanExpectationsEnum.PARTNER_DEATH_PROTECTION]
  getPensionNeedCode(pensionNeedToTest); // 'code'
  getPensionNeedExternalId(pensionNeedToTest); // 'externalId'
  getPensionNeedExternalCode(pensionNeedToTest); // 'externalCode'
  getPensionNeedPensionNeedId(pensionNeedToTest); // 'pensionNeedId'
  getPensionNeedLabel(pensionNeedToTest); // 'label'
})();