refactor(test): Adds wait() step to tests

gpestana · Nov 11, 2017 · 6a72525 · 6a72525
1 parent c10119c
commit 6a72525
Show file tree

Hide file tree

Showing 5 changed files with 12 additions and 5 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,2 +1,4 @@
 kapacitor-unit
 notes.md
+*.swp
+*.swo
diff --git a/README.md b/README.md
@@ -6,7 +6,7 @@
 
 
 Kapacitor-unit is a testing framework to make TICK scripts testing easy and
-automated. Testing with Kapacitor-unit is as easy as defining the test configuration defining which alerts are expected to trigger when the TICK script ingests specific data. 
+automated. Testing with Kapacitor-unit is as easy as defining the test configuration saying which alerts are expected to trigger when the TICK script processes specific data. 
 
 
 Read more about the idea and motivation behind kapacitor-unit in 

diff --git a/sample/.alert_weather_batch.tick.swp b/sample/.alert_weather_batch.tick.swp
diff --git a/sample/.test_case_batch.yaml.swp b/sample/.test_case_batch.yaml.swp
diff --git a/test/test.go b/test/test.go
@@ -40,10 +40,7 @@ func (t *Test) Run(k io.Kapacitor, i io.Influxdb) error {
 	if err != nil {
 		return err
 	}
-
-	// Change to better mechanism
-	time.Sleep(3 * time.Second)
-
+	t.wait()
 	err = t.results(k)
 	if err != nil {
 		return err
@@ -118,6 +115,14 @@ func (t *Test) setup(k io.Kapacitor, i io.Influxdb) error {
 	return nil
 }
 
+func (t *Test) wait() {
+	switch t.Type {
+		case "batch":
+			// If batch script, waits 3 seconds for batch queries being processed
+			time.Sleep(3 * time.Second)
+	}
+}
+
 // Deletes data, database and retention policies created to run the test
 func (t *Test) teardown(k io.Kapacitor, i io.Influxdb) error {
 	glog.Info("Teardown test:: ", t.Name)