cgrates/migrator/thresholds_test.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 migrator

import (
	"reflect"
	"testing"
	"time"

	"github.com/cgrates/cgrates/config"
	"github.com/cgrates/cgrates/engine"
	"github.com/cgrates/cgrates/utils"
)

func TestV2ActionTriggerAsThreshold(t *testing.T) {
	var filters []*engine.FilterRule
	v2ATR := &v2ActionTrigger{
		ID:             "test2",              // original csv tag
		UniqueID:       "testUUID",           // individual id
		ThresholdType:  "*min_event_counter", //*min_event_counter, *max_event_counter, *min_balance_counter, *max_balance_counter, *min_balance, *max_balance, *balance_expired
		ThresholdValue: 5.32,
		Recurrent:      false,           // reset excuted flag each run
		MinSleep:       5 * time.Second, // Minimum duration between two executions in case of recurrent triggers
		ExpirationDate: time.Now(),
		ActivationDate: time.Now(),
		Balance: &engine.BalanceFilter{
			ID: utils.StringPointer(utils.MetaMonetary),
			DestinationIDs: &utils.StringMap{
				"1002": true,
			},
			RatingSubject: utils.StringPointer("1001"),
			Categories: &utils.StringMap{
				utils.MetaVoice: true,
			},
			SharedGroups: &utils.StringMap{
				"SHG1": true,
			},
			TimingIDs: &utils.StringMap{
				"TIMINGID": true,
			},
		},
		Weight:            0,
		ActionsID:         "Action1",
		MinQueuedItems:    10, // Trigger actions only if this number is hit (stats only)
		Executed:          false,
		LastExecutionTime: time.Now(),
	}
	x, _ := engine.NewFilterRule(utils.MetaDestinations, "DestinationIDs", v2ATR.Balance.DestinationIDs.Slice())
	filters = append(filters, x)
	x, _ = engine.NewFilterRule(utils.MetaPrefix, "RatingSubject", []string{*v2ATR.Balance.RatingSubject})
	filters = append(filters, x)
	x, _ = engine.NewFilterRule(utils.MetaPrefix, "Categories", v2ATR.Balance.Categories.Slice())
	filters = append(filters, x)
	x, _ = engine.NewFilterRule(utils.MetaPrefix, "SharedGroups", v2ATR.Balance.SharedGroups.Slice())
	filters = append(filters, x)
	x, _ = engine.NewFilterRule(utils.MetaPrefix, "TimingIDs", v2ATR.Balance.TimingIDs.Slice())
	filters = append(filters, x)

	filter := &engine.Filter{
		Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
		ID:     *v2ATR.Balance.ID,
		Rules:  filters}

	thp := &engine.ThresholdProfile{
		ID:                 v2ATR.ID,
		FilterIDs:          make([]string, 0),
		Tenant:             config.CgrConfig().GeneralCfg().DefaultTenant,
		Blocker:            false,
		Weight:             v2ATR.Weight,
		ActivationInterval: &utils.ActivationInterval{ExpiryTime: v2ATR.ExpirationDate, ActivationTime: v2ATR.ActivationDate},
		MinSleep:           v2ATR.MinSleep,
	}
	th := &engine.Threshold{
		Tenant: config.CgrConfig().GeneralCfg().DefaultTenant,
		ID:     v2ATR.ID,
	}

	newthp, newth, fltr, err := v2ATR.AsThreshold()
	if err != nil {
		t.Error(err)
	}
	if !reflect.DeepEqual(thp, newthp) {
		t.Errorf("Expecting: %+v, received: %+v", utils.ToJSON(thp), utils.ToJSON(newthp))
	}
	if !reflect.DeepEqual(th, newth) {
		t.Errorf("Expecting: %+v, received: %+v", th, newth)
	}
	if !reflect.DeepEqual(filter, fltr) {
		t.Errorf("Expecting: %+v, received: %+v", filter, fltr)
	}
}
func TestV2toV3Threshold(t *testing.T) {

	v2T := v2Threshold{
		Tenant:             "test-tenant",
		ID:                 "test-id",
		FilterIDs:          []string{"filter1", "filter2"},
		ActivationInterval: &utils.ActivationInterval{},
		Recurrent:          true,
		MinHits:            5,
		MinSleep:           10 * time.Second,
		Blocker:            true,
		Weight:             2.5,
		ActionIDs:          []string{"action1", "action2"},
		Async:              true,
	}

	result := v2T.V2toV3Threshold()

	if result.Tenant != v2T.Tenant {
		t.Errorf("expected Tenant %v, got %v", v2T.Tenant, result.Tenant)
	}
	if result.ID != v2T.ID {
		t.Errorf("expected ID %v, got %v", v2T.ID, result.ID)
	}
	if result.FilterIDs[0] != v2T.FilterIDs[0] || result.FilterIDs[1] != v2T.FilterIDs[1] {
		t.Errorf("expected FilterIDs %v, got %v", v2T.FilterIDs, result.FilterIDs)
	}
	if result.ActivationInterval != v2T.ActivationInterval {
		t.Errorf("expected ActivationInterval %v, got %v", v2T.ActivationInterval, result.ActivationInterval)
	}
	if result.MinHits != v2T.MinHits {
		t.Errorf("expected MinHits %v, got %v", v2T.MinHits, result.MinHits)
	}
	if result.MinSleep != v2T.MinSleep {
		t.Errorf("expected MinSleep %v, got %v", v2T.MinSleep, result.MinSleep)
	}
	if result.Blocker != v2T.Blocker {
		t.Errorf("expected Blocker %v, got %v", v2T.Blocker, result.Blocker)
	}
	if result.Weight != v2T.Weight {
		t.Errorf("expected Weight %v, got %v", v2T.Weight, result.Weight)
	}
	if result.ActionIDs[0] != v2T.ActionIDs[0] || result.ActionIDs[1] != v2T.ActionIDs[1] {
		t.Errorf("expected ActionIDs %v, got %v", v2T.ActionIDs, result.ActionIDs)
	}
	if result.Async != v2T.Async {
		t.Errorf("expected Async %v, got %v", v2T.Async, result.Async)
	}
	if result.MaxHits != -1 {
		t.Errorf("expected MaxHits %v, got %v", -1, result.MaxHits)
	}

	v2T.Recurrent = false
	result = v2T.V2toV3Threshold()
	if result.MaxHits != 1 {
		t.Errorf("expected MaxHits %v, got %v", 1, result.MaxHits)
	}
}