test: add tests for the api routes (#3666)

twilio-labs · Dec 11, 2023 · 6a4f6e1 · 6a4f6e1 · vercel · Dec 11, 2023
1 parent 71c481a
commit 6a4f6e1
Show file tree

Hide file tree

Showing 10 changed files with 100 additions and 0 deletions.
diff --git a/.github/workflows/on_deployment_status_cypress.yml b/.github/workflows/on_deployment_status_cypress.yml
@@ -31,6 +31,7 @@ jobs:
       CYPRESS_RECORD_KEY: ${{ secrets.CYPRESS_RECORD_KEY }}
       GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
       PERCY_TOKEN: ${{ secrets.PERCY_TOKEN }}
+      OPENAI_API_SECRET: ${{ secrets.OPENAI_API_SECRET }}
       USE_CYPRESS_VRT: false
 
     steps:

diff --git a/.github/workflows/on_pull_request_cypress.yml b/.github/workflows/on_pull_request_cypress.yml
@@ -35,6 +35,7 @@ jobs:
       AIRTABLE_APIKEY: ${{ secrets.AIRTABLE_APIKEY }}
       AIRTABLE_BASEID: ${{ secrets.AIRTABLE_BASEID }}
       OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
+      OPENAI_API_SECRET: ${{ secrets.OPENAI_API_SECRET }}
       CYPRESS_BASE_URL: http://localhost:3000
       SUPABASE_KEY: ${{ secrets.SUPABASE_STAGING_KEY }}
       SUPABASE_URL: ${{ secrets.SUPABASE_STAGING_URL }}

diff --git a/cypress.config.e2e.ts b/cypress.config.e2e.ts
@@ -8,6 +8,7 @@ export default defineConfig({
     env: {
       USE_CYPRESS_VRT: process.env.USE_CYPRESS_VRT,
       CYPRESS_BASE_URL: process.env.CYPRESS_BASE_URL,
+      OPENAI_API_SECRET: process.env.OPENAI_API_SECRET,
     },
     viewportWidth: 1440,
     viewportHeight: 1440,

diff --git a/cypress.config.ts b/cypress.config.ts
@@ -8,6 +8,7 @@ export default defineConfig({
     env: {
       USE_CYPRESS_VRT: process.env.USE_CYPRESS_VRT,
       CYPRESS_BASE_URL: process.env.CYPRESS_BASE_URL,
+      OPENAI_API_SECRET: process.env.OPENAI_API_SECRET,
     },
     viewportWidth: 1440,
     viewportHeight: 1440,

diff --git a/cypress/integration/api/ai.spec.ts b/cypress/integration/api/ai.spec.ts
@@ -0,0 +1,11 @@
+context("POST /api/ai", () => {
+  it("gets an ai response for a given prompty", () => {
+    cy.request("POST", "/api/ai", {
+      prompt: "How do I create a primary button",
+      secret: Cypress.env("OPENAI_API_SECRET"),
+    }).then((response) => {
+      expect(response.status).to.eq(200);
+      expect(response.body).length.to.be.greaterThan(1);
+    });
+  });
+});
diff --git a/cypress/integration/api/discussions-search.spec.ts b/cypress/integration/api/discussions-search.spec.ts
@@ -0,0 +1,8 @@
+context("GET /api/discussions-search", () => {
+  it("gets a list of discussions", () => {
+    cy.request("POST", "/api/discussions-search", { prompt: "creating a button" }).then((response) => {
+      expect(response.status).to.eq(200);
+      expect(response.body.data).length.to.be.greaterThan(1);
+    });
+  });
+});
diff --git a/cypress/integration/api/docs-search.spec.ts b/cypress/integration/api/docs-search.spec.ts
@@ -0,0 +1,8 @@
+context("GET /api/docs-search", () => {
+  it("gets a list of docs", () => {
+    cy.request("POST", "/api/docs-search", { prompt: "creating a button" }).then((response) => {
+      expect(response.status).to.eq(200);
+      expect(response.body.data).length.to.be.greaterThan(1);
+    });
+  });
+});
diff --git a/cypress/integration/api/og-image.spec.ts b/cypress/integration/api/og-image.spec.ts
@@ -0,0 +1,9 @@
+context("GET /api/og-image", () => {
+  it("gets an image", () => {
+    cy.request("GET", "/api/og-image?componentRequested=primitives/box").then((response) => {
+      expect(response.status).to.eq(200);
+      expect(response.headers["content-type"]).to.eq("image/png");
+      console.log(response);
+    });
+  });
+});
diff --git a/cypress/integration/api/paste-assistant-message.spec.ts b/cypress/integration/api/paste-assistant-message.spec.ts
@@ -0,0 +1,31 @@
+context("POST /api/paste-assistant-message", () => {
+  let threadId: string;
+
+  before(() => {
+    // create a thread for the message
+    cy.request("POST", "/api/paste-assistant-thread", {}).then((response) => {
+      threadId = response.body.id;
+    });
+  });
+
+  after(() => {
+    // delete the thread
+    cy.request("DELETE", "/api/paste-assistant-thread", { id: threadId });
+  });
+
+  it("creates an message on an ai thread", () => {
+    // create a message on the thread
+    cy.request("POST", "/api/paste-assistant-message", { threadId, message: "create a button" }).then((response) => {
+      expect(response.status).to.eq(200);
+    });
+  });
+
+  it("gets messages on an ai thread", () => {
+    // get messages on the thread
+    cy.request("GET", `/api/paste-assistant-messages/${threadId}`).then((response) => {
+      expect(response.status).to.eq(200);
+      expect(response.body.data).length.to.be.greaterThan(0);
+      expect(response.body.data[0].content[0].text.value).to.eq("create a button");
+    });
+  });
+});
diff --git a/cypress/integration/api/paste-assistant-thread.spec.ts b/cypress/integration/api/paste-assistant-thread.spec.ts
@@ -0,0 +1,29 @@
+context("POST /api/paste-assistant-thread", () => {
+  let threadId: string;
+  it("creates an ai thread", () => {
+    cy.request("POST", "/api/paste-assistant-thread", {}).then((response) => {
+      expect(response.status).to.eq(200);
+      threadId = response.body.id;
+    });
+  });
+  it("updates an ai thread", () => {
+    cy.request("PUT", "/api/paste-assistant-thread", { id: threadId, metadata: { testKey: "testData" } }).then(
+      (response) => {
+        expect(response.status).to.eq(200);
+        expect(response.body.metadata.testKey).to.eq("testData");
+      },
+    );
+  });
+  it("gets an ai thread", () => {
+    cy.request("GET", `/api/paste-assistant-thread/${threadId}`).then((response) => {
+      expect(response.status).to.eq(200);
+      expect(response.body.metadata.testKey).to.eq("testData");
+    });
+  });
+  it("deletes an ai thread", () => {
+    cy.request("DELETE", "/api/paste-assistant-thread", { id: threadId }).then((response) => {
+      expect(response.status).to.eq(200);
+      expect(response.body.deleted).to.eq(true);
+    });
+  });
+});