import { Entities } from '@fasstech/oip-core';
import { getCompanyId } from '@fasstech/oip-core/entities/Company';
import { getDemographyChildrenCount, getDemographyCode, getDemographyCollege, getDemographyDemographyId, getDemographyEmployeeAverageAge, getDemographyEmployeeCount, getDemographyExternalCode, getDemographyExternalId, getDemographyId, getDemographyLabel, getDemographyMenNumber, getDemographyObjectiveCategory, getDemographyPartnerCount, getDemographyPayroll, getDemographyProfessionalCategoryGroup, getDemographyRiskAndRange, setDemographyChildrenCount, setDemographyCode, setDemographyCollege, setDemographyDemographyId, setDemographyEmployeeAverageAge, setDemographyEmployeeCount, setDemographyExternalCode, setDemographyExternalId, setDemographyLabel, setDemographyMenNumber, setDemographyObjectiveCategory, setDemographyPartnerCount, setDemographyPayroll, setDemographyProfessionalCategoryGroup, setDemographyRiskAndRange } from '@fasstech/oip-core/entities/Demography';
import { DemographyObjectiveCategoryEnum } from '@fasstech/oip-core/entities/Demography/enums';
import { getProjectId } from '@fasstech/oip-core/entities/Project';
import { CollegeEnum, ProfessionalCategoryGroupEnum, RiskAndRangeEnum } from '@fasstech/oip-core/entities/enums';

(async () => {
  let project = await Entities('Project').create();
  let company = await Entities('Company').create();
  const demography = await Entities('Demography').create();
  
  setDemographyChildrenCount(99, demography);
  setDemographyCollege(CollegeEnum.ALL, demography);
  setDemographyEmployeeAverageAge(101, demography);
  setDemographyEmployeeCount(101, demography);
  setDemographyMenNumber(109, demography);
  setDemographyObjectiveCategory(DemographyObjectiveCategoryEnum.EMPLOYEE_NOT_ART21_22_ANI_20171117, demography);
  setDemographyPartnerCount(112, demography);
  setDemographyPayroll([112], demography);
  setDemographyProfessionalCategoryGroup(ProfessionalCategoryGroupEnum.SUPERVISOR, demography);
  setDemographyRiskAndRange(RiskAndRangeEnum.GROUP_PROTECTION, demography);
  setDemographyCode('code', demography);
  setDemographyExternalId('externalId', demography);
  setDemographyExternalCode('externalCode', demography);
  setDemographyDemographyId('demographyId', demography);
  setDemographyLabel('label', demography);
  
  company = await Entities('Demography').save(company, demography);
  project = await Entities('Company').save(project, company);
  await Entities('Project').save(project);
  
  const projectToTest = await Entities('Project').get({ id: getProjectId(project) });
  const companyToTest = await Entities('Company').get(projectToTest, { filter: ({ id }) => id === getCompanyId(company) });
  const demographyToTest = await Entities('Demography').get(companyToTest, { filter: ({ id }) => id === getDemographyId(demography) });
  
  getDemographyChildrenCount(demographyToTest); // 99
  getDemographyCollege(demographyToTest); // CollegeEnum.ALL
  getDemographyEmployeeAverageAge(demographyToTest); // 101
  getDemographyEmployeeCount(demographyToTest); // 101
  getDemographyMenNumber(demographyToTest); // 109
  getDemographyObjectiveCategory(demographyToTest); // DemographyObjectiveCategoryEnum.EMPLOYEE_NOT_ART21_22_ANI_20171117
  getDemographyPartnerCount(demographyToTest); // 112
  getDemographyPayroll(demographyToTest); // [112]
  getDemographyProfessionalCategoryGroup(demographyToTest); // ProfessionalCategoryGroupEnum.SUPERVISOR
  getDemographyRiskAndRange(demographyToTest); // RiskAndRangeEnum.GROUP_PROTECTION
  getDemographyCode(demographyToTest); // 'code'
  getDemographyExternalId(demographyToTest); // 'externalId'
  getDemographyExternalCode(demographyToTest); // 'externalCode'
  getDemographyDemographyId(demographyToTest); // 'demographyId'
  getDemographyLabel(demographyToTest); // 'label'
})();