import { Entities } from '@fasstech/oip-core';
import { getCollectionNeedCode, getCollectionNeedCollectionNeedId, getCollectionNeedCollege, getCollectionNeedExternalCode, getCollectionNeedExternalId, getCollectionNeedId, getCollectionNeedIsDutyOfAdviceSkipped, getCollectionNeedIsEnforcedGuarantee, getCollectionNeedLabel, getCollectionNeedRiskAndRange, setCollectionNeedCode, setCollectionNeedCollectionNeedId, setCollectionNeedCollege, setCollectionNeedExternalCode, setCollectionNeedExternalId, setCollectionNeedIsDutyOfAdviceSkipped, setCollectionNeedIsEnforcedGuarantee, setCollectionNeedLabel, setCollectionNeedRiskAndRange } from '@fasstech/oip-core/entities/CollectionNeed';
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 collectionNeed = await Entities('CollectionNeed').create();
  
  setCollectionNeedCollege(CollegeEnum.ALL, collectionNeed);
  setCollectionNeedIsDutyOfAdviceSkipped(false, collectionNeed);
  setCollectionNeedIsEnforcedGuarantee(false, collectionNeed);
  setCollectionNeedRiskAndRange(RiskAndRangeEnum.GROUP_PROTECTION, collectionNeed);
  setCollectionNeedCode('code', collectionNeed);
  setCollectionNeedExternalId('externalId', collectionNeed);
  setCollectionNeedExternalCode('externalCode', collectionNeed);
  setCollectionNeedCollectionNeedId('collectionNeedId', collectionNeed);
  setCollectionNeedLabel('label', collectionNeed);
  
  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) });
  
  getCollectionNeedCollege(collectionNeedToTest); // CollegeEnum.ALL
  getCollectionNeedIsDutyOfAdviceSkipped(collectionNeedToTest); // false
  getCollectionNeedIsEnforcedGuarantee(collectionNeedToTest); // false
  getCollectionNeedRiskAndRange(collectionNeedToTest); // RiskAndRangeEnum.GROUP_PROTECTION
  getCollectionNeedCode(collectionNeedToTest); // 'code'
  getCollectionNeedExternalId(collectionNeedToTest); // 'externalId'
  getCollectionNeedExternalCode(collectionNeedToTest); // 'externalCode'
  getCollectionNeedCollectionNeedId(collectionNeedToTest); // 'collectionNeedId'
  getCollectionNeedLabel(collectionNeedToTest); // 'label'
})();