From 6df0768803089eda2dad47a402fd08a348dc9944 Mon Sep 17 00:00:00 2001 From: Juan Fernandez Date: Fri, 7 Jul 2023 09:59:16 -0400 Subject: [PATCH] Fixed broken tests --- server/service/async/async_scheduled_query_stats_test.go | 8 ++++---- server/service/integration_core_test.go | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/server/service/async/async_scheduled_query_stats_test.go b/server/service/async/async_scheduled_query_stats_test.go index f03f6a0675..784919c8ec 100644 --- a/server/service/async/async_scheduled_query_stats_test.go +++ b/server/service/async/async_scheduled_query_stats_test.go @@ -28,10 +28,10 @@ func testCollectScheduledQueryStats(t *testing.T, ds *mysql.Datastore, pool flee p2 := test.NewPack(t, ds, "p2") p3 := test.NewPack(t, ds, "p3") - q1 := test.NewQuery(t, ds, "q1", "select 1", user.ID, true) - q2 := test.NewQuery(t, ds, "q2", "select 2", user.ID, true) - q3 := test.NewQuery(t, ds, "q3", "select 3", user.ID, true) - q4 := test.NewQuery(t, ds, "q4", "select 4", user.ID, true) + q1 := test.NewQuery(t, ds, nil, "q1", "select 1", user.ID, true) + q2 := test.NewQuery(t, ds, nil, "q2", "select 2", user.ID, true) + q3 := test.NewQuery(t, ds, nil, "q3", "select 3", user.ID, true) + q4 := test.NewQuery(t, ds, nil, "q4", "select 4", user.ID, true) sq1 := test.NewScheduledQuery(t, ds, p1.ID, q1.ID, 60, false, false, "sq1") sq2 := test.NewScheduledQuery(t, ds, p2.ID, q2.ID, 60, false, false, "sq2") diff --git a/server/service/integration_core_test.go b/server/service/integration_core_test.go index 9924fc62b7..b456efd107 100644 --- a/server/service/integration_core_test.go +++ b/server/service/integration_core_test.go @@ -1196,7 +1196,7 @@ func (s *integrationTestSuite) TestListHosts() { assert.Greater(t, resp.Hosts[0].SoftwareUpdatedAt, resp.Hosts[0].CreatedAt) user1 := test.NewUser(t, s.ds, "Alice", "alice@example.com", true) - q := test.NewQuery(t, s.ds, "query1", "select 1", 0, true) + q := test.NewQuery(t, s.ds, nil, "query1", "select 1", 0, true) defer cleanupQuery(s, q.ID) p, err := s.ds.NewGlobalPolicy(context.Background(), &user1.ID, fleet.PolicyPayload{ QueryID: &q.ID,