modified: scheduler_test.go

epazote · Sep 16, 2016 · c2f0887 · c2f0887
1 parent 4cf2dc5
commit c2f0887
Showing 1 changed file with 22 additions and 2 deletions.
diff --git a/scheduler_test.go b/scheduler_test.go
@@ -26,7 +26,27 @@ func TestScheduler(t *testing.T) {
 	case <-time.After(3 * time.Second):
 		sk.StopAll()
 	}
-	if c.Get() != 3 {
-		t.Error("Expecting 3")
+	if c.Get() <= 1 {
+		t.Error("Expecting c > 0")
+	}
+}
+
+func TestStopError(t *testing.T) {
+	sk := New()
+	c := &count{}
+	sk.AddScheduler("print", 1, func() { fmt.Println(c.Get()); c.Add(1) })
+	err := sk.Stop("none")
+	if err == nil {
+		t.Error("Expecint error")
+	}
+}
+
+func TestStop(t *testing.T) {
+	sk := New()
+	c := &count{}
+	sk.AddScheduler("print", 1, func() { fmt.Println(c.Get()); c.Add(1) })
+	err := sk.Stop("print")
+	if err != nil {
+		t.Error(err)
 	}
 }