gf/g/os/gcron/gcron_unit_1_test.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_AddOnce(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_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)
    })
}