import { Entities } from '@fasstech/oip-core';
import { getCompanyId } from '@fasstech/oip-core/entities/company';
import { getDemographyChildrenCount, getDemographyCollege, getDemographyEmployeeAverageAge, getDemographyEmployeeCount, getDemographyMenNumber, getDemographyObjectiveCategory, getDemographyPartnerCount, getDemographyPayroll, getDemographyProfessionalCategoryGroup, getDemographyRiskAndRange, getdemographyId, setDemographyChildrenCount, setDemographyCollege, setDemographyEmployeeAverageAge, setDemographyEmployeeCount, setDemographyMenNumber, setDemographyObjectiveCategory, setDemographyPartnerCount, setDemographyPayroll, setDemographyProfessionalCategoryGroup, setDemographyRiskAndRange } from '@fasstech/oip-core/entities/demography';
import { collegeDemographyEnum, objectiveCategoryEnum, demographyRiskAndRangeEnum, demographyProfessionalCategoryGroupEnum } from '@fasstech/oip-core/entities/demography/enums';
import { getProjectId } from '@fasstech/oip-core/entities/project';

(async () => {
  let ProjectEntity = await Entities('Project').get({ id: projectId });
  let CompanyEntity = await Entities('Company').create();
  const DemographyEntity = await Entities('Demography').create();
  
  setDemographyChildrenCount(99, DemographyEntity);
  setDemographyCollege(collegeDemographyEnum.ALL, DemographyEntity);
  setDemographyEmployeeAverageAge(101, DemographyEntity);
  setDemographyEmployeeCount(101, DemographyEntity);
  setDemographyMenNumber(109, DemographyEntity);
  setDemographyObjectiveCategory(objectiveCategoryEnum.EMPLOYEE_NOT_ART21_22_ANI_20171117, DemographyEntity);
  setDemographyPartnerCount(112, DemographyEntity);
  setDemographyPayroll([112], DemographyEntity);
  setDemographyProfessionalCategoryGroup(demographyProfessionalCategoryGroupEnum.SUPERVISOR, DemographyEntity);
  setDemographyRiskAndRange(demographyRiskAndRangeEnum.GROUP_PROTECTION, DemographyEntity);
  
  CompanyEntity = await Entities('Demography').save(CompanyEntity, DemographyEntity);
  ProjectEntity = await Entities('Company').save(ProjectEntity, CompanyEntity);
  await Entities('Project').save(ProjectEntity);
  
  const ProjectEntityToTest = await Entities('Project').get({ id: getProjectId(ProjectEntity) });
  const CompanyEntityToTest = await Entities('Company').get(ProjectEntityToTest, getCompanyId(CompanyEntity));
  const DemographyEntityToTest = await Entities('Demography').get(CompanyEntityToTest, getDemographyId(DemographyEntity));
  
  getDemographyChildrenCount(DemographyEntityToTest); // 99
  getDemographyCollege(DemographyEntityToTest); // collegeDemographyEnum.ALL
  getDemographyEmployeeAverageAge(DemographyEntityToTest); // 101
  getDemographyEmployeeCount(DemographyEntityToTest); // 101
  getDemographyMenNumber(DemographyEntityToTest); // 109
  getDemographyObjectiveCategory(DemographyEntityToTest); // objectiveCategoryEnum.EMPLOYEE_NOT_ART21_22_ANI_20171117
  getDemographyPartnerCount(DemographyEntityToTest); // 112
  getDemographyPayroll(DemographyEntityToTest); // [112]
  getDemographyProfessionalCategoryGroup(DemographyEntityToTest); // demographyProfessionalCategoryGroupEnum.SUPERVISOR
  getDemographyRiskAndRange(DemographyEntityToTest); // demographyRiskAndRangeEnum.GROUP_PROTECTION
})();