Initial commit after restructure

taskmanager/scheduledtasks_test.go

@@ -0,0 +1,95 @@
+// Copyright Safing ICS Technologies GmbH. Use of this source code is governed by the AGPL license that can be found in the LICENSE file.
+
+package taskmanager
+
+import (
+	"sync"
+	"testing"
+	"time"
+)
+
+// test waiting
+
+// globals
+var stWg sync.WaitGroup
+var stOutputChannel chan string
+var stSleepDuration time.Duration
+var stWaitCh chan bool
+
+// functions
+func scheduledTaskTester(s string, sched time.Time) {
+	t := NewScheduledTask(s, sched)
+	go func() {
+		<-stWaitCh
+		<-t.WaitForStart()
+		time.Sleep(stSleepDuration)
+		stOutputChannel <- s
+		t.Done()
+		stWg.Done()
+	}()
+}
+
+// test
+func TestScheduledTaskWaiting(t *testing.T) {
+
+	// skip
+	if testing.Short() {
+		t.Skip("skipping test in short mode.")
+	}
+
+	// init
+	expectedOutput := "0123456789"
+	stSleepDuration = 10 * time.Millisecond
+	stOutputChannel = make(chan string, 100)
+	stWaitCh = make(chan bool, 0)
+
+	// test queue length
+	c := TotalScheduledTasks()
+	if c != 0 {
+		t.Errorf("Error in calculating Task Queue, expected 0, got %d", c)
+	}
+
+	stWg.Add(10)
+
+	// TEST
+	scheduledTaskTester("4", time.Now().Add(stSleepDuration*4))
+	scheduledTaskTester("0", time.Now().Add(stSleepDuration*1))
+	scheduledTaskTester("8", time.Now().Add(stSleepDuration*8))
+	scheduledTaskTester("1", time.Now().Add(stSleepDuration*2))
+	scheduledTaskTester("7", time.Now().Add(stSleepDuration*7))
+
+	// test queue length
+	time.Sleep(1 * time.Millisecond)
+	c = TotalScheduledTasks()
+	if c != 5 {
+		t.Errorf("Error in calculating Task Queue, expected 5, got %d", c)
+	}
+
+	scheduledTaskTester("9", time.Now().Add(stSleepDuration*9))
+	scheduledTaskTester("3", time.Now().Add(stSleepDuration*3))
+	scheduledTaskTester("2", time.Now().Add(stSleepDuration*2))
+	scheduledTaskTester("6", time.Now().Add(stSleepDuration*6))
+	scheduledTaskTester("5", time.Now().Add(stSleepDuration*5))
+
+	// wait for test to finish
+	close(stWaitCh)
+	stWg.Wait()
+
+	// test queue length
+	c = TotalScheduledTasks()
+	if c != 0 {
+		t.Errorf("Error in calculating Task Queue, expected 0, got %d", c)
+	}
+
+	// collect output
+	close(stOutputChannel)
+	completeOutput := ""
+	for s := <-stOutputChannel; s != ""; s = <-stOutputChannel {
+		completeOutput += s
+	}
+	// check if test succeeded
+	if completeOutput != expectedOutput {
+		t.Errorf("ScheduledTask test failed, expected sequence %s, got %s", expectedOutput, completeOutput)
+	}
+
+}