add tests with sam

narval-xyz · Aug 30, 2024 · 641028b · 641028b
1 parent ae4350f
commit 641028b
Show file tree

Hide file tree

Showing 2 changed files with 94 additions and 1 deletion.
diff --git a/apps/policy-engine/src/resource/open-policy-agent/rego/__test__/policies/spendings.rego b/apps/policy-engine/src/resource/open-policy-agent/rego/__test__/policies/spendings.rego
@@ -170,7 +170,7 @@ permit[{"policyId": "spendingLimitWithFixedPeriod"}] = reason {
 		"operator": "lte",
 		"timeWindow": {
 			"type": "fixed",
-			"value": "1d",
+			"period": "1d",
 		},
 		"filters": {
 			"tokens": {"eip155:137/erc20:0x2791bca1f2de4661ed88a30c99a7a9449aa84174"},
@@ -185,3 +185,38 @@ permit[{"policyId": "spendingLimitWithFixedPeriod"}] = reason {
 		"approvalsMissing": [],
 	}
 }
+
+permit[{"policyId": "spendingLimitWithRange"}] = reason {
+	checkIntentToken({"eip155:1/slip44:60"})
+	checkSpendingLimit({
+		"limit": "1000000000000000000",
+		"operator": "gt",
+		"timeWindow": {
+			"type": "rolling",
+			"value": 86400,
+		},
+		"filters": {
+			"perPrincipal": true,
+			"tokens": {"eip155:1/slip44:60"},
+		},
+	})
+	checkSpendingLimit({
+		"limit": "10000000000000000000",
+		"operator": "lte",
+		"timeWindow": {
+			"type": "rolling",
+			"value": 86400,
+		},
+		"filters": {
+			"perPrincipal": true,
+			"tokens": {"eip155:1/slip44:60"},
+		},
+	})
+
+	reason = {
+		"type": "permit",
+		"policyId": "spendingLimitWithRange",
+		"approvalsSatisfied": [],
+		"approvalsMissing": [],
+	}
+}
diff --git a/apps/policy-engine/src/resource/open-policy-agent/rego/__test__/policies/spendings_test.rego b/apps/policy-engine/src/resource/open-policy-agent/rego/__test__/policies/spendings_test.rego
@@ -315,3 +315,61 @@ test_spendingLimitWithoutHistoricalData {
 
 	print("res", res)
 }
+
+test_spendingLimitWithRange {
+	spendingLimitWithRangeReq = {
+		"action": "signTransaction",
+		"transactionRequest": {
+			"chainId": 1,
+			"from": "0x0301e2724a40e934cce3345928b88956901aa127",
+			"to": "0x76d1b7f9b3f69c435eef76a98a415332084a856f",
+			"value": "0x4563918244F40000"
+		},
+		"principal": {"userId": "test-alice-uid"},
+		"resource": {"uid": "eip155:eoa:0xddcf208f219a6e6af072f2cfdc615b2c1805f98e"},
+		"intent": {
+			"to": "eip155:1:0x76d1b7f9b3f69c435eef76a98a415332084a856f",
+			"from": "eip155:1:0xddcf208f219a6e6af072f2cfdc615b2c1805f98e",
+			"type": "transferNative",
+			"amount": "5000000000000000000",
+			"token": "eip155:1/slip44:60",
+			"chainId": 1,
+		},
+		"feeds": [
+			{
+				"source": "armory/price-feed",
+				"sig": "",
+				"data": {},
+			},
+			{
+				"source": "armory/historical-transfer-feed",
+				"sig": "",
+				"data": [
+				    {
+						"id": "7a2c9055-0b1f-41df-91c8-e76f06b12a4f",
+						"resourceId": "eip155:eoa:0xddcf208f219a6e6af072f2cfdc615b2c1805f98e",
+						"requestId": "67c09036-650a-4287-84d9-edc349d66748",
+						"chainId": 1,
+						"from": "eip155:1:0xddcf208f219a6e6af072f2cfdc615b2c1805f98e",
+						"to": "eip155:1:0x76d1b7f9b3f69c435eef76a98a415332084a856f",
+						"token": "eip155:1/slip44:60",
+						"amount": "1000000000000000000",
+						"rates": {},
+						"initiatedBy": "test-alice-uid",
+						"createdAt": "2024-08-30T12:24:22.081Z",
+						"timestamp": 1725023081 * 1000,
+					}
+				],
+			},
+		],
+	}
+
+	res = permit[{"policyId": "spendingLimitWithRange"}] with input as spendingLimitWithRangeReq with data.entities as entities
+
+	res == {
+		"approvalsMissing": [],
+		"approvalsSatisfied": [],
+		"policyId": "spendingLimitWithRange",
+		"type": "permit",
+	}
+}