need to fix unit test

opentdf · Sep 7, 2023 · 1e1beb6 · 1e1beb6
1 parent 5c63e38
commit 1e1beb6
Showing 1 changed file with 62 additions and 62 deletions.
diff --git a/lib/tests/mocha/encrypt-decrypt.spec.ts b/lib/tests/mocha/encrypt-decrypt.spec.ts
@@ -1,9 +1,9 @@
 import getMocks from '../mocks/index.js';
-import {  TDF } from '../../tdf3/src/index.js';
-// import * as cryptoService from '../../tdf3/src/crypto/index.js';
-// import { createSandbox } from 'sinon';
-// import { assert } from 'chai';
-// import { Binary } from '../../tdf3/src/binary.js';
+import { Client, TDF } from '../../tdf3/src/index.js';
+import * as cryptoService from '../../tdf3/src/crypto/index.js';
+import { createSandbox } from 'sinon';
+import { assert } from 'chai';
+import { Binary } from '../../tdf3/src/binary.js';
 import { HttpRequest } from '../../src/auth/auth.js';
 const Mocks = getMocks();
 
@@ -18,61 +18,61 @@ describe('encrypt decrypt test', async function () {
   const kasUrl = `http://localhost:4000/`;
   const kasPublicKey = await TDF.extractPemFromKeyString(Mocks.kasPublicKey);
 
-  // it('encrypt-decrypt stream source happy path', async function () {
-  //   const sandbox = createSandbox();
-  //   try {
-  //     const tdf1 = TDF.create({ cryptoService });
-  //     sandbox.replace(
-  //       TDF,
-  //       'create',
-  //       sandbox.fake(() => tdf1)
-  //     );
-  //     sandbox.spy(tdf1, '_generateManifest');
-  //     sandbox.stub(tdf1, 'unwrapKey').callsFake(async () => {
-  //       // @ts-ignore
-  //       const keyInfo = tdf1._generateManifest.lastCall.args[0];
-  //       return {
-  //         reconstructedKeyBinary: keyInfo.unwrappedKeyBinary as Binary,
-  //         metadata: undefined,
-  //       };
-  //     });
-  //     const client = new Client.Client({
-  //       kasEndpoint: kasUrl,
-  //       kasPublicKey,
-  //       keypair: { publicKey: Mocks.entityPublicKey, privateKey: Mocks.entityPrivateKey },
-  //       clientId: 'id',
-  //       authProvider,
-  //     });
-  //
-  //     const eo = await Mocks.getEntityObject();
-  //     const scope = Mocks.getScope();
-  //
-  //     const encryptedStream = await client.encrypt({
-  //       eo,
-  //       metadata: Mocks.getMetadataObject(),
-  //       offline: true,
-  //       scope,
-  //       source: new ReadableStream({
-  //         start(controller) {
-  //           controller.enqueue(new TextEncoder().encode(expectedVal));
-  //           controller.close();
-  //         },
-  //       }),
-  //     });
-  //
-  //     const decryptStream = await client.decrypt({
-  //       eo,
-  //       source: {
-  //         type: 'stream',
-  //         location: encryptedStream.stream,
-  //       },
-  //     });
-  //
-  //     const { value: decryptedText } = await decryptStream.stream.getReader().read();
-  //
-  //     assert.equal(new TextDecoder().decode(decryptedText), expectedVal);
-  //   } finally {
-  //     sandbox.restore();
-  //   }
-  // });
+  it('encrypt-decrypt stream source happy path', async function () {
+    const sandbox = createSandbox();
+    try {
+      const tdf1 = TDF.create({ cryptoService });
+      sandbox.replace(
+        TDF,
+        'create',
+        sandbox.fake(() => tdf1)
+      );
+      sandbox.spy(tdf1, '_generateManifest');
+      sandbox.stub(tdf1, 'unwrapKey').callsFake(async () => {
+        // @ts-ignore
+        const keyInfo = tdf1._generateManifest.lastCall.args[0];
+        return {
+          reconstructedKeyBinary: keyInfo.unwrappedKeyBinary as Binary,
+          metadata: undefined,
+        };
+      });
+      const client = new Client.Client({
+        kasEndpoint: kasUrl,
+        kasPublicKey,
+        keypair: { publicKey: Mocks.entityPublicKey, privateKey: Mocks.entityPrivateKey },
+        clientId: 'id',
+        authProvider,
+      });
+
+      const eo = await Mocks.getEntityObject();
+      const scope = Mocks.getScope();
+
+      const encryptedStream = await client.encrypt({
+        eo,
+        metadata: Mocks.getMetadataObject(),
+        offline: true,
+        scope,
+        source: new ReadableStream({
+          start(controller) {
+            controller.enqueue(new TextEncoder().encode(expectedVal));
+            controller.close();
+          },
+        }),
+      });
+
+      const decryptStream = await client.decrypt({
+        eo,
+        source: {
+          type: 'stream',
+          location: encryptedStream.stream,
+        },
+      });
+
+      const { value: decryptedText } = await decryptStream.stream.getReader().read();
+
+      assert.equal(new TextDecoder().decode(decryptedText), expectedVal);
+    } finally {
+      sandbox.restore();
+    }
+  });
 });