From dc32ccfac8f8ddfcc33b84df0b1d68cafdd25547 Mon Sep 17 00:00:00 2001
From: Juan Fernandez <juan@fleetdm.com>
Date: Fri, 7 Jul 2023 08:54:09 -0400
Subject: [PATCH] Refactored Delete test

---
 server/datastore/mysql/queries_test.go | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/server/datastore/mysql/queries_test.go b/server/datastore/mysql/queries_test.go
index d49767dcc4..961a078225 100644
--- a/server/datastore/mysql/queries_test.go
+++ b/server/datastore/mysql/queries_test.go
@@ -21,7 +21,7 @@ func TestQueries(t *testing.T) {
 		fn   func(t *testing.T, ds *Datastore)
 	}{
 		{"Apply", testQueriesApply},
-		// {"Delete", testQueriesDelete},
+		{"Delete", testQueriesDelete},
 		// {"GetByName", testQueriesGetByName},
 		// {"DeleteMany", testQueriesDeleteMany},
 		// {"Save", testQueriesSave},
@@ -140,16 +140,16 @@ func testQueriesDelete(t *testing.T, ds *Datastore) {
 		AuthorID: &user.ID,
 	}
 	query, err := ds.NewQuery(context.Background(), query)
-	require.Nil(t, err)
+	require.NoError(t, err)
 	require.NotNil(t, query)
 	assert.NotEqual(t, query.ID, 0)
 
-	err = ds.DeleteQuery(context.Background(), nil, query.Name)
-	require.Nil(t, err)
+	err = ds.DeleteQuery(context.Background(), query.TeamID, query.Name)
+	require.NoError(t, err)
 
-	assert.NotEqual(t, query.ID, 0)
+	require.NotEqual(t, query.ID, 0)
 	_, err = ds.Query(context.Background(), query.ID)
-	assert.NotNil(t, err)
+	require.Error(t, err)
 }
 
 func testQueriesGetByName(t *testing.T, ds *Datastore) {