feat(compliance): add support for IsIdentity condition

PolymeshAssociation · Nov 10, 2020 · 8201d5c · 8201d5c
1 parent 6786fd9
commit 8201d5c
Show file tree

Hide file tree

Showing 7 changed files with 239 additions and 30 deletions.
diff --git a/src/api/entities/SecurityToken/Compliance/Requirements.ts b/src/api/entities/SecurityToken/Compliance/Requirements.ts
@@ -73,7 +73,7 @@ export class Requirements extends Namespace<SecurityToken> {
       const defaultTrustedClaimIssuers = claimIssuers.map(identityIdToString);
 
       return assetCompliance.requirements.map(complianceRequirement => {
-        const requirement = complianceRequirementToRequirement(complianceRequirement);
+        const requirement = complianceRequirementToRequirement(complianceRequirement, context);
 
         requirement.conditions.forEach(condition => {
           if (!condition.trustedClaimIssuers || !condition.trustedClaimIssuers.length) {
@@ -174,7 +174,7 @@ export class Requirements extends Namespace<SecurityToken> {
       primaryIssuanceAgent ? stringToIdentityId(primaryIssuanceAgent.did, context) : null
     );
 
-    return assetComplianceResultToRequirementCompliance(res);
+    return assetComplianceResultToRequirementCompliance(res, context);
   }
 
   /**

diff --git a/src/api/procedures/__tests__/setAssetRequirements.ts b/src/api/procedures/__tests__/setAssetRequirements.ts
@@ -22,7 +22,10 @@ describe('setAssetRequirements procedure', () => {
     [Requirement, Context],
     ComplianceRequirement
   >;
-  let complianceRequirementToRequirementStub: sinon.SinonStub<[ComplianceRequirement], Requirement>;
+  let complianceRequirementToRequirementStub: sinon.SinonStub<
+    [ComplianceRequirement, Context],
+    Requirement
+  >;
   let assetCompliancesStub: sinon.SinonStub;
   let ticker: string;
   let requirements: Condition[][];
@@ -124,7 +127,8 @@ describe('setAssetRequirements procedure', () => {
             sender_conditions: senderConditions[index],
             receiver_conditions: receiverConditions[index],
             /* eslint-enable @typescript-eslint/camelcase */
-          })
+          }),
+          mockContext
         )
         .returns({ conditions: condition, id: 1 });
     });

diff --git a/src/api/procedures/setAssetRequirements.ts b/src/api/procedures/setAssetRequirements.ts
@@ -40,7 +40,7 @@ export async function prepareSetAssetRequirements(
   const rawCurrentAssetCompliance = await query.complianceManager.assetCompliances(rawTicker);
 
   const currentRequirements = rawCurrentAssetCompliance.requirements.map(
-    requirement => complianceRequirementToRequirement(requirement).conditions
+    requirement => complianceRequirementToRequirement(requirement, context).conditions
   );
 
   const comparator = (a: Condition[], b: Condition[]): boolean => {

diff --git a/src/testUtils/mocks/dataSources.ts b/src/testUtils/mocks/dataSources.ts
@@ -74,6 +74,7 @@ import {
   SecurityToken,
   SettlementType,
   Signatory,
+  TargetIdentity,
   Ticker,
   TickerRegistration,
   TickerRegistrationConfig,
@@ -1639,6 +1640,14 @@ export const createMockClaim = (
     | 'NoData'
 ): Claim => createMockEnum(claim) as Claim;
 
+/**
+ * @hidden
+ * NOTE: `isEmpty` will be set to true if no value is passed
+ */
+export const createMockTargetIdentity = (
+  targetIdentity?: { Specific: IdentityId } | 'PrimaryIssuanceAgent'
+): TargetIdentity => createMockEnum(targetIdentity) as TargetIdentity;
+
 /**
  * @hidden
  * NOTE: `isEmpty` will be set to true if no value is passed
@@ -1649,6 +1658,7 @@ export const createMockConditionType = (
     | { IsAbsent: Claim }
     | { IsAnyOf: Claim[] }
     | { IsNoneOf: Claim[] }
+    | { IsIdentity: TargetIdentity }
 ): ConditionType => createMockEnum(conditionType) as ConditionType;
 
 /**

diff --git a/src/types/index.ts b/src/types/index.ts
@@ -290,6 +290,8 @@ export enum ConditionType {
   IsAbsent = 'IsAbsent',
   IsAnyOf = 'IsAnyOf',
   IsNoneOf = 'IsNoneOf',
+  IsPrimaryIssuanceAgent = 'IsPrimaryIssuanceAgent',
+  IsIdentity = 'IsIdentity',
 }
 
 export type ConditionBase = { target: ConditionTarget; trustedClaimIssuers?: string[] };
@@ -304,7 +306,20 @@ export type MultiClaimCondition = ConditionBase & {
   claims: Claim[];
 };
 
-export type Condition = SingleClaimCondition | MultiClaimCondition;
+export type IdentityCondition = ConditionBase & {
+  type: ConditionType.IsIdentity;
+  identity: Identity;
+};
+
+export type PrimaryIssuanceAgentCondition = ConditionBase & {
+  type: ConditionType.IsPrimaryIssuanceAgent;
+};
+
+export type Condition =
+  | SingleClaimCondition
+  | MultiClaimCondition
+  | IdentityCondition
+  | PrimaryIssuanceAgentCondition;
 
 /**
  * @hidden