rewrite tests

moira-alert · Sep 27, 2018 · bd194c3 · bd194c3
1 parent 253004c
commit bd194c3
Showing 1 changed file with 5 additions and 12 deletions.
diff --git a/protectors/matched/matched_test.go b/protectors/matched/matched_test.go
@@ -97,19 +97,12 @@ func TestMatchedProtector(t *testing.T) {
 		for _, val := range databaseResponse {
 			database.EXPECT().GetMatchedMetricsUpdatesCount().Return(val, nil)
 		}
-		//database.EXPECT().GetMatchedMetricsUpdatesCount().Return(int64(0), nil).AnyTimes()
 		stream := protector.GetStream()
-		numChecks := 0
-		for {
-			select {
-			case protectorData := <- stream:
-				fmt.Println(protectorData)
-				protector.Protect(protectorData)
-				if numChecks == 10 {
-					return
-				}
-				numChecks ++
-			}
+		numChecks := 10
+		for i := 0; i < numChecks; i++ {
+			protectorData := <- stream
+			fmt.Println(protectorData)
+			protector.Protect(protectorData)
 		}
 	})
 }