ZeroBot-Plugin/plugin_moyu/holiday_test.go

package moyu

import (
	"fmt"
	"testing"
	"time"

	reg "github.com/fumiama/go-registry"
)

var sr = reg.NewRegedit("reilia.fumiama.top:32664", "fumiama", "--")

func TestGetHoliday(t *testing.T) {
	registry.Connect()
	holidaymap = make(map[string]*Holiday)
	h := GetHoliday("元旦")
	registry.Close()
	t.Fatal(h)
}

func TestSetHoliday(t *testing.T) {
	err := sr.Connect()
	if err != nil {
		t.Fatal(err)
	}

	err = SetHoliday("元旦", 1, 2023, 1, 1)
	if err != nil {
		t.Fatal(err)
	}
	err = SetHoliday("春节", 7, 2022, 1, 31)
	if err != nil {
		t.Fatal(err)
	}
	err = SetHoliday("清明节", 1, 2022, 4, 3)
	if err != nil {
		t.Fatal(err)
	}
	err = SetHoliday("劳动节", 1, 2022, 4, 30)
	if err != nil {
		t.Fatal(err)
	}
	err = SetHoliday("端午节", 1, 2022, 6, 3)
	if err != nil {
		t.Fatal(err)
	}
	err = SetHoliday("中秋节", 1, 2022, 9, 10)
	if err != nil {
		t.Fatal(err)
	}
	err = SetHoliday("国庆节", 7, 2022, 10, 1)
	if err != nil {
		t.Fatal(err)
	}

	err = sr.Close()
	if err != nil {
		t.Fatal(err)
	}
}

func SetHoliday(name string, dur, year int, month time.Month, day int) error {
	return sr.Set("holiday/"+name, fmt.Sprintf("%d_%d_%d_%d", dur, year, month, day))
}