import { Entities } from '@fasstech/oip-core';
import { getCollectionNeedId } from '@fasstech/oip-core/entities/CollectionNeed';
import { getProjectId } from '@fasstech/oip-core/entities/Project';
import { getProtectionNeedCategory, getProtectionNeedCode, getProtectionNeedExternalCode, getProtectionNeedExternalId, getProtectionNeedGuarantee, getProtectionNeedId, getProtectionNeedIsSelected, getProtectionNeedLabel, getProtectionNeedLevel, getProtectionNeedProtectionNeedId, setProtectionNeedCategory, setProtectionNeedCode, setProtectionNeedExternalCode, setProtectionNeedExternalId, setProtectionNeedGuarantee, setProtectionNeedIsSelected, setProtectionNeedLabel, setProtectionNeedLevel, setProtectionNeedProtectionNeedId } from '@fasstech/oip-core/entities/ProtectionNeed';
import { ProtectionNeedLevelEnum } from '@fasstech/oip-core/entities/ProtectionNeed/enums';

(async () => {
  let project = await Entities('Project').create();
  let collectionNeed = await Entities('CollectionNeed').create();
  const protectionNeed = await Entities('ProtectionNeed').create();
  
  setProtectionNeedCategory('category', protectionNeed);
  setProtectionNeedGuarantee('guarantee', protectionNeed);
  setProtectionNeedIsSelected(false, protectionNeed);
  setProtectionNeedLevel(ProtectionNeedLevelEnum.FOUR, protectionNeed);
  setProtectionNeedCode('code', protectionNeed);
  setProtectionNeedExternalId('externalId', protectionNeed);
  setProtectionNeedExternalCode('externalCode', protectionNeed);
  setProtectionNeedProtectionNeedId('protectionNeedId', protectionNeed);
  setProtectionNeedLabel('label', protectionNeed);
  
  collectionNeed = await Entities('ProtectionNeed').save(collectionNeed, protectionNeed);
  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 protectionNeedToTest = await Entities('ProtectionNeed').get(collectionNeedToTest, { filter: ({ id }) => id === getProtectionNeedId(protectionNeed) });
  
  getProtectionNeedCategory(protectionNeedToTest); // 'category'
  getProtectionNeedGuarantee(protectionNeedToTest); // 'guarantee'
  getProtectionNeedIsSelected(protectionNeedToTest); // false
  getProtectionNeedLevel(protectionNeedToTest); // ProtectionNeedLevelEnum.FOUR
  getProtectionNeedCode(protectionNeedToTest); // 'code'
  getProtectionNeedExternalId(protectionNeedToTest); // 'externalId'
  getProtectionNeedExternalCode(protectionNeedToTest); // 'externalCode'
  getProtectionNeedProtectionNeedId(protectionNeedToTest); // 'protectionNeedId'
  getProtectionNeedLabel(protectionNeedToTest); // 'label'
})();