Add cleanup step to all tests

Author: alexleventer

__tests__/astra.test.ts

@@ -3,31 +3,30 @@ import { describe, test } from "mocha";
 import { expect } from "chai";
 
 const { ASTRA_DB_TOKEN, ASTRA_DB_ID } = process.env;
+const epoch = new Date().getTime();
 
 describe("Astra", () => {
+  let astra;
+  before(async () => {
+    astra = new Astra({
+      token: ASTRA_DB_TOKEN,
+      databaseId: ASTRA_DB_ID,
+      databaseRegion: "us-east1",
+      namespace: "test",
+    });
+  })
   describe("Collections", () => {
     test("should create collection", async () => {
-      const astra = new Astra({
-        token: ASTRA_DB_TOKEN,
-        databaseId: ASTRA_DB_ID,
-        databaseRegion: "us-east1",
-        namespace: "test",
-      });
-
-      const results = await astra.createCollection({ name: "blah" });
+      const collectionName = `test${epoch}`;
+      const results = await astra.createCollection({ name: collectionName });
       expect(results.status.ok).to.equal(1);
+      await astra.deleteCollection({ name: collectionName });
     });
 
     test("should create vector collection", async () => {
-      const astra = new Astra({
-        token: ASTRA_DB_TOKEN,
-        databaseId: ASTRA_DB_ID,
-        databaseRegion: "us-east1",
-        namespace: "test",
-      });
-
+      const collectionName = `test${epoch}`;
       const results = await astra.createCollection({
-        name: "blahs",
+        name: collectionName,
         options: {
           vector: {
             size: 2,
@@ -36,28 +35,26 @@ describe("Astra", () => {
         },
       });
       expect(results.status.ok).to.equal(1);
+      await astra.deleteCollection({ name: collectionName });
     });
 
     test("should find collections", async () => {
-      const astra = new Astra({
-        token: ASTRA_DB_TOKEN,
-        databaseId: ASTRA_DB_ID,
-        databaseRegion: "us-east1",
-        namespace: "test",
+      const collectionName = `test${epoch}`;
+      await astra.createCollection({
+        name: collectionName,
+        options: {
+          vector: {
+            size: 2,
+            function: "cosine",
+          },
+        },
       });
-
       const results = await astra.findCollections();
-      expect(results.status.collections[0]).to.equal("blah");
+      expect(results.status.collections[0]).to.equal(collectionName);
+      await astra.deleteCollection({ name: collectionName });
     });
 
     test("should delete collection that doesn't exist", async () => {
-      const astra = new Astra({
-        token: ASTRA_DB_TOKEN,
-        databaseId: ASTRA_DB_ID,
-        databaseRegion: "us-east1",
-        namespace: "test",
-      });
-
       const collection = await astra.deleteCollection({ name: "bah" });
       expect(collection.status.ok).to.equal(1);
     });

__tests__/collection.test.ts

@@ -79,7 +79,6 @@ describe("Collections", () => {
           },
         },
       });
-      console.log(updateResults);
       await astra.deleteCollection({ name: collectionName });
     });
     test("should delete one", async () => {