try to fix the test failure in travis

intelsdi-x · Feb 18, 2016 · 61984a5 · 61984a5
1 parent cd9252d
commit 61984a5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/control/available_plugin.go b/control/available_plugin.go
@@ -252,7 +252,7 @@ func (a *availablePlugin) CheckHealth() {
 		} else {
 			a.healthCheckFailed()
 		}
-	case <-time.After(time.Second * 1):
+	case <-time.After(DefaultHealthCheckTimeout):
 		a.healthCheckFailed()
 	}
 }

diff --git a/control/control_test.go b/control/control_test.go
@@ -1134,7 +1134,7 @@ func TestFailedPlugin(t *testing.T) {
 				var err []error
 				var cr []core.Metric
 				eventMap := map[string]int{}
-				for i := 0; i <= MaxPluginRestartCount; i++ {
+				for i := 0; i < MaxPluginRestartCount+1; i++ {
 					cr, err = c.CollectMetrics(m, time.Now().Add(time.Second*1), uuid.New())
 					So(err, ShouldNotBeNil)
 					So(cr, ShouldBeNil)
@@ -1145,13 +1145,12 @@ func TestFailedPlugin(t *testing.T) {
 						<-lpe.done
 						eventMap[lpe.plugin.EventNamespace]++
 						So(pool.RestartCount(), ShouldEqual, i+1)
+						So(lpe.plugin.EventNamespace, ShouldEqual, control_event.AvailablePluginRestarted)
 					}
 				}
 				<-lpe.done
-				eventMap[lpe.plugin.EventNamespace]++
-				So(eventMap[control_event.AvailablePluginDead], ShouldEqual, MaxPluginRestartCount+1)
+				So(lpe.plugin.EventNamespace, ShouldEqual, control_event.PluginRestartsExceeded)
 				So(eventMap[control_event.AvailablePluginRestarted], ShouldEqual, MaxPluginRestartCount)
-				So(eventMap[control_event.PluginRestartsExceeded], ShouldEqual, 1)
 				So(len(pool.Plugins()), ShouldEqual, 0)
 				So(pool.RestartCount(), ShouldEqual, MaxPluginRestartCount)
 			})