(chore) Wire up saves for endpoint tests

backend/src/api/tests/index.ts

@@ -1,6 +1,8 @@
 import { Request, Response } from "express";
 import ApiResponseHandler from "api-response-handler";
 import { runTest } from "~/services/testing/runTests";
+import { AppDataSource } from "~/data-source";
+import { ApiEndpointTest } from "~/models";
 
 export const runTestHandler = async (
   req: Request,
@@ -10,3 +12,27 @@ export const runTestHandler = async (
   runTest(req.body.test);
   await ApiResponseHandler.success(res, null);
 };
+
+export const saveTest = async (req: Request, res: Response): Promise<void> => {
+  const {
+    test: { uuid, name, tags, requests },
+    endpointUuid,
+  } = req.body;
+  let resp = await AppDataSource.getRepository(ApiEndpointTest)
+    .createQueryBuilder()
+    .insert()
+    .into(ApiEndpointTest)
+    .values({
+      uuid: uuid,
+      name,
+      tags,
+      requests,
+      apiEndpoint: {
+        uuid: endpointUuid,
+      },
+    })
+    .orUpdate(["name", "tags", "requests"], ["uuid"])
+    .execute();
+
+  await ApiResponseHandler.success(res, resp);
+};

backend/src/data-source.ts

@@ -8,6 +8,7 @@ import {
   Alert,
   Session,
   Connections,
+  ApiEndpointTest,
 } from "models";
 
 export const AppDataSource: DataSource = new DataSource({
@@ -22,6 +23,7 @@ export const AppDataSource: DataSource = new DataSource({
     Alert,
     Session,
     Connections,
+    ApiEndpointTest,
   ],
   migrations: [],
   logging: false,

backend/src/index.ts

@@ -35,7 +35,7 @@ import {
   aws_os_choices,
   setup_connection,
 } from "./api/setup";
-import { runTestHandler } from "./api/tests";
+import { runTestHandler, saveTest } from "./api/tests";
 import {
   get_connection_for_uuid,
   get_ssh_key_for_connection_uuid,
@@ -102,9 +102,10 @@ app.get(
   get_ssh_key_for_connection_uuid
 );
 app.post("/api/v1/update_connection", update_connection);
-
 app.post("/api/v1/test/run", runTestHandler);
 
+app.post("/api/v1/tests/save", saveTest);
+
 const main = async () => {
   try {
     const datasource = await AppDataSource.initialize();

backend/src/models/index.ts

@@ -5,6 +5,7 @@ import { OpenApiSpec } from "models/openapi-spec";
 import { Alert } from "models/alert";
 import { Session } from "models/sessions";
 import { Connections } from "./connections";
+import { ApiEndpointTest } from "./api-endpoint-test";
 
 export {
   ApiEndpoint,
@@ -14,4 +15,5 @@ export {
   Alert,
   Session,
   Connections,
+  ApiEndpointTest,
 };

common/src/testing/types.ts

@@ -61,6 +61,7 @@ export interface Request {
 }
 
 export interface Test {
+  uuid: string;
   name: string;
   tags: string[];
   requests: Request[];

frontend/src/components/TestEditor/index.tsx

@@ -15,6 +15,8 @@ import RequestList from "./requestsList";
 import RequestEditor from "./requestEditor";
 import { makeNewEmptyRequest, sendRequest } from "./requestUtils";
 import { runTest } from "~/api/tests";
+import axios from "axios";
+import { getAPIURL } from "~/constants";
 
 interface TestEditorProps {
   endpoint: ApiEndpointDetailed;
@@ -141,6 +143,17 @@ const TestEditor: React.FC<TestEditorProps> = React.memo(
         });
     };
 
+    const onSaveRequest = () => {
+      axios
+        .post(`${getAPIURL()}/tests/save`, {
+          test: state.test,
+          endpointUuid: endpoint.uuid,
+        })
+        .then((v) => console.log(v))
+        .catch((e) => console.log(e))
+        .finally(() => console.log("Request done"));
+    };
+
     const onRunClick = () => {
       runTest(test);
     };
@@ -170,7 +183,9 @@ const TestEditor: React.FC<TestEditorProps> = React.memo(
               <Button colorScheme="blue" onClick={onRunClick}>
                 Run
               </Button>
-              <Button colorScheme="blue">Save</Button>
+              <Button colorScheme="blue" onClick={onSaveRequest}>
+                Save
+              </Button>
             </HStack>
           </HStack>
         </VStack>

frontend/src/pages/endpoint/[endpointUUID]/test/new.tsx

@@ -7,6 +7,7 @@ import TestEditor from "components/TestEditor";
 import { getEndpoint } from "api/endpoints";
 import { ApiEndpointDetailed } from "@common/types";
 import { makeNewRequest } from "~/components/TestEditor/requestUtils";
+import { v4 as uuidv4 } from "uuid";
 
 const NewTest = ({ endpoint }) => {
   const parsedEndpoint = superjson.parse(
@@ -23,6 +24,7 @@ const NewTest = ({ endpoint }) => {
       <TestEditor
         endpoint={parsedEndpoint}
         initTest={{
+          uuid: uuidv4(),
           name: "Untitled Test",
           tags: [],
           requests: [makeNewRequest(parsedEndpoint)],