62 lines
949 B
Go
62 lines
949 B
Go
|
package timer
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
var job = mockJob{}
|
||
|
|
|
||
|
|
type mockJob struct{}
|
||
|
|
|
||
|
|
func (job mockJob) Run() {
|
||
|
|
mockFunc()
|
||
|
|
}
|
||
|
|
func mockFunc() {
|
||
|
|
time.Sleep(time.Second)
|
||
|
|
fmt.Println("1s...")
|
||
|
|
}
|
||
|
|
func TestNewTimerTask(t *testing.T) {
|
||
|
|
tm := NewTimerTask()
|
||
|
|
_tm := tm.(*timer)
|
||
|
|
{
|
||
|
|
_, err := tm.AddTaskByFunc("func", "@every 1s", mockFunc)
|
||
|
|
assert.Nil(t, err)
|
||
|
|
_, ok := _tm.taskList["func"]
|
||
|
|
if !ok {
|
||
|
|
t.Error("no find func")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
{
|
||
|
|
_, err := tm.AddTaskByJob("job", "@every 1s", job)
|
||
|
|
assert.Nil(t, err)
|
||
|
|
_, ok := _tm.taskList["job"]
|
||
|
|
if !ok {
|
||
|
|
t.Error("no find job")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
{
|
||
|
|
_, ok := tm.FindCron("func")
|
||
|
|
if !ok {
|
||
|
|
t.Error("no find func")
|
||
|
|
}
|
||
|
|
_, ok = tm.FindCron("job")
|
||
|
|
if !ok {
|
||
|
|
t.Error("no find job")
|
||
|
|
}
|
||
|
|
_, ok = tm.FindCron("none")
|
||
|
|
if ok {
|
||
|
|
t.Error("find none")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
{
|
||
|
|
tm.Clear("func")
|
||
|
|
_, ok := tm.FindCron("func")
|
||
|
|
if ok {
|
||
|
|
t.Error("find func")
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|