feat: added tests

[ci skip]
MatrixAI · Sep 10, 2024 · e324e4e · e324e4e
1 parent 150040e
commit e324e4e
Showing 1 changed file with 89 additions and 2 deletions.
diff --git a/tests/secrets/remove.test.ts b/tests/secrets/remove.test.ts
@@ -41,7 +41,7 @@ describe('commandDeleteSecret', () => {
     });
   });
 
-  test('should remove secrets', async () => {
+  test('should remove secret', async () => {
     const vaultName = 'Vault2' as VaultName;
     const vaultId = await polykeyAgent.vaultManager.createVault(vaultName);
 
@@ -53,7 +53,7 @@ describe('commandDeleteSecret', () => {
 
     command = ['secrets', 'rm', '-np', dataDir, `${vaultName}:MySecret`];
 
-    const result = await testUtils.pkStdio([...command], {
+    const result = await testUtils.pkStdio(command, {
       env: { PK_PASSWORD: password },
       cwd: dataDir,
     });
@@ -64,4 +64,91 @@ describe('commandDeleteSecret', () => {
       expect(list.sort()).toStrictEqual([]);
     });
   });
+  test('should remove multiple secrets', async () => {
+    const vaultName = 'Vault2' as VaultName;
+    const vaultId = await polykeyAgent.vaultManager.createVault(vaultName);
+    const secretNames = ['secret1', 'secret2', 'secret3'];
+
+    await polykeyAgent.vaultManager.withVaults([vaultId], async (vault) => {
+      for (const secretName of secretNames) {
+        await vaultOps.addSecret(vault, secretName, secretName);
+      }
+      const list = await vaultOps.listSecrets(vault);
+      expect(list.sort()).toStrictEqual(secretNames);
+    });
+
+    const secretPaths = secretNames.map((v) => `${vaultName}:${v}`);
+    command = ['secrets', 'rm', '-np', dataDir, ...secretPaths];
+
+    const result = await testUtils.pkStdio(command, {
+      env: { PK_PASSWORD: password },
+      cwd: dataDir,
+    });
+    expect(result.exitCode).toBe(0);
+
+    await polykeyAgent.vaultManager.withVaults([vaultId], async (vault) => {
+      const list = await vaultOps.listSecrets(vault);
+      expect(list.sort()).toStrictEqual([]);
+    });
+  });
+  test('should remove secrets recursively', async () => {
+    const vaultName = 'Vault2' as VaultName;
+    const vaultId = await polykeyAgent.vaultManager.createVault(vaultName);
+    const secretDir = 'secretDir';
+    const secretNames = ['secret1', 'secret2', 'secret3'].map((v) =>
+      path.join(secretDir, v),
+    );
+
+    await polykeyAgent.vaultManager.withVaults([vaultId], async (vault) => {
+      vaultOps.mkdir(vault, secretDir);
+      for (const secretName of secretNames) {
+        await vaultOps.addSecret(vault, secretName, secretName);
+      }
+      const list = await vaultOps.listSecrets(vault);
+      expect(list.sort()).toStrictEqual(secretNames);
+    });
+
+    command = ['secrets', 'rm', '-np', dataDir, `${vaultName}:secretDir`, '-r'];
+
+    const result = await testUtils.pkStdio(command, {
+      env: { PK_PASSWORD: password },
+      cwd: dataDir,
+    });
+    expect(result.exitCode).toBe(0);
+
+    await polykeyAgent.vaultManager.withVaults([vaultId], async (vault) => {
+      const list = await vaultOps.listSecrets(vault);
+      expect(list.sort()).toStrictEqual([]);
+    });
+  });
+  test('should fail to remove directory without recursive flag', async () => {
+    const vaultName = 'Vault2' as VaultName;
+    const vaultId = await polykeyAgent.vaultManager.createVault(vaultName);
+    const secretDir = 'secretDir';
+    const secretNames = ['secret1', 'secret2', 'secret3'].map((v) =>
+      path.join(secretDir, v),
+    );
+
+    await polykeyAgent.vaultManager.withVaults([vaultId], async (vault) => {
+      vaultOps.mkdir(vault, secretDir);
+      for (const secretName of secretNames) {
+        await vaultOps.addSecret(vault, secretName, secretName);
+      }
+      const list = await vaultOps.listSecrets(vault);
+      expect(list.sort()).toStrictEqual(secretNames);
+    });
+
+    command = ['secrets', 'rm', '-np', dataDir, `${vaultName}:secretDir`];
+
+    const result = await testUtils.pkStdio(command, {
+      env: { PK_PASSWORD: password },
+      cwd: dataDir,
+    });
+    expect(result.exitCode).not.toBe(0);
+
+    await polykeyAgent.vaultManager.withVaults([vaultId], async (vault) => {
+      const list = await vaultOps.listSecrets(vault);
+      expect(list.sort()).toStrictEqual(secretNames);
+    });
+  });
 });