Fix race condition on the shadow tests

GrowwOld · Jan 6, 2019 · cc20a10 · cc20a10
1 parent 6b4e477
commit cc20a10
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/proxy/shadow_test.go b/proxy/shadow_test.go
@@ -52,7 +52,7 @@ func TestShadowMiddleware(t *testing.T) {
 	request := &Request{}
 	p(context.Background(), request)
 	time.Sleep(100 * time.Millisecond)
-	if counter != 2 {
+	if atomic.LoadUint64(&counter) != 2 {
 		t.Errorf("The shadow proxy should have been called 2 times, not %d", counter)
 	}
 }
@@ -107,7 +107,7 @@ func TestNewShadowFactory(t *testing.T) {
 		t.Error(err)
 	}
 	time.Sleep(100 * time.Millisecond)
-	if counter != 2 {
+	if atomic.LoadUint64(&counter) != 2 {
 		t.Errorf("The shadow proxy should have been called 2 times, not %d", counter)
 	}
 }