Test getNext

distribworks · Mar 12, 2019 · f6a5536 · f6a5536
1 parent 0435228
commit f6a5536
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/dkron/job.go b/dkron/job.go
@@ -264,6 +264,7 @@ func (j *Job) Unlock() error {
 	return nil
 }
 
+// GetNext returns the job's next schedule
 func (j *Job) GetNext() (time.Time, error) {
 	s, err := cron.Parse(j.Schedule)
 	if err != nil {

diff --git a/dkron/job_test.go b/dkron/job_test.go
@@ -2,6 +2,7 @@ package dkron
 
 import (
 	"testing"
+	"time"
 
 	s "github.com/abronan/valkeyrie/store"
 	"github.com/stretchr/testify/assert"
@@ -66,3 +67,16 @@ func TestJobGetParent(t *testing.T) {
 	assert.NoError(t, err)
 	assert.Equal(t, []string{}, ptj.DependentJobs)
 }
+
+func TestJobGetNext(t *testing.T) {
+	j := Job{
+		Schedule: "@daily",
+	}
+
+	td := time.Now()
+	tonight := time.Date(td.Year(), td.Month(), td.Day()+1, 0, 0, 0, 0, td.Location())
+	n, err := j.GetNext()
+
+	assert.NoError(t, err)
+	assert.Equal(t, tonight, n)
+}