/*
Rating system designed to be used in VoIP Carriers World
Copyright (C) 2012-2015 ITsysCOM

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU 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 General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>
*/

package engine

import (
	"bufio"
	"io"
	"reflect"
	"strings"
	"testing"

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

var timingsSample = `#Tag,Years,Months,MonthDays,WeekDays,Time
ALWAYS,*any,*any,*any,*any,00:00:00
DUMMY,INVALID;DATA
ASAP,*any,*any,*any,*any,*asap
`

var destsSample = `#Tag,Prefix
GERMANY,+49
DUMMY,INVALID;DATA
GERMANY_MOBILE,+4915
`
var ratesSample = `#Tag,DestinationRatesTag,TimingTag,Weight
RT_1CENT,0,1,1s,1s,0s
DUMMY,INVALID;DATA
RT_DATA_2c,0,0.002,10,10,0
`

var destRatesSample = `#Tag,DestinationsTag,RatesTag,MaxCost,MaxCostStrategy
DR_RETAIL,GERMANY,RT_1CENT,*up,0,0,
DUMMY,INVALID;DATA
DR_DATA_1,*any,RT_DATA_2c,*up,2,0,
_TNT_1211_01_V_ANY,CST_1246534_BRB02,C_TNT_1211_01_V_1246534_BRB02_ANY,*up,2,,
DR_1007_MAXCOST_DISC,DST_1007,RT_1CNT,*up,4,0.62,*disconnect
DR_1007_MAXCOST_FREE,DST_1007,RT_1CNT,*up,4,0.62,*free
`
var ratingPlansSample = `#Tag,DestinationRatesTag,TimingTag,Weight
RP_RETAIL,DR_RETAIL,ALWAYS,10
DUMMY,INVALID;DATA
`

var ratingProfilesSample = `#Tenant,TOR,Direction,Subject,ActivationTime,RatingPlanTag,FallbackSubject
*out,cgrates.org,call,*any,2012-01-01T00:00:00Z,RP_RETAIL,,
DUMMY,INVALID;DATA
*out,cgrates.org,call,subj1;alias1,2012-01-01T00:00:00Z,RP_RETAIL,,
`

var actionsSample = `#ActionsTag[0],Action[1],ExtraParameters[2],BalanceTag[3],BalanceType[4],Direction[5],Category[6],DestinationTag[7],RatingSubject[8],SharedGroup[9],ExpiryTime[10],TimingTags[11],Units[12],BalanceWeight[13],Weight[14]
PREPAID_10,*topup_reset,,,*monetary,*out,,*any,,,*unlimited,ALWAYS;ASAP,5,10,10
WARN_HTTP,*call_url,http://localhost:8000,,,,,,,,,,,,10
LOG_BALANCE,*log,,,,,,,,,,,,,10
DUMMY,INVALID;DATA
PREPAID_10,*topup_reset,,,*monetary,*out,,*any,,*unlimited,,5,10,10
TOPUP_RST_SHARED_5,*topup_reset,param&some,,*monetary,*out,,*any,subj,SHARED_A,*unlimited,,5,20,10
DEFEE,*cdrlog,"{""Category"":""^ddi"",""MediationRunId"":""^did_run""}",,,,,,,,,,,,10
`

var actionTimingsSample = `#Tag,ActionsTag,TimingTag,Weight
PREPAID_10,PREPAID_10,ASAP,10
DUMMY,INVALID;DATA
`

var actionTriggersSample = `#Tag[0],ThresholdType[1],ThresholdValue[2],Recurrent[3],MinSleep[4],BalanceTag[5],BalanceType[6],BalanceDirection[7],BalanceCategory[8],BalanceDestinationTag[9],BalanceRatingSubject[10],BalanceSharedGroup[11],BalanceExpiryTime[12],BalanceTimingTags[13],BalanceWeight[14],StatsMinQueuedItems[15],ActionsTag[16],Weight[17]
STANDARD_TRIGGERS,,*min_balance,2,false,0,,*monetary,*out,,,,,,ALWAYS;ASAP,,,LOG_WARNING,10
STANDARD_TRIGGERS,,*max_balance,20,false,0,,*monetary,*out,,,,,,,,,LOG_WARNING,10
STANDARD_TRIGGERS,,*max_counter,15,false,0,,*monetary,*out,,FS_USERS,,,,,,,LOG_WARNING,10
CDRST1_WARN_ASR,,*min_asr,45,true,1h,,,,,,,,,,,3,CDRST_WARN_HTTP,10
DUMMY,INVALID;DATA
`

var sharedGroupsSample = `#Id,Account,Strategy,RatingSubject
SHARED_A,*any,*lowest_first,
DUMMY,INVALID;DATA
`

var accountActionsSample = `#Tenant,Account,Direction,ActionTimingsTag,ActionTriggersTag
cgrates.org,1001,*out,PREPAID_10,STANDARD_TRIGGERS
DUMMY,INVALID;DATA
cgrates.org,1002;1006,*out,PACKAGE_10,STANDARD_TRIGGERS
`
var derivedChargesSample = `#Direction,Tenant,Tor,Account,Subject,RunId,ReqTypeField,DirectionField,TenantField,TorField,AccountField,SubjectField,DestinationField,SetupTimeField,AnswerTimeField,DurationField
*out,cgrates.org,call,dan,dan,extra1,^filteredHeader1/filterValue1,^prepaid,,,,rif,rif,,,,,*default,*default
*out,cgrates.org,,dan,dan,extra1,^filteredHeader1/filterValue1,^prepaid,,,,rif,rif,,,,,
*in,cgrates.org,call,dan,dan,extra1,^filteredHeader1/filterValue1,^prepaid,,,,rif,rif,,,,,
DUMMY_DATA
*out,cgrates.org,call,dan,dan,extra2,,,,,,ivo,ivo,,,,,,
*out,cgrates.org,call,dan,*any,extra1,,,,,,rif2,rif2,,,,,,
*out,cgrates.org,call,dan,*any,*any,,,,,,rif2,rif2,,,,
*out,cgrates.org,call,dan,*any,*default,,*default,*default,*default,*default,rif2,rif2,*default,*default,*default,*default,*default
*out,cgrates.org,call,dan,*any,test,^test,^test,^test,^test,^test,^test,^test,^test,^test,^test,^test,^test,^test
*out,cgrates.org,call,dan,*any,run1,,,,,,,,,,,,,
*out,cgrates.org,call,dan,*default,,,,,,,,,,,,,
*out,cgrates.org,call,dan,dan,extra3,~filterhdr1:s/(.+)/special_run3/,,,,,^runusr3,^runusr3,,,,,,
*out,cgrates.org,call,1001,1001,derived_run1,,^*rated,*default,*default,*default,*default,^1002,*default,*default,*default,*default,*default,*default
`

func TestTimingsValidator(t *testing.T) {
	reader := bufio.NewReader(strings.NewReader(timingsSample))
	lnValidator := FileValidators[utils.TIMINGS_CSV]
	lineNr := 0
	for {
		lineNr++
		ln, _, err := reader.ReadLine()
		if err == io.EOF { // Reached end of the string
			break
		}
		valid := lnValidator.Rule.Match(ln)
		switch lineNr {
		case 1, 3:
			if valid {
				t.Error("Validation passed for invalid line", ln)
			}
		case 2, 4:
			if !valid {
				t.Error("Validation did not pass for valid line", ln)
			}
		}
	}
}

func TestDestinationsValidator(t *testing.T) {
	reader := bufio.NewReader(strings.NewReader(destsSample))
	lnValidator := FileValidators[utils.DESTINATIONS_CSV]
	lineNr := 0
	for {
		lineNr++
		ln, _, err := reader.ReadLine()
		if err == io.EOF { // Reached end of the string
			break
		}
		valid := lnValidator.Rule.Match(ln)
		switch lineNr {
		case 1, 3:
			if valid {
				t.Error("Validation passed for invalid line", string(ln))
			}
		case 2, 4:
			if !valid {
				t.Error("Validation did not pass for valid line", string(ln))
			}
		}
	}
}

func TestRatesValidator(t *testing.T) {
	reader := bufio.NewReader(strings.NewReader(ratesSample))
	lnValidator := FileValidators[utils.RATES_CSV]
	lineNr := 0
	for {
		lineNr++
		ln, _, err := reader.ReadLine()
		if err == io.EOF { // Reached end of the string
			break
		}
		valid := lnValidator.Rule.Match(ln)
		switch lineNr {
		case 1, 3:
			if valid {
				t.Error("Validation passed for invalid line", string(ln))
			}
		case 2, 4:
			if !valid {
				t.Error("Validation did not pass for valid line", string(ln))
			}
		}
	}
}

func TestDestRatesValidator(t *testing.T) {
	reader := bufio.NewReader(strings.NewReader(destRatesSample))
	lnValidator := FileValidators[utils.DESTINATION_RATES_CSV]
	lineNr := 0
	for {
		lineNr++
		ln, _, err := reader.ReadLine()
		if err == io.EOF { // Reached end of the string
			break
		}
		valid := lnValidator.Rule.Match(ln)
		switch lineNr {
		case 1, 3:
			if valid {
				t.Error("Validation passed for invalid line", string(ln))
			}
		case 2, 4, 5, 6, 7:
			if !valid {
				t.Error("Validation did not pass for valid line", string(ln))
			}
		}
	}
}

func TestRatingPlansValidator(t *testing.T) {
	reader := bufio.NewReader(strings.NewReader(ratingPlansSample))
	lnValidator := FileValidators[utils.RATING_PLANS_CSV]
	lineNr := 0
	for {
		lineNr++
		ln, _, err := reader.ReadLine()
		if err == io.EOF { // Reached end of the string
			break
		}
		valid := lnValidator.Rule.Match(ln)
		switch lineNr {
		case 1, 3:
			if valid {
				t.Error("Validation passed for invalid line", string(ln))
			}
		case 2:
			if !valid {
				t.Error("Validation did not pass for valid line", string(ln))
			}
		}
	}
}

func TestRatingProfilesValidator(t *testing.T) {
	reader := bufio.NewReader(strings.NewReader(ratingProfilesSample))
	lnValidator := FileValidators[utils.RATING_PROFILES_CSV]
	lineNr := 0
	for {
		lineNr++
		ln, _, err := reader.ReadLine()
		if err == io.EOF { // Reached end of the string
			break
		}
		valid := lnValidator.Rule.Match(ln)
		switch lineNr {
		case 1, 3:
			if valid {
				t.Error("Validation passed for invalid line", string(ln))
			}
		case 2, 4:
			if !valid {
				t.Error("Validation did not pass for valid line", string(ln))
			}
		}
	}
}

func TestActionsValidator(t *testing.T) {
	reader := bufio.NewReader(strings.NewReader(actionsSample))
	lnValidator := FileValidators[utils.ACTIONS_CSV]
	lineNr := 0
	for {
		lineNr++
		ln, _, err := reader.ReadLine()
		if err == io.EOF { // Reached end of the string
			break
		}
		valid := lnValidator.Rule.Match(ln)
		switch lineNr {
		case 1, 5, 6:
			if valid {
				t.Error("Validation passed for invalid line", string(ln))
			}
		case 2, 3, 4, 7, 8:
			if !valid {
				t.Error("Validation did not pass for valid line", string(ln))
			}
		}
	}
}

func TestActionTimingsValidator(t *testing.T) {
	reader := bufio.NewReader(strings.NewReader(actionTimingsSample))
	lnValidator := FileValidators[utils.ACTION_PLANS_CSV]
	lineNr := 0
	for {
		lineNr++
		ln, _, err := reader.ReadLine()
		if err == io.EOF { // Reached end of the string
			break
		}
		valid := lnValidator.Rule.Match(ln)
		switch lineNr {
		case 1, 3:
			if valid {
				t.Error("Validation passed for invalid line", string(ln))
			}
		case 2:
			if !valid {
				t.Error("Validation did not pass for valid line", string(ln))
			}
		}
	}
}

func TestActionTriggersValidator(t *testing.T) {
	reader := bufio.NewReader(strings.NewReader(actionTriggersSample))
	lnValidator := FileValidators[utils.ACTION_TRIGGERS_CSV]
	lineNr := 0
	for {
		lineNr++
		ln, _, err := reader.ReadLine()
		if err == io.EOF { // Reached end of the string
			break
		}
		valid := lnValidator.Rule.Match(ln)
		switch lineNr {
		case 1, 6:
			if valid {
				t.Error("Validation passed for invalid line", string(ln))
			}
		case 2, 3, 4, 5:
			if !valid {
				t.Error("Validation did not pass for valid line", string(ln))
			}
		}
	}
}

func TestSharedGroupsValidator(t *testing.T) {
	reader := bufio.NewReader(strings.NewReader(sharedGroupsSample))
	lnValidator := FileValidators[utils.SHARED_GROUPS_CSV]
	lineNr := 0
	for {
		lineNr++
		ln, _, err := reader.ReadLine()
		if err == io.EOF { // Reached end of the string
			break
		}
		valid := lnValidator.Rule.Match(ln)
		switch lineNr {
		case 1, 3:
			if valid {
				t.Error("Validation passed for invalid line", string(ln))
			}
		case 2:
			if !valid {
				t.Error("Validation did not pass for valid line", string(ln))
			}
		}
	}
}

func TestAccountActionsValidator(t *testing.T) {
	reader := bufio.NewReader(strings.NewReader(accountActionsSample))
	lnValidator := FileValidators[utils.ACCOUNT_ACTIONS_CSV]
	lineNr := 0
	for {
		lineNr++
		ln, _, err := reader.ReadLine()
		if err == io.EOF { // Reached end of the string
			break
		}
		valid := lnValidator.Rule.Match(ln)
		switch lineNr {
		case 1, 3:
			if valid {
				t.Error("Validation passed for invalid line", string(ln))
			}
		case 2, 4:
			if !valid {
				t.Error("Validation did not pass for valid line", string(ln))
			}
		}
	}
}

func TestDerivedChargersValidator(t *testing.T) {
	reader := bufio.NewReader(strings.NewReader(derivedChargesSample))
	lnValidator := FileValidators[utils.DERIVED_CHARGERS_CSV]
	lineNr := 0
	for {
		lineNr++
		ln, _, err := reader.ReadLine()
		if err == io.EOF { // Reached end of the string
			break
		}
		valid := lnValidator.Rule.Match(ln)
		switch lineNr {
		case 1, 3, 4, 5, 8, 9, 12:
			if valid {
				t.Error("Validation passed for invalid line", string(ln))
			}
		case 2, 6, 7, 10, 11, 13, 14:
			if !valid {
				t.Error("Validation did not pass for valid line", string(ln))
			}
		}
	}
}

func TestTPCSVFileParser(t *testing.T) {
	bfRdr := bufio.NewReader(strings.NewReader(ratesSample))
	fParser := &TPCSVFileParser{FileValidators[utils.RATES_CSV], bfRdr}
	lineNr := 0
	for {
		lineNr++
		record, err := fParser.ParseNextLine()
		if err == io.EOF { // Reached end of the string
			break
		}
		switch lineNr {
		case 1:
			if err == nil || err.Error() != "Line starts with comment character." {
				t.Error("Failed to detect comment character")
			}
		case 2:
			if err != nil {
				t.Error(err)
			}
			if !reflect.DeepEqual(record, []string{"RT_1CENT", "0", "1", "1s", "1s", "0s"}) {
				t.Error("Unexpected record extracted", record)
			}
		case 3:
			if err == nil {
				t.Error("Expecting invalid line at row 3")
			}
		}
	}
}

func TestValueOrDefault(t *testing.T) {
	if res := ValueOrDefault("someval", "*any"); res != "someval" {
		t.Error("Unexpected value received", res)
	}
	if res := ValueOrDefault("", "*any"); res != "*any" {
		t.Error("Unexpected value received", res)
	}
}