From 1c58c132a337b4d10eac82525ec84790b1fde309 Mon Sep 17 00:00:00 2001
From: jason plumb <75337021+breedx-splk@users.noreply.github.com>
Date: Wed, 4 Aug 2021 02:06:56 -0700
Subject: [PATCH] test script cleans up the data it inserts. (#3765)

---
 testing-overhead/k6/basic.js | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/testing-overhead/k6/basic.js b/testing-overhead/k6/basic.js
index e6c4f2de93..407cfcc23f 100644
--- a/testing-overhead/k6/basic.js
+++ b/testing-overhead/k6/basic.js
@@ -1,5 +1,5 @@
 import http from "k6/http";
-import { check } from "k6";
+import {check} from "k6";
 import names from "./names.js";
 
 const baseUri = `http://petclinic:9966/petclinic/api`;
@@ -62,6 +62,25 @@ export default function() {
     check(responses[1], { "pet exists 200": r => r.status === 200});
     check(responses[2], { "pet exists 200": r => r.status === 200});
 
+    // Clean up after ourselves.
+    // Delete pets
+    const petDeletes = http.batch([
+      ["DELETE", `${baseUri}/pets/${JSON.parse(newPetResponses[0].body).id}`],
+      ["DELETE", `${baseUri}/pets/${JSON.parse(newPetResponses[1].body).id}`],
+      ["DELETE", `${baseUri}/pets/${JSON.parse(newPetResponses[2].body).id}`]
+    ]);
+
+    check(petDeletes[0], { "pet deleted 204": r => r.status === 204});
+    check(petDeletes[1], { "pet deleted 204": r => r.status === 204});
+    check(petDeletes[2], { "pet deleted 204": r => r.status === 204});
+
+    // Delete owner
+    const delOwner = http.del(`${baseUri}/owners/${ownerId}`);
+    check(delOwner, { "owner deleted 204": r => r.status === 204});
+
+    // Delete vet
+    const delVet = http.del(`${baseUri}/vets/${vetId}`);
+    check(delVet, { "owner deleted 204": r => r.status === 204});
 
     //TODO: Set up a visit or two
     //TODO: Fetch out the owner again because their model has been updated.