From 068c2142b92b4e5f896035cc0db68b967fbb1ac3 Mon Sep 17 00:00:00 2001 From: Liran Cohen Date: Sat, 10 Aug 2024 17:07:28 -0400 Subject: [PATCH] permissions-api tests --- packages/agent/src/permissions-api.ts | 41 +++ packages/agent/tests/permissions-api.spec.ts | 254 +++++++++++++++++++ 2 files changed, 295 insertions(+) create mode 100644 packages/agent/tests/permissions-api.spec.ts diff --git a/packages/agent/src/permissions-api.ts b/packages/agent/src/permissions-api.ts index 06e0eb2e1..990165f17 100644 --- a/packages/agent/src/permissions-api.ts +++ b/packages/agent/src/permissions-api.ts @@ -12,6 +12,12 @@ export type FetchPermissionsParams = { protocol?: string; } +export type FetchPermissionRequestParams = { + author: string; + target: string; + protocol?: string; +} + export type PermissionGrantEntry = { grant: DwnPermissionGrant; message: DwnDataEncodedRecordsWriteMessage; @@ -58,6 +64,8 @@ export interface PermissionsApi { */ fetchGrants: (params: FetchPermissionsParams) => Promise; + fetchRequests: (params: FetchPermissionRequestParams) => Promise; + /** * Check whether a grant is revoked by reading the revocation record for a given grant recordId. */ @@ -137,6 +145,39 @@ export class AgentPermissionsApi implements PermissionsApi { return grants; } + async fetchRequests({ + author, + target, + protocol, + }:FetchPermissionRequestParams):Promise { + // filter by a protocol using tags if provided + const tags = protocol ? { protocol } : undefined; + + const { reply: requestsReply } = await this.agent.processDwnRequest({ + author : author, + target : target, + messageType : DwnInterface.RecordsQuery, + messageParams : { + filter: { + ...DwnPermissionsUtil.permissionsProtocolParams('request'), + tags + } + } + }); + + if (requestsReply.status.code !== 200) { + throw new Error(`AgentDwnApi: Failed to fetch requests: ${requestsReply.status.detail}`); + } + + const requests: PermissionRequestEntry[] = []; + for (const entry of requestsReply.entries! as DwnDataEncodedRecordsWriteMessage[]) { + const request = await DwnPermissionRequest.parse(entry); + requests.push({ request, message: entry }); + } + + return requests; + } + async isGrantRevoked(author:string, target: string, grantRecordId: string): Promise { const { reply: revocationReply } = await this.agent.processDwnRequest({ author, diff --git a/packages/agent/tests/permissions-api.spec.ts b/packages/agent/tests/permissions-api.spec.ts new file mode 100644 index 000000000..07180c806 --- /dev/null +++ b/packages/agent/tests/permissions-api.spec.ts @@ -0,0 +1,254 @@ +import { expect } from 'chai'; +import { AgentPermissionsApi } from '../src/permissions-api.js'; +import { PlatformAgentTestHarness } from '../src/test-harness.js'; +import { TestAgent } from './utils/test-agent.js'; +import { BearerDid } from '@web5/dids'; + +import { testDwnUrl } from './utils/test-config.js'; +import { DwnInterfaceName, DwnMethodName, Time } from '@tbd54566975/dwn-sdk-js'; +import { DwnPermissionGrant } from '../src/index.js'; + +let testDwnUrls: string[] = [testDwnUrl]; + +describe('AgentPermissionsApi', () => { + let testHarness: PlatformAgentTestHarness; + let aliceDid: BearerDid; + + before(async () => { + testHarness = await PlatformAgentTestHarness.setup({ + agentClass : TestAgent, + agentStores : 'dwn' + }); + }); + + after(async () => { + await testHarness.clearStorage(); + await testHarness.closeStorage(); + }); + + beforeEach(async () => { + await testHarness.clearStorage(); + await testHarness.createAgentDid(); + + // Create an "alice" Identity to author the DWN messages. + const alice = await testHarness.createIdentity({ name: 'Alice', testDwnUrls }); + await testHarness.agent.identity.manage({ portableIdentity: await alice.export() }); + aliceDid = alice.did; + }); + + describe('get agent', () => { + it(`returns the 'agent' instance property`, async () => { + // we are only mocking + const permissionsApi = new AgentPermissionsApi({ agent: testHarness.agent }); + const agent = permissionsApi.agent; + expect(agent).to.exist; + expect(agent.agentDid).to.equal(testHarness.agent.agentDid); + }); + + it(`throws an error if the 'agent' instance property is undefined`, () => { + const permissionsApi = new AgentPermissionsApi(); + expect(() => + permissionsApi.agent + ).to.throw(Error, 'AgentPermissionsApi: Agent is not set'); + }); + }); + + describe('createGrant', () => { + it('creates and stores a grant', async () => { + // scenario: create a grant for deviceX, confirm the grant exists + + // create an identity for deviceX + const aliceDeviceX = await testHarness.agent.identity.create({ + store : false, + metadata : { name: 'Alice Device X' }, + didMethod : 'jwk' + }); + + + // create a grant for deviceX + const deviceXGrant = await testHarness.agent.permissions.createGrant({ + store : true, + author : aliceDid.uri, + grantedTo : aliceDeviceX.did.uri, + dateExpires : Time.createOffsetTimestamp({ seconds: 60 }), + scope : { + interface : DwnInterfaceName.Records, + method : DwnMethodName.Write, + protocol : 'http://example.com/protocol' + } + }); + + const grants = await testHarness.agent.permissions.fetchGrants({ + author : aliceDid.uri, + target : aliceDid.uri, + }); + + // expect to have the 1 grant created for deviceX + expect(grants.length).to.equal(1); + expect(grants[0].message.recordId).to.equal(deviceXGrant.message.recordId); + }); + + it('creates a grant without storing it', async () => { + // scenario: create a grant for deviceX, confirm the grant does not exist + + // create an identity for deviceX + const aliceDeviceX = await testHarness.agent.identity.create({ + store : false, + metadata : { name: 'Alice Device X' }, + didMethod : 'jwk' + }); + + // create a grant for deviceX store is set to false by default + const deviceXGrant = await testHarness.agent.permissions.createGrant({ + author : aliceDid.uri, + grantedTo : aliceDeviceX.did.uri, + dateExpires : Time.createOffsetTimestamp({ seconds: 60 }), + scope : { + interface : DwnInterfaceName.Records, + method : DwnMethodName.Write, + protocol : 'http://example.com/protocol' + } + }); + + const grantDataObject = { ...deviceXGrant.grant }; + const parsedGrant = await DwnPermissionGrant.parse(deviceXGrant.message); + + expect(grantDataObject).to.deep.equal(parsedGrant); + }); + }); + + describe('createRevocation', () => { + it('creates and stores a grant revocation', async () => { + // scenario: create a grant for deviceX, revoke the grant, confirm the grant is revoked + + // create an identity for deviceX + const aliceDeviceX = await testHarness.agent.identity.create({ + store : true, + metadata : { name: 'Alice Device X' }, + didMethod : 'jwk' + }); + + // create a grant for deviceX + const deviceXGrant = await testHarness.agent.permissions.createGrant({ + store : true, + author : aliceDid.uri, + grantedTo : aliceDeviceX.did.uri, + dateExpires : Time.createOffsetTimestamp({ seconds: 60 }), + scope : { + interface : DwnInterfaceName.Records, + method : DwnMethodName.Write, + protocol : 'http://example.com/protocol' + } + }); + + // parse the grant + const writeGrant = await DwnPermissionGrant.parse(deviceXGrant.message); + + // check if the grant is revoked + let isRevoked = await testHarness.agent.permissions.isGrantRevoked(aliceDid.uri, aliceDid.uri, deviceXGrant.grant.id); + expect(isRevoked).to.equal(false); + + // create a revocation for the grant + await testHarness.agent.permissions.createRevocation({ + author : aliceDid.uri, + store : true, + grant : writeGrant, + }); + + // check if the grant is revoked again, should be true + isRevoked = await testHarness.agent.permissions.isGrantRevoked(aliceDid.uri, aliceDid.uri, deviceXGrant.grant.id); + expect(isRevoked).to.equal(true); + }); + + it('creates a grant revocation without storing it', async () => { + // scenario: create a grant for deviceX, revoke the grant, confirm the grant is revoked + + // create an identity for deviceX + const aliceDeviceX = await testHarness.agent.identity.create({ + store : true, + metadata : { name: 'Alice Device X' }, + didMethod : 'jwk' + }); + + // create a grant for deviceX + const deviceXGrant = await testHarness.agent.permissions.createGrant({ + store : true, + author : aliceDid.uri, + grantedTo : aliceDeviceX.did.uri, + dateExpires : Time.createOffsetTimestamp({ seconds: 60 }), + scope : { + interface : DwnInterfaceName.Records, + method : DwnMethodName.Write, + protocol : 'http://example.com/protocol' + } + }); + + // parse the grant + const writeGrant = await DwnPermissionGrant.parse(deviceXGrant.message); + + // check if the grant is revoked + let isRevoked = await testHarness.agent.permissions.isGrantRevoked(aliceDid.uri, aliceDid.uri, deviceXGrant.grant.id); + expect(isRevoked).to.equal(false); + + // create a revocation for the grant without storing it + await testHarness.agent.permissions.createRevocation({ + author : aliceDid.uri, + grant : writeGrant, + }); + + // check if the grant is revoked again, should be true + isRevoked = await testHarness.agent.permissions.isGrantRevoked(aliceDid.uri, aliceDid.uri, deviceXGrant.grant.id); + expect(isRevoked).to.equal(false); + }); + }); + + describe('createRequest', () => { + it('creates a permission request and stores it', async () => { + // scenario: create a permission request confirm the request exists + + // create a permission request + const deviceXRequest = await testHarness.agent.permissions.createRequest({ + author : aliceDid.uri, + store : true, + scope : { + interface : DwnInterfaceName.Records, + method : DwnMethodName.Write, + protocol : 'http://example.com/protocol' + } + }); + + // query for the request + const fetchedRequests = await testHarness.agent.permissions.fetchRequests({ + author : aliceDid.uri, + target : aliceDid.uri, + }); + + // expect to have the 1 request created + expect(fetchedRequests.length).to.equal(1); + expect(fetchedRequests[0].request.id).to.equal(deviceXRequest.message.recordId); + }); + + it('creates a permission request without storing it', async () => { + // scenario: create a permission request confirm the request does not exist + + // create a permission request store is set to false by default + await testHarness.agent.permissions.createRequest({ + author : aliceDid.uri, + scope : { + interface : DwnInterfaceName.Records, + method : DwnMethodName.Write, + protocol : 'http://example.com/protocol' + } + }); + + // query for the request + const fetchedRequests = await testHarness.agent.permissions.fetchRequests({ + author : aliceDid.uri, + target : aliceDid.uri, + }); + + // expect to have no requests + expect(fetchedRequests.length).to.equal(0); + }); + }); +}); \ No newline at end of file