112 lines
1.8 KiB
Go
112 lines
1.8 KiB
Go
package scheduler
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"local/logb"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
func TestNewBashJobBadCron(t *testing.T) {
|
|
_, err := newBashJob("1 1 1 ", "hostname")
|
|
if err == nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func TestNewBashJobAndRun(t *testing.T) {
|
|
cases := []struct {
|
|
sched string
|
|
cmd string
|
|
out string
|
|
}{
|
|
{
|
|
sched: "1 1 1 1 1",
|
|
cmd: "hostname",
|
|
out: os.Getenv("HOSTNAME"),
|
|
},
|
|
{
|
|
sched: "@hourly",
|
|
cmd: "hostname",
|
|
out: os.Getenv("HOSTNAME"),
|
|
},
|
|
{
|
|
sched: "@hourly",
|
|
cmd: "exit 1",
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
b, clean := captureLog()
|
|
defer clean()
|
|
j, err := newBashJob(c.sched, c.cmd)
|
|
if err != nil {
|
|
t.Error(err)
|
|
continue
|
|
}
|
|
func() {
|
|
defer func() {
|
|
recover()
|
|
}()
|
|
j.Run()
|
|
}()
|
|
if !bytes.Contains(b.Bytes(), []byte(c.out)) {
|
|
t.Errorf("(%s, %s) => %s", c.sched, c.cmd, b.Bytes())
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestJobEncodeDecode(t *testing.T) {
|
|
buff, clean := captureLog()
|
|
defer clean()
|
|
|
|
j, err := newBashJob("1 1 1 1 1", "hostname")
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
j.foo()
|
|
if !bytes.Contains(buff.Bytes(), []byte(os.Getenv("HOSTNAME"))) {
|
|
t.Errorf("%s", buff.Bytes())
|
|
}
|
|
buff.Reset()
|
|
|
|
b, err := j.Encode()
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
k := &Job{}
|
|
if err := k.Decode(b); err != nil {
|
|
t.Logf("decoded %+v", k)
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if k.Schedule != j.Schedule {
|
|
t.Error(k.Schedule, "vs", j.Schedule)
|
|
}
|
|
if k.Name != j.Name {
|
|
t.Error(k.Name, "vs", j.Name)
|
|
}
|
|
|
|
k.foo()
|
|
if !bytes.Contains(buff.Bytes(), []byte(os.Getenv("HOSTNAME"))) {
|
|
t.Errorf("%s", buff.Bytes())
|
|
}
|
|
}
|
|
|
|
func captureLog() (*bytes.Buffer, func()) {
|
|
was := logb.Writer()
|
|
wase := os.Stderr
|
|
f, _ := ioutil.TempFile(os.TempDir(), "test.newBashJobAndRun")
|
|
os.Stderr = f
|
|
b := bytes.NewBuffer(nil)
|
|
logb.SetWriter(b)
|
|
return b, func() {
|
|
os.Remove(f.Name())
|
|
logb.SetWriter(was)
|
|
os.Stderr = wase
|
|
}
|
|
}
|