Fix data race

* dont stub after calling method under test [#85545422]
ihocho · Apr 2, 2015 · 8ddfe9b · 8ddfe9b
1 parent 25a6b14
commit 8ddfe9b
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/route_fetcher/route_fetcher_test.go b/route_fetcher/route_fetcher_test.go
@@ -211,9 +211,6 @@ var _ = Describe("RouteFetcher", func() {
 				eventSource := fake_routing_api.FakeEventSource{}
 				client.SubscribeToEventsReturns(&eventSource, nil)
 
-				tokenFetcher.FetchTokenReturns(token, nil)
-				fetcher.StartEventCycle()
-
 				received := make(chan struct{})
 
 				eventSource.NextStub = func() (routing_api.Event, error) {
@@ -230,6 +227,9 @@ var _ = Describe("RouteFetcher", func() {
 					return event, nil
 				}
 
+				tokenFetcher.FetchTokenReturns(token, nil)
+				fetcher.StartEventCycle()
+
 				<-received
 
 				Expect(registry.UnregisterCallCount()).To(Equal(1))
@@ -240,16 +240,16 @@ var _ = Describe("RouteFetcher", func() {
 				eventSource := fake_routing_api.FakeEventSource{}
 				client.SubscribeToEventsReturns(&eventSource, nil)
 
-				tokenFetcher.FetchTokenReturns(token, nil)
-				fetcher.StartEventCycle()
-
 				received := make(chan struct{})
 
 				eventSource.NextStub = func() (routing_api.Event, error) {
 					received <- struct{}{}
 					return routing_api.Event{}, errors.New("beep boop im a robot")
 				}
 
+				tokenFetcher.FetchTokenReturns(token, nil)
+				fetcher.StartEventCycle()
+
 				<-received
 
 				Expect(sink.Records()).ToNot(BeNil())