gf/g/os/gcron/gcron_unit_1_test.go

233 lines
6.6 KiB
Go

// Copyright 2018 gf Author(https://github.com/gogf/gf). All Rights Reserved.
//
// This Source Code Form is subject to the terms of the MIT License.
// If a copy of the MIT was not distributed with this file,
// You can obtain one at https://github.com/gogf/gf.
package gcron_test
import (
"github.com/gogf/gf/g/container/garray"
"github.com/gogf/gf/g/os/gcron"
"github.com/gogf/gf/g/test/gtest"
"testing"
"time"
)
func TestCron_Add_Close(t *testing.T) {
gtest.Case(t, func() {
cron := gcron.New()
array := garray.New()
_, err1 := cron.Add("* * * * * *", func() {
//glog.Println("cron1")
array.Append(1)
})
_, err2 := cron.Add("* * * * * *", func() {
//glog.Println("cron2")
array.Append(1)
}, "test")
_, err3 := cron.Add("* * * * * *", func() {
array.Append(1)
}, "test")
_, err4 := cron.Add("@every 2s", func() {
//glog.Println("cron3")
array.Append(1)
})
gtest.Assert(err1, nil)
gtest.Assert(err2, nil)
gtest.AssertNE(err3, nil)
gtest.Assert(err4, nil)
gtest.Assert(cron.Size(), 3)
time.Sleep(1200*time.Millisecond)
gtest.Assert(array.Len(), 2)
time.Sleep(1200*time.Millisecond)
gtest.Assert(array.Len(), 5)
cron.Close()
time.Sleep(1200*time.Millisecond)
fixedLength := array.Len()
time.Sleep(1200*time.Millisecond)
gtest.Assert(array.Len(), fixedLength)
})
}
func TestCron_Basic(t *testing.T) {
gtest.Case(t, func() {
cron := gcron.New()
cron.Add("* * * * * *", func() {}, "add")
//fmt.Println("start", time.Now())
cron.DelayAdd(time.Second, "* * * * * *", func() {}, "delay_add")
gtest.Assert(cron.Size(), 1)
time.Sleep(1200*time.Millisecond)
gtest.Assert(cron.Size(), 2)
cron.Remove("delay_add")
gtest.Assert(cron.Size(), 1)
entry1 := cron.Search("add")
entry2 := cron.Search("test-none")
gtest.AssertNE(entry1, nil)
gtest.Assert(entry2, nil)
})
}
func TestCron_Remove(t *testing.T) {
gtest.Case(t, func() {
cron := gcron.New()
array := garray.New()
cron.Add("* * * * * *", func() {
array.Append(1)
}, "add")
gtest.Assert(array.Len(), 0)
time.Sleep(1200*time.Millisecond)
gtest.Assert(array.Len(), 1)
cron.Remove("add")
gtest.Assert(array.Len(), 1)
time.Sleep(1200*time.Millisecond)
gtest.Assert(array.Len(), 1)
})
}
func TestCron_AddSingleton(t *testing.T) {
// un used, can be removed
gtest.Case(t, func() {
cron := gcron.New()
cron.Add("* * * * * *", func() {}, "add")
cron.DelayAdd(time.Second, "* * * * * *", func() {}, "delay_add")
gtest.Assert(cron.Size(), 1)
time.Sleep(1200*time.Millisecond)
gtest.Assert(cron.Size(), 2)
cron.Remove("delay_add")
gtest.Assert(cron.Size(), 1)
entry1 := cron.Search("add")
entry2 := cron.Search("test-none")
gtest.AssertNE(entry1, nil)
gtest.Assert(entry2, nil)
})
// keep this
gtest.Case(t, func() {
cron := gcron.New()
array := garray.New()
cron.AddSingleton("* * * * * *", func() {
array.Append(1)
time.Sleep(50*time.Second)
})
gtest.Assert(cron.Size(), 1)
time.Sleep(3500*time.Millisecond)
gtest.Assert(array.Len(), 1)
})
}
func TestCron_AddOnce1(t *testing.T) {
gtest.Case(t, func() {
cron := gcron.New()
array := garray.New()
cron.AddOnce("* * * * * *", func() {
array.Append(1)
})
cron.AddOnce("* * * * * *", func() {
array.Append(1)
})
gtest.Assert(cron.Size(), 2)
time.Sleep(2500*time.Millisecond)
gtest.Assert(array.Len(), 2)
gtest.Assert(cron.Size(), 0)
})
}
func TestCron_AddOnce2(t *testing.T) {
gtest.Case(t, func() {
cron := gcron.New()
array := garray.New()
cron.AddOnce("@every 2s", func() {
array.Append(1)
})
gtest.Assert(cron.Size(), 1)
time.Sleep(2500*time.Millisecond)
gtest.Assert(array.Len(), 1)
gtest.Assert(cron.Size(), 0)
})
}
func TestCron_AddTimes(t *testing.T) {
gtest.Case(t, func() {
cron := gcron.New()
array := garray.New()
cron.AddTimes("* * * * * *", 2, func() {
array.Append(1)
})
time.Sleep(3500*time.Millisecond)
gtest.Assert(array.Len(), 2)
gtest.Assert(cron.Size(), 0)
})
}
func TestCron_DelayAdd(t *testing.T) {
gtest.Case(t, func() {
cron := gcron.New()
array := garray.New()
cron.DelayAdd(500*time.Millisecond, "* * * * * *", func() {
array.Append(1)
})
gtest.Assert(cron.Size(), 0)
time.Sleep(800*time.Millisecond)
gtest.Assert(array.Len(), 0)
gtest.Assert(cron.Size(), 1)
time.Sleep(1000*time.Millisecond)
gtest.Assert(array.Len(), 1)
gtest.Assert(cron.Size(), 1)
})
}
func TestCron_DelayAddSingleton(t *testing.T) {
gtest.Case(t, func() {
cron := gcron.New()
array := garray.New()
cron.DelayAddSingleton(500*time.Millisecond, "* * * * * *", func() {
array.Append(1)
time.Sleep(10*time.Second)
})
gtest.Assert(cron.Size(), 0)
time.Sleep(2200*time.Millisecond)
gtest.Assert(array.Len(), 1)
gtest.Assert(cron.Size(), 1)
})
}
func TestCron_DelayAddOnce(t *testing.T) {
gtest.Case(t, func() {
cron := gcron.New()
array := garray.New()
cron.DelayAddOnce(500*time.Millisecond, "* * * * * *", func() {
array.Append(1)
})
gtest.Assert(cron.Size(), 0)
time.Sleep(800*time.Millisecond)
gtest.Assert(array.Len(), 0)
gtest.Assert(cron.Size(), 1)
time.Sleep(2200*time.Millisecond)
gtest.Assert(array.Len(), 1)
gtest.Assert(cron.Size(), 0)
})
}
func TestCron_DelayAddTimes(t *testing.T) {
gtest.Case(t, func() {
cron := gcron.New()
array := garray.New()
cron.DelayAddTimes(500*time.Millisecond, "* * * * * *", 2, func() {
array.Append(1)
})
gtest.Assert(cron.Size(), 0)
time.Sleep(800*time.Millisecond)
gtest.Assert(array.Len(), 0)
gtest.Assert(cron.Size(), 1)
time.Sleep(3000*time.Millisecond)
gtest.Assert(array.Len(), 2)
gtest.Assert(cron.Size(), 0)
})
}