github.com/zhongdalu/gf@v1.0.0/g/os/gcron/gcron_unit_1_test.go

// Copyright 2018 gf Author(https://github.com/zhongdalu/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/zhongdalu/gf.

package gcron_test

import (
	"testing"
	"time"

	"github.com/zhongdalu/gf/g/container/garray"
	"github.com/zhongdalu/gf/g/os/gcron"
	"github.com/zhongdalu/gf/g/test/gtest"
)

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(3000 * 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)
	})
}