/* 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 */ 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.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) } }) } }