Fix tests to pass on linux

This commit is contained in:
bel
2020-03-13 03:40:53 +00:00
parent 6d39ef9aa2
commit 0d6be1e9d8
6 changed files with 185 additions and 3 deletions

View File

@@ -2,8 +2,10 @@ package scheduler
import (
"bytes"
"io/ioutil"
"local/firestormy/config"
"local/storage"
"os"
"testing"
)
@@ -59,3 +61,91 @@ func TestSchedulerStartStop(t *testing.T) {
t.Errorf("%v: %s", n, b.Bytes())
}
}
func TestSchedulerFromFile(t *testing.T) {
was := config.Store
defer func() {
config.Store = was
}()
cases := map[string]struct {
content string
want int
}{
"just a job": {
content: `10 */12 * * * /bin/bash -c "hostname"`,
want: 1,
},
"all wild": {
content: `* * * * * /bin/bash -c "hostname"`,
want: 1,
},
"all single numbers": {
content: `1 1 1 1 1 /bin/bash -c "hostname"`,
want: 1,
},
"all double numbers": {
content: `10 10 10 10 2 /bin/bash -c "hostname"`,
want: 1,
},
"all /\\d+": {
content: `*/11 */2 */3 */4 */1 /bin/bash -c "hostname"`,
want: 1,
},
"2 jobs with 1 comment no whitespace leading": {
content: `# this is my comment
10 */12 * * * /bin/bash -c "hostname"
10 */12 * * * /bin/bash -c "hostname"
`,
want: 2,
},
"2 jobs with 1 comment whitespace leading": {
content: ` # this is my comment
10 */12 * * * /bin/bash -c "hostname"
10 */12 * * * /bin/bash -c "hostname"
`,
want: 2,
},
"2 jobs with crazy whitespace between cron spec": {
content: ` # this is my comment
10 */12 * * * /bin/bash -c "hostname"
10 */12 * * * /bin/bash -c "hostname"
`,
want: 2,
},
"2 jobs with 2 comemnts and 2 empty lines": {
content: ` # this is my comment
# this is a second comment
10 */12 * * * /bin/bash -c "hostname"
10 */12 * * * /bin/bash -c "hostname"
`,
want: 2,
},
}
for name, c := range cases {
t.Run(name, func(t *testing.T) {
config.Store, _ = storage.New(storage.MAP)
f, err := ioutil.TempFile(os.TempDir(), "testSchedulerFromFile")
if err != nil {
t.Fatal(err)
}
defer os.Remove(f.Name())
f.Write([]byte(c.content))
f.Close()
s, err := NewFromFile(f.Name())
if err != nil {
t.Fatal(err)
}
jobs, err := s.List()
if err != nil {
t.Fatal(err)
}
if len(jobs) != c.want {
t.Fatalf("want %v, got %v jobs", c.want, jobs)
}
})
}
}