Fix tests

distribworks · Apr 25, 2019 · 007d6b0 · 007d6b0
1 parent 2cfcce1
commit 007d6b0
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/dkron/api_test.go b/dkron/api_test.go
@@ -36,7 +36,7 @@ func setupAPITest(t *testing.T) (a *Agent) {
 	time.Sleep(1 * time.Second)
 
 	// clean up the keyspace to ensure clean runs
-	a.Store.Client.DeleteTree("dkron-test")
+	a.Store.Client().DeleteTree("dkron-test")
 
 	return
 }

diff --git a/dkron/grpc_test.go b/dkron/grpc_test.go
@@ -15,7 +15,7 @@ func TestGRPCExecutionDone(t *testing.T) {
 	viper.Reset()
 
 	// Cleanup everything
-	err := store.Client.DeleteTree("dkron")
+	err := store.Client().DeleteTree("dkron")
 	if err != nil {
 		t.Logf("error cleaning up: %s", err)
 	}

diff --git a/dkron/job_test.go b/dkron/job_test.go
@@ -16,7 +16,7 @@ func TestJobGetParent(t *testing.T) {
 	store.agent = a
 
 	// Cleanup everything
-	err := store.Client.DeleteTree("dkron-test")
+	err := store.Client().DeleteTree("dkron-test")
 	if err != nil && err != s.ErrKeyNotFound {
 		t.Logf("error cleaning up: %s", err)
 	}

diff --git a/dkron/store_test.go b/dkron/store_test.go
@@ -215,7 +215,7 @@ func createTestStore() *Store {
 }
 
 func cleanTestKVSpace(s *Store) error {
-	err := s.Client.DeleteTree("dkron-test")
+	err := s.Client().DeleteTree("dkron-test")
 	if err != nil && err != store.ErrKeyNotFound {
 		return err
 	}