@aamp/protocol 1.1.2 → 1.1.4

package/test/handshake.spec.ts

@@ -1,67 +1,63 @@
-/**
- * Integration Test: Protocol Handshake
- * Run using: npm test
- * Location: sdk/typescript/test/handshake.spec.ts
- */
-import { AAMPAgent } from '../src/agent';
-import { AAMPPublisher } from '../src/publisher';
-import { AccessPurpose } from '../src/types';
-
-function base64ToUint8Array(base64: string): Uint8Array {
-  const binaryString = atob(base64);
-  const len = binaryString.length;
-  const bytes = new Uint8Array(len);
-  for (let i = 0; i < len; i++) {
-    bytes[i] = binaryString.charCodeAt(i);
-  }
-  return bytes;
-}
-
-async function runTest() {
-  console.log("--- STARTING AAMP HANDSHAKE TEST ---");
-
-  // 1. Setup Publisher with Economic Policy
-  const publisher = new AAMPPublisher({
-    version: '1.1',
-    allowTraining: false,
-    allowRAG: true,
-    attributionRequired: true,
-    requiresPayment: true,        // Publisher wants money...
-    allowAdSupportedAccess: true, // ...OR ads
-    paymentPointer: '$wallet.example.com/publisher'
-  });
-
-  // 2. Initialize Agent
-  const agent = new AAMPAgent();
-  await agent.initialize();
-
-  // TEST CASE A: Requesting RAG without Ads (Should FAIL due to Payment Requirement)
-  console.log("\n[TEST A] Requesting RAG (No Ads)...");
-  const reqA = await agent.createAccessRequest('/doc/1', AccessPurpose.RAG_RETRIEVAL, { adsDisplayed: false });
-  
-  // Fix: Cast to any to avoid TS2769 error (Uint8Array vs BufferSource mismatch in ts-node)
-  const keyA = await crypto.subtle.importKey(
-    "spki", 
-    base64ToUint8Array(reqA.publicKey!) as any, 
-    { name: "ECDSA", namedCurve: "P-256" }, 
-    true, 
-    ["verify"]
-  );
-
-  const resA = await publisher.verifyRequest(reqA, keyA);
-  console.log("Result A (Expect Deny):", resA);
-  if (resA.allowed) throw new Error("Test A Failed (Should require payment)");
-
-
-  // TEST CASE B: Requesting RAG WITH Ads (Should SUCCEED via Exemption)
-  console.log("\n[TEST B] Requesting RAG (With Ads)...");
-  const reqB = await agent.createAccessRequest('/doc/1', AccessPurpose.RAG_RETRIEVAL, { adsDisplayed: true });
-  
-  const resB = await publisher.verifyRequest(reqB, keyA);
-  console.log("Result B (Expect Allow):", resB);
-  if (!resB.allowed) throw new Error("Test B Failed (Should allow via Ad exemption)");
-
-  console.log("\n--- ALL TESTS PASSED ---");
-}
-
+/**
+ * Integration Test: Protocol Handshake
+ * Run using: npm test
+ * Location: sdk/typescript/test/handshake.spec.ts
+ */
+import { AAMPAgent } from '../src/agent';
+import { AAMPPublisher } from '../src/publisher';
+import { AccessPurpose } from '../src/types';
+import { HEADERS } from '../src/constants';
+
+async function runTest() {
+  console.log("--- STARTING AAMP HANDSHAKE TEST ---");
+
+  // 1. Setup Publisher with Economic Policy
+  const publisher = new AAMPPublisher({
+    version: '1.1',
+    allowTraining: false,
+    allowRAG: true,
+    attributionRequired: true,
+    requiresPayment: true,        // Publisher wants money...
+    allowAdSupportedAccess: true, // ...OR ads
+    paymentPointer: '$wallet.example.com/publisher'
+  });
+
+  // 2. Initialize Agent
+  const agent = new AAMPAgent();
+  await agent.initialize();
+
+  // TEST CASE A: Requesting RAG without Ads (Should FAIL due to Payment Requirement)
+  console.log("\n[TEST A] Requesting RAG (No Ads)...");
+  const reqA = await agent.createAccessRequest('/doc/1', AccessPurpose.RAG_RETRIEVAL, { adsDisplayed: false });
+  
+  const payloadA = JSON.stringify(reqA.header);
+  const headersA = {
+    [HEADERS.PAYLOAD]: btoa(payloadA),
+    [HEADERS.SIGNATURE]: reqA.signature,
+    [HEADERS.PUBLIC_KEY]: reqA.publicKey!
+  };
+
+  const resA = await publisher.evaluateVisitor(headersA, payloadA);
+  console.log("Result A (Expect Deny):", resA);
+  if (resA.allowed) throw new Error("Test A Failed (Should require payment)");
+
+
+  // TEST CASE B: Requesting RAG WITH Ads (Should SUCCEED via Exemption)
+  console.log("\n[TEST B] Requesting RAG (With Ads)...");
+  const reqB = await agent.createAccessRequest('/doc/1', AccessPurpose.RAG_RETRIEVAL, { adsDisplayed: true });
+  
+  const payloadB = JSON.stringify(reqB.header);
+  const headersB = {
+    [HEADERS.PAYLOAD]: btoa(payloadB),
+    [HEADERS.SIGNATURE]: reqB.signature,
+    [HEADERS.PUBLIC_KEY]: reqB.publicKey!
+  };
+
+  const resB = await publisher.evaluateVisitor(headersB, payloadB);
+  console.log("Result B (Expect Allow):", resB);
+  if (!resB.allowed) throw new Error("Test B Failed (Should allow via Ad exemption)");
+
+  console.log("\n--- ALL TESTS PASSED ---");
+}
+
 runTest().catch(console.error);

package/tsconfig.json

@@ -1,15 +1,15 @@
-{
-  "compilerOptions": {
-    "target": "ES2020",
-    "module": "CommonJS",
-    "lib": ["ES2020", "DOM"],
-    "declaration": true,
-    "outDir": "./dist",
-    "rootDir": "./src",
-    "strict": true,
-    "esModuleInterop": true,
-    "skipLibCheck": true,
-    "forceConsistentCasingInFileNames": true
-  },
-  "include": ["src/**/*"]
+{
+  "compilerOptions": {
+    "target": "ES2020",
+    "module": "CommonJS",
+    "lib": ["ES2020", "DOM"],
+    "declaration": true,
+    "outDir": "./dist",
+    "rootDir": "./src",
+    "strict": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true,
+    "forceConsistentCasingInFileNames": true
+  },
+  "include": ["src/**/*"]
 }