import { Entities } from '@fasstech/oip-core';
import { getChosenRiskAndRangeChosenRiskAndRangeId, getChosenRiskAndRangeCode, getChosenRiskAndRangeColleges, getChosenRiskAndRangeEffectDate, getChosenRiskAndRangeExternalCode, getChosenRiskAndRangeExternalId, getChosenRiskAndRangeId, getChosenRiskAndRangeLabel, getChosenRiskAndRangeRiskAndRange, setChosenRiskAndRangeChosenRiskAndRangeId, setChosenRiskAndRangeCode, setChosenRiskAndRangeColleges, setChosenRiskAndRangeEffectDate, setChosenRiskAndRangeExternalCode, setChosenRiskAndRangeExternalId, setChosenRiskAndRangeLabel, setChosenRiskAndRangeRiskAndRange } from '@fasstech/oip-core/entities/ChosenRiskAndRange';
import { getProjectId } from '@fasstech/oip-core/entities/Project';
import { CollegeEnum, RiskAndRangeEnum } from '@fasstech/oip-core/entities/enums';

(async () => {
  let project = await Entities('Project').create();
  const chosenRiskAndRange = await Entities('ChosenRiskAndRange').create();
  
  setChosenRiskAndRangeColleges([CollegeEnum.ALL], chosenRiskAndRange);
  setChosenRiskAndRangeEffectDate(new Date('2017-01-01'), chosenRiskAndRange);
  setChosenRiskAndRangeRiskAndRange(RiskAndRangeEnum.GROUP_PROTECTION, chosenRiskAndRange);
  setChosenRiskAndRangeCode('code', chosenRiskAndRange);
  setChosenRiskAndRangeExternalId('externalId', chosenRiskAndRange);
  setChosenRiskAndRangeExternalCode('externalCode', chosenRiskAndRange);
  setChosenRiskAndRangeChosenRiskAndRangeId('chosenRiskAndRangeId', chosenRiskAndRange);
  setChosenRiskAndRangeLabel('label', chosenRiskAndRange);
  
  project = await Entities('ChosenRiskAndRange').save(project, chosenRiskAndRange);
  await Entities('Project').save(project);
  
  const projectToTest = await Entities('Project').get({ id: getProjectId(project) });
  const chosenRiskAndRangeToTest = await Entities('ChosenRiskAndRange').get(projectToTest, { filter: ({ id }) => id === getChosenRiskAndRangeId(chosenRiskAndRange) });
  
  getChosenRiskAndRangeColleges(chosenRiskAndRangeToTest); // [CollegeEnum.ALL]
  getChosenRiskAndRangeEffectDate(chosenRiskAndRangeToTest); // new Date('2017-01-01')
  getChosenRiskAndRangeRiskAndRange(chosenRiskAndRangeToTest); // RiskAndRangeEnum.GROUP_PROTECTION
  getChosenRiskAndRangeCode(chosenRiskAndRangeToTest); // 'code'
  getChosenRiskAndRangeExternalId(chosenRiskAndRangeToTest); // 'externalId'
  getChosenRiskAndRangeExternalCode(chosenRiskAndRangeToTest); // 'externalCode'
  getChosenRiskAndRangeChosenRiskAndRangeId(chosenRiskAndRangeToTest); // 'chosenRiskAndRangeId'
  getChosenRiskAndRangeLabel(chosenRiskAndRangeToTest); // 'label'
})();