mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
890 lines
24 KiB
Go
890 lines
24 KiB
Go
/*
|
|
Real-time Online/Offline Charging System (OCS) for Telecom & ISP environments
|
|
Copyright (C) ITsysCOM GmbH
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU Affero General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU Affero General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Affero General Public License
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>
|
|
*/
|
|
package engine
|
|
|
|
import (
|
|
"reflect"
|
|
"sort"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/cgrates/cgrates/config"
|
|
"github.com/cgrates/cgrates/utils"
|
|
"github.com/cgrates/ltcache"
|
|
)
|
|
|
|
func TestActionTimingTasks(t *testing.T) {
|
|
//empty check
|
|
actionTiming := new(ActionTiming)
|
|
eOut := []*Task{{Uuid: "", ActionsID: ""}}
|
|
rcv := actionTiming.Tasks()
|
|
if !reflect.DeepEqual(eOut, rcv) {
|
|
t.Errorf("Expecting: %+v, received: %+v", eOut, rcv)
|
|
}
|
|
//multiple check
|
|
actionTiming.ActionsID = "test"
|
|
actionTiming.Uuid = "test"
|
|
actionTiming.accountIDs = utils.StringMap{"1001": true, "1002": true, "1003": true}
|
|
eOut = []*Task{
|
|
{Uuid: "test", AccountID: "1001", ActionsID: "test"},
|
|
{Uuid: "test", AccountID: "1002", ActionsID: "test"},
|
|
{Uuid: "test", AccountID: "1003", ActionsID: "test"},
|
|
}
|
|
rcv = actionTiming.Tasks()
|
|
sort.Slice(rcv, func(i, j int) bool { return rcv[i].AccountID < rcv[j].AccountID })
|
|
if !reflect.DeepEqual(eOut, rcv) {
|
|
t.Errorf("Expecting: %+v, received: %+v", eOut, rcv)
|
|
}
|
|
}
|
|
|
|
func TestActionTimingRemoveAccountID(t *testing.T) {
|
|
actionTiming := &ActionTiming{
|
|
accountIDs: utils.StringMap{"1001": true, "1002": true, "1003": true},
|
|
}
|
|
eOut := utils.StringMap{"1002": true, "1003": true}
|
|
rcv := actionTiming.RemoveAccountID("1001")
|
|
if !rcv {
|
|
t.Errorf("Account ID not found ")
|
|
}
|
|
if !reflect.DeepEqual(eOut, actionTiming.accountIDs) {
|
|
t.Errorf("Expecting: %+v, received: %+v", eOut, actionTiming.accountIDs)
|
|
}
|
|
//check for Account ID not found
|
|
rcv = actionTiming.RemoveAccountID("1001")
|
|
if rcv {
|
|
t.Errorf("Expected AccountID to be not found")
|
|
}
|
|
}
|
|
|
|
func TestActionPlanRemoveAccountID(t *testing.T) {
|
|
actionPlan := &ActionPlan{
|
|
AccountIDs: utils.StringMap{"1001": true, "1002": true, "1003": true},
|
|
}
|
|
eOut := utils.StringMap{"1002": true, "1003": true}
|
|
rcv := actionPlan.RemoveAccountID("1001")
|
|
if !rcv {
|
|
t.Errorf("Account ID not found ")
|
|
}
|
|
if !reflect.DeepEqual(eOut, actionPlan.AccountIDs) {
|
|
t.Errorf("Expecting: %+v, received: %+v", eOut, actionPlan.AccountIDs)
|
|
}
|
|
//check for Account ID not found
|
|
rcv = actionPlan.RemoveAccountID("1001")
|
|
if rcv {
|
|
t.Errorf("Expected AccountID to be not found")
|
|
}
|
|
}
|
|
|
|
func TestActionPlanClone(t *testing.T) {
|
|
at1 := &ActionPlan{
|
|
Id: "test",
|
|
AccountIDs: utils.StringMap{"one": true, "two": true, "three": true},
|
|
ActionTimings: []*ActionTiming{
|
|
{
|
|
Uuid: "Uuid_test1",
|
|
ActionsID: "ActionsID_test1",
|
|
Weight: 0.8,
|
|
Timing: &RateInterval{
|
|
Weight: 0.7,
|
|
},
|
|
},
|
|
},
|
|
}
|
|
clned := at1.Clone()
|
|
if !reflect.DeepEqual(at1, clned) {
|
|
t.Errorf("Expecting: %+v,\n received: %+v", at1, clned)
|
|
}
|
|
}
|
|
|
|
func TestActionTimingClone(t *testing.T) {
|
|
var at *ActionTiming
|
|
val := at.Clone()
|
|
if val != nil {
|
|
t.Errorf("expected nil ,received %v", val)
|
|
}
|
|
at = &ActionTiming{
|
|
Uuid: "Uuid_test",
|
|
ActionsID: "ActionsID_test",
|
|
Weight: 0.7,
|
|
}
|
|
if cloned := at.Clone(); !reflect.DeepEqual(at, cloned) {
|
|
t.Errorf("Expecting: %+v,\n received: %+v", at, cloned)
|
|
}
|
|
}
|
|
|
|
func TestActionTimindSetActions(t *testing.T) {
|
|
actionTiming := new(ActionTiming)
|
|
|
|
actions := Actions{
|
|
&Action{ActionType: "test"},
|
|
&Action{ActionType: "test1"},
|
|
}
|
|
actionTiming.SetActions(actions)
|
|
if !reflect.DeepEqual(actions, actionTiming.actions) {
|
|
t.Errorf("Expecting: %+v, received: %+v", actions, actionTiming.actions)
|
|
}
|
|
}
|
|
|
|
func TestActionTimingSetAccountIDs(t *testing.T) {
|
|
actionTiming := new(ActionTiming)
|
|
accountIDs := utils.StringMap{"one": true, "two": true, "three": true}
|
|
actionTiming.SetAccountIDs(accountIDs)
|
|
|
|
if !reflect.DeepEqual(accountIDs, actionTiming.accountIDs) {
|
|
t.Errorf("Expecting: %+v, received: %+v", accountIDs, actionTiming.accountIDs)
|
|
}
|
|
}
|
|
|
|
func TestActionTimingGetAccountIDs(t *testing.T) {
|
|
actionTiming := &ActionTiming{
|
|
accountIDs: utils.StringMap{"one": true, "two": true, "three": true},
|
|
}
|
|
accIDs := utils.StringMap{"one": true, "two": true, "three": true}
|
|
rcv := actionTiming.GetAccountIDs()
|
|
|
|
if !reflect.DeepEqual(accIDs, rcv) {
|
|
t.Errorf("Expecting: %+v, received: %+v", accIDs, rcv)
|
|
}
|
|
}
|
|
func TestActionTimingSetActionPlanID(t *testing.T) {
|
|
actionTiming := new(ActionTiming)
|
|
id := "test"
|
|
actionTiming.SetActionPlanID(id)
|
|
if !reflect.DeepEqual(id, actionTiming.actionPlanID) {
|
|
t.Errorf("Expecting: %+v, received: %+v", id, actionTiming.actionPlanID)
|
|
}
|
|
}
|
|
|
|
func TestActionTimingGetActionPlanID(t *testing.T) {
|
|
id := "test"
|
|
actionTiming := new(ActionTiming)
|
|
actionTiming.actionPlanID = id
|
|
|
|
rcv := actionTiming.GetActionPlanID()
|
|
if !reflect.DeepEqual(id, rcv) {
|
|
t.Errorf("Expecting: %+v, received: %+v", id, rcv)
|
|
}
|
|
}
|
|
|
|
func TestActionTimingIsASAP(t *testing.T) {
|
|
actionTiming := new(ActionTiming)
|
|
if rcv := actionTiming.IsASAP(); rcv {
|
|
t.Error("Expecting false return")
|
|
}
|
|
}
|
|
|
|
func TestAtplLen(t *testing.T) {
|
|
atpl := &ActionTimingWeightOnlyPriorityList{
|
|
&ActionTiming{Uuid: "first", accountIDs: utils.StringMap{"1001": true, "1002": true}},
|
|
&ActionTiming{Uuid: "second", accountIDs: utils.StringMap{"1004": true, "1005": true}},
|
|
&ActionTiming{Uuid: "third", accountIDs: utils.StringMap{"1001": true, "1002": true}},
|
|
}
|
|
eOut := len(*atpl)
|
|
rcv := atpl.Len()
|
|
if !reflect.DeepEqual(eOut, rcv) {
|
|
t.Errorf("Expecting: %+v, received: %+v", eOut, rcv)
|
|
}
|
|
}
|
|
func TestAtplSwap(t *testing.T) {
|
|
atpl := &ActionTimingWeightOnlyPriorityList{
|
|
&ActionTiming{Uuid: "first", accountIDs: utils.StringMap{"1001": true, "1002": true}},
|
|
&ActionTiming{Uuid: "second", accountIDs: utils.StringMap{"1004": true, "1005": true}},
|
|
}
|
|
atpl2 := &ActionTimingWeightOnlyPriorityList{
|
|
&ActionTiming{Uuid: "second", accountIDs: utils.StringMap{"1004": true, "1005": true}},
|
|
&ActionTiming{Uuid: "first", accountIDs: utils.StringMap{"1001": true, "1002": true}},
|
|
}
|
|
atpl.Swap(0, 1)
|
|
if !reflect.DeepEqual(atpl, atpl2) {
|
|
t.Errorf("Expecting: %+v, received: %+v", atpl, atpl2)
|
|
}
|
|
}
|
|
|
|
func TestAtplLess(t *testing.T) {
|
|
atpl := &ActionTimingWeightOnlyPriorityList{
|
|
&ActionTiming{Uuid: "first", Weight: 0.07},
|
|
&ActionTiming{Uuid: "second", Weight: 1.07},
|
|
}
|
|
rcv := atpl.Less(1, 0)
|
|
if !rcv {
|
|
t.Errorf("Expecting false, Received: true")
|
|
}
|
|
rcv = atpl.Less(0, 1)
|
|
if rcv {
|
|
t.Errorf("Expecting true, Received: false")
|
|
}
|
|
}
|
|
|
|
func TestAtplSort(t *testing.T) {
|
|
|
|
atpl := &ActionTimingWeightOnlyPriorityList{
|
|
&ActionTiming{Uuid: "first", accountIDs: utils.StringMap{"1001": true, "1002": true}},
|
|
&ActionTiming{Uuid: "second", accountIDs: utils.StringMap{"1004": true, "1005": true}},
|
|
}
|
|
atpl2 := &ActionTimingWeightOnlyPriorityList{
|
|
&ActionTiming{Uuid: "first", accountIDs: utils.StringMap{"1001": true, "1002": true}},
|
|
&ActionTiming{Uuid: "second", accountIDs: utils.StringMap{"1004": true, "1005": true}},
|
|
}
|
|
|
|
sort.Sort(atpl)
|
|
atpl2.Sort()
|
|
if !reflect.DeepEqual(atpl, atpl2) {
|
|
t.Errorf("Expecting: %+v, received: %+v", atpl, atpl2)
|
|
}
|
|
}
|
|
|
|
func TestCacheGetCloned(t *testing.T) {
|
|
at1 := &ActionPlan{
|
|
Id: "test",
|
|
AccountIDs: utils.StringMap{"one": true, "two": true, "three": true},
|
|
}
|
|
if err := Cache.Set(utils.CacheActionPlans, "MYTESTAPL", at1, nil, true, ""); err != nil {
|
|
t.Errorf("Expecting nil, received: %s", err)
|
|
}
|
|
clned, ok := Cache.Get(utils.CacheActionPlans, "MYTESTAPL")
|
|
if !ok {
|
|
t.Error(ltcache.ErrNotFound)
|
|
}
|
|
at1Cloned := clned.(*ActionPlan)
|
|
if !reflect.DeepEqual(at1, at1Cloned) {
|
|
t.Errorf("Expecting: %+v, received: %+v", at1, at1Cloned)
|
|
}
|
|
}
|
|
|
|
func TestActionTimingExErr(t *testing.T) {
|
|
tmpDm := dm
|
|
tmp := Cache
|
|
cfg := config.NewDefaultCGRConfig()
|
|
defer func() {
|
|
dm = tmpDm
|
|
Cache = tmp
|
|
config.SetCgrConfig(config.NewDefaultCGRConfig())
|
|
}()
|
|
db, dErr := NewInternalDB(nil, nil, true, nil, cfg.DataDbCfg().Items)
|
|
if dErr != nil {
|
|
t.Error(dErr)
|
|
}
|
|
dm := NewDataManager(db, cfg.CacheCfg(), nil)
|
|
at := &ActionTiming{
|
|
Timing: &RateInterval{},
|
|
actions: []*Action{
|
|
{
|
|
Filters: []string{},
|
|
ExpirationString: "*yearly",
|
|
ActionType: "test",
|
|
Id: "ZeroMonetary",
|
|
Balance: &BalanceFilter{
|
|
Type: utils.StringPointer(utils.MetaMonetary),
|
|
|
|
Value: &utils.ValueFormula{Static: 11},
|
|
Weight: utils.Float64Pointer(30),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
fltrs := NewFilterS(cfg, nil, dm)
|
|
if err := at.Execute(nil, ""); err == nil || err != utils.ErrPartiallyExecuted {
|
|
t.Error(err)
|
|
}
|
|
at.actions[0].ActionType = utils.MetaDebitReset
|
|
if err := at.Execute(nil, ""); err == nil || err != utils.ErrPartiallyExecuted {
|
|
t.Error(err)
|
|
}
|
|
at.accountIDs = utils.StringMap{"cgrates.org:zeroNegative": true}
|
|
at.actions[0].ActionType = utils.MetaResetStatQueue
|
|
if err := at.Execute(nil, ""); err == nil || err != utils.ErrPartiallyExecuted {
|
|
t.Error(err)
|
|
}
|
|
Cache.Set(utils.CacheFilters, "cgrates.org:*string:~*req.BalanceMap.*monetary[0].ID:*default", nil, []string{}, true, utils.NonTransactional)
|
|
at.actions[0].Filters = []string{"*string:~*req.BalanceMap.*monetary[0].ID:*default"}
|
|
if err := at.Execute(fltrs, ""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
SetDataStorage(nil)
|
|
if err := at.Execute(nil, ""); err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestGetDayOrEndOfMonth(t *testing.T) {
|
|
if val := getDayOrEndOfMonth(31, time.Date(2022, 12, 22, 12, 0, 0, 0, time.UTC)); val != 31 {
|
|
t.Errorf("Should Receive Last Day %v", val)
|
|
}
|
|
}
|
|
|
|
func TestActionTimingGetNextStartTimesMonthlyEstimated(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
t1 time.Time
|
|
at *ActionTiming
|
|
expected time.Time
|
|
}{
|
|
{
|
|
name: "February 7 to February 29",
|
|
t1: time.Date(2020, 2, 7, 14, 25, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{31},
|
|
StartTime: "00:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2020, 2, 29, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "February 17 to March 16",
|
|
t1: time.Date(2020, 2, 17, 14, 25, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{16},
|
|
StartTime: "00:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2020, 3, 16, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "December 17 to January 16",
|
|
t1: time.Date(2020, 12, 17, 14, 25, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{16},
|
|
StartTime: "00:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2021, 1, 16, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "December 17 to December 31",
|
|
t1: time.Date(2020, 12, 17, 14, 25, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{31},
|
|
StartTime: "00:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2020, 12, 31, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "July 31 same day",
|
|
t1: time.Date(2020, 7, 31, 14, 25, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{31},
|
|
StartTime: "15:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2020, 7, 31, 15, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "February 17 same day",
|
|
t1: time.Date(2020, 2, 17, 14, 25, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{17},
|
|
StartTime: "15:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2020, 2, 17, 15, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "February 17 to March 17",
|
|
t1: time.Date(2020, 2, 17, 15, 25, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{17},
|
|
StartTime: "10:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2020, 3, 17, 10, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "September 29 to September 30",
|
|
t1: time.Date(2020, 9, 29, 14, 25, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{31},
|
|
StartTime: "00:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2020, 9, 30, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "September 30 to October 31",
|
|
t1: time.Date(2020, 9, 30, 14, 25, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{31},
|
|
StartTime: "00:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2020, 10, 31, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "September 30 same day",
|
|
t1: time.Date(2020, 9, 30, 14, 25, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{31},
|
|
StartTime: "15:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2020, 9, 30, 15, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "September 30 same second",
|
|
t1: time.Date(2020, 9, 30, 14, 25, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{31},
|
|
StartTime: "14:25:01",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2020, 9, 30, 14, 25, 1, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "December 31 same second",
|
|
t1: time.Date(2020, 12, 31, 14, 25, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{31},
|
|
StartTime: "14:25:01",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2020, 12, 31, 14, 25, 1, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "December 31 to January 31",
|
|
t1: time.Date(2020, 12, 31, 14, 25, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{31},
|
|
StartTime: "14:25:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2021, 1, 31, 14, 25, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "Non-Leap Year: January 29 to Feb 28",
|
|
t1: time.Date(2021, 1, 29, 0, 0, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{29},
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2021, 2, 28, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "Non-Leap Year: February 28 to March 28",
|
|
t1: time.Date(2021, 2, 28, 0, 0, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{28},
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2021, 3, 28, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "October 31 to November 30",
|
|
t1: time.Date(2021, 10, 31, 0, 0, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{30},
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2021, 11, 30, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "Leap Year: January 29 to February 29",
|
|
t1: time.Date(2020, 1, 29, 0, 0, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{29},
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2020, 2, 29, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "Leap Year: February 29 to March 29",
|
|
t1: time.Date(2020, 2, 29, 0, 0, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{29},
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2020, 3, 29, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "Leap Year: February 29 to March 30",
|
|
t1: time.Date(2020, 2, 29, 0, 0, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{30},
|
|
StartTime: "00:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2020, 3, 30, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "November 30 to December 31",
|
|
t1: time.Date(2021, 11, 30, 0, 0, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{31},
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2021, 12, 31, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "September 29 to September 30",
|
|
t1: time.Date(2021, 9, 29, 0, 0, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{31},
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2021, 9, 30, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
|
|
{
|
|
name: "Non-Leap Year: Jan 29 to Feb 28",
|
|
t1: time.Date(2021, 1, 29, 0, 0, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{29},
|
|
StartTime: "00:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2021, 2, 28, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "Leap Year: Jan 30 to Feb 29 ",
|
|
t1: time.Date(2020, 1, 30, 0, 0, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{30},
|
|
StartTime: "00:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2020, 2, 29, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "Non-Leap Year: Jan 30 to Feb 28 ",
|
|
t1: time.Date(2021, 1, 30, 0, 0, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{30},
|
|
StartTime: "00:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2021, 2, 28, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "Non-Leap Year: Jan 15 to Feb 15 ",
|
|
t1: time.Date(2021, 1, 15, 0, 0, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{15},
|
|
StartTime: "00:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2021, 2, 15, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "Jan 14 to Jan 15 ",
|
|
t1: time.Date(2021, 1, 14, 0, 0, 0, 0, time.UTC),
|
|
at: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
MonthDays: utils.MonthDays{15},
|
|
StartTime: "00:00:00",
|
|
},
|
|
},
|
|
},
|
|
expected: time.Date(2021, 1, 15, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
if st := tt.at.GetNextStartTime(tt.t1); !st.Equal(tt.expected) {
|
|
t.Errorf("Expecting: %+v, received: %+v", tt.expected, st)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestVerifyFormat(t *testing.T) {
|
|
tests := []struct {
|
|
tStr string
|
|
expectedBool bool
|
|
}{
|
|
|
|
{"12:34:56", true},
|
|
{"23:59:59", true},
|
|
{"12:34", false},
|
|
{"12:34:56:78", false},
|
|
{"12:abc:56", false},
|
|
{"123:456:789", false},
|
|
{"00:00:00", true},
|
|
{"12:34:56", true},
|
|
{"t:01:t", false},
|
|
{"1,1,1", false},
|
|
{"0:0:0", true},
|
|
{"119911", false},
|
|
{"00/01/03", false},
|
|
{"t1:t2:t3", false},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.tStr, func(t *testing.T) {
|
|
result := verifyFormat(tt.tStr)
|
|
if result != tt.expectedBool {
|
|
t.Errorf("verifyFormat(%q) = %v; want %v", tt.tStr, result, tt.expectedBool)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestCheckDefaultTiming(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
tStr string
|
|
wantID string
|
|
wantIsDef bool
|
|
}{
|
|
{"Every Minute", utils.MetaEveryMinute, utils.MetaEveryMinute, true},
|
|
{"Hourly", utils.MetaHourly, utils.MetaHourly, true},
|
|
{"Daily", utils.MetaDaily, utils.MetaDaily, true},
|
|
{"Weekly", utils.MetaWeekly, utils.MetaWeekly, true},
|
|
{"Monthly", utils.MetaMonthly, utils.MetaMonthly, true},
|
|
{"Monthly Estimated", utils.MetaMonthlyEstimated, utils.MetaMonthlyEstimated, true},
|
|
{"Month End", utils.MetaMonthEnd, utils.MetaMonthEnd, true},
|
|
{"Yearly", utils.MetaYearly, utils.MetaYearly, true},
|
|
{"Unknown", "unknown", "", false},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, isDef := checkDefaultTiming(tt.tStr)
|
|
if isDef != tt.wantIsDef {
|
|
t.Errorf("checkDefaultTiming(%q) isDefault = %v, want %v", tt.tStr, isDef, tt.wantIsDef)
|
|
}
|
|
if isDef && got.ID != tt.wantID {
|
|
t.Errorf("checkDefaultTiming(%q) got.ID = %v, want %v", tt.tStr, got.ID, tt.wantID)
|
|
}
|
|
if !isDef && got != nil {
|
|
t.Errorf("checkDefaultTiming(%q) expected nil, got non-nil", tt.tStr)
|
|
}
|
|
})
|
|
}
|
|
}
|
|
func TestActionTimingGetNextStartTime2(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
actionTiming *ActionTiming
|
|
startTime time.Time
|
|
expectedTime time.Time
|
|
expectedError bool
|
|
}{
|
|
{
|
|
name: "MonthlyEstimatedTiming",
|
|
actionTiming: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
ID: utils.MetaMonthlyEstimated,
|
|
StartTime: "00:00:00",
|
|
Years: []int{2024},
|
|
Months: []time.Month{1, 2, 3},
|
|
MonthDays: []int{31},
|
|
},
|
|
},
|
|
},
|
|
startTime: time.Date(2024, 1, 15, 0, 0, 0, 0, time.UTC),
|
|
expectedTime: time.Date(2024, 1, 31, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "WeekDaysCron",
|
|
actionTiming: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
StartTime: "09:00:00",
|
|
Years: []int{2024},
|
|
Months: []time.Month{1, 2, 3},
|
|
MonthDays: []int{1, 15},
|
|
WeekDays: []time.Weekday{time.Monday, time.Wednesday, time.Friday},
|
|
EndTime: "17:00:00",
|
|
},
|
|
},
|
|
},
|
|
startTime: time.Date(2024, 1, 1, 10, 0, 0, 0, time.UTC),
|
|
expectedTime: time.Date(2024, 1, 3, 9, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "YearTransition",
|
|
actionTiming: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
StartTime: "00:00:00",
|
|
Years: []int{2024, 2025},
|
|
Months: []time.Month{12, 1},
|
|
MonthDays: []int{31, 1},
|
|
},
|
|
},
|
|
},
|
|
startTime: time.Date(2024, 12, 31, 23, 0, 0, 0, time.UTC),
|
|
expectedTime: time.Date(2025, 1, 1, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "LeapYear",
|
|
actionTiming: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: &RITiming{
|
|
StartTime: "00:00:00",
|
|
Years: []int{2024},
|
|
Months: []time.Month{2},
|
|
MonthDays: []int{29},
|
|
},
|
|
},
|
|
},
|
|
startTime: time.Date(2024, 2, 28, 0, 0, 0, 0, time.UTC),
|
|
expectedTime: time.Date(2024, 2, 29, 0, 0, 0, 0, time.UTC),
|
|
},
|
|
{
|
|
name: "NilTiming",
|
|
actionTiming: &ActionTiming{
|
|
Timing: nil,
|
|
},
|
|
startTime: time.Date(2024, 1, 1, 0, 0, 0, 0, time.UTC),
|
|
expectedTime: time.Time{},
|
|
expectedError: true,
|
|
},
|
|
{
|
|
name: "EmptyTiming",
|
|
actionTiming: &ActionTiming{
|
|
Timing: &RateInterval{
|
|
Timing: nil,
|
|
},
|
|
},
|
|
startTime: time.Date(2024, 1, 1, 0, 0, 0, 0, time.UTC),
|
|
expectedTime: time.Time{},
|
|
expectedError: true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
if tt.actionTiming != nil {
|
|
tt.actionTiming.ResetStartTimeCache()
|
|
}
|
|
|
|
result := tt.actionTiming.GetNextStartTime(tt.startTime)
|
|
|
|
if tt.expectedError {
|
|
if !result.IsZero() {
|
|
t.Errorf("Expected zero time for error case, got: %v", result)
|
|
}
|
|
return
|
|
}
|
|
if !result.Equal(tt.expectedTime) {
|
|
t.Errorf("GetNextStartTime(%v) = %v; want %v",
|
|
tt.startTime, result, tt.expectedTime)
|
|
}
|
|
cachedResult := tt.actionTiming.GetNextStartTime(tt.startTime)
|
|
if !cachedResult.Equal(result) {
|
|
t.Errorf("Cached result differs: got %v, want %v",
|
|
cachedResult, result)
|
|
}
|
|
})
|
|
}
|
|
}
|