Fixed racy tests

gaia-pipeline · Jun 25, 2019 · 78c33fa · 78c33fa
1 parent 376c8ca
commit 78c33fa
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/handlers/workers_test.go b/handlers/workers_test.go
@@ -50,6 +50,7 @@ func TestRegisterWorker(t *testing.T) {
 	m := &mockStorageService{}
 	services.MockStorageService(m)
 	dataStore, _ := services.StorageService()
+	defer func() { services.MockStorageService(nil) }()
 
 	// Initialize certificate store
 	_, err = services.CertificateService()
@@ -189,6 +190,7 @@ func TestDeregisterWorker(t *testing.T) {
 	m := &mockStorageService{}
 	services.MockStorageService(m)
 	dataStore, _ := services.StorageService()
+	defer func() { services.MockStorageService(nil) }()
 
 	// Initialize certificate store
 	_, err = services.CertificateService()
@@ -380,6 +382,7 @@ func TestGetWorkerStatusOverview(t *testing.T) {
 	m := &mockStorageService{}
 	services.MockStorageService(m)
 	dataStore, _ := services.StorageService()
+	defer func() { services.MockStorageService(nil) }()
 
 	// Initialize certificate store
 	_, err = services.CertificateService()
@@ -531,6 +534,7 @@ func TestGetWorker(t *testing.T) {
 	m := &mockStorageService{}
 	services.MockStorageService(m)
 	dataStore, _ := services.StorageService()
+	defer func() { services.MockStorageService(nil) }()
 
 	// Initialize certificate store
 	_, err = services.CertificateService()