cgrates/sessionmanager/session_test.go

/*
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 sessionmanager

import (
	"testing"
	"time"

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

//"github.com/cgrates/cgrates/config"
//"testing"

var (
	newEventBody = `
"Event-Name":	"HEARTBEAT",
"Core-UUID":	"d5abc5b0-95c6-11e1-be05-43c90197c914",
"FreeSWITCH-Hostname":	"grace",
"FreeSWITCH-Switchname":	"grace",
"FreeSWITCH-IPv4":	"172.17.77.126",
"variable_sip_full_from":	"rif",
"variable_cgr_account":	"rif",
"variable_sip_full_to":	"0723045326",
"Caller-Dialplan":	"vdf",
"FreeSWITCH-IPv6":	"::1",
"Event-Date-Local":	"2012-05-04 14:38:23",
"Event-Date-GMT":	"Fri, 03 May 2012 11:38:23 GMT",
"Event-Date-Timestamp":	"1336131503218867",
"Event-Calling-File":	"switch_core.c",
"Event-Calling-Function":	"send_heartbeat",
"Event-Calling-Line-Number":	"68",
"Event-Sequence":	"4171",
"Event-Info":	"System Ready",
"Up-Time":	"0 years, 0 days, 2 hours, 43 minutes, 21 seconds, 349 milliseconds, 683 microseconds",
"Session-Count":	"0",
"Max-Sessions":	"1000",
"Session-Per-Sec":	"30",
"Session-Since-Startup":	"122",
"Idle-CPU":	"100.000000"
`
	conf_data = []byte(`
### Test data, not for production usage

[global]
default_reqtype=
`)
)

/* Missing parameter is not longer tested in NewSession. ToDo: expand this test for more util information
func TestSessionNilSession(t *testing.T) {
	var errCfg error
	cfg, errCfg = config.NewCGRConfigBytes(conf_data) // Needed here to avoid nil on cfg variable
	if errCfg != nil {
		t.Errorf("Cannot get configuration %v", errCfg)
	}
	newEvent := new(FSEvent).New("")
	sm := &FSSessionManager{}
	s := NewSession(newEvent, sm)
	if s != nil {
		t.Error("no account and it still created session.")
	}
}
*/

type MockConnector struct {
	refundCd *engine.CallDescriptor
}

func (mc *MockConnector) GetCost(*engine.CallDescriptor, *engine.CallCost) error  { return nil }
func (mc *MockConnector) Debit(*engine.CallDescriptor, *engine.CallCost) error    { return nil }
func (mc *MockConnector) MaxDebit(*engine.CallDescriptor, *engine.CallCost) error { return nil }
func (mc *MockConnector) RefundIncrements(cd *engine.CallDescriptor, reply *float64) error {
	mc.refundCd = cd
	return nil
}
func (mc *MockConnector) GetMaxSessionTime(*engine.CallDescriptor, *float64) error { return nil }
func (mc *MockConnector) GetDerivedChargers(*utils.AttrDerivedChargers, *utils.DerivedChargers) error {
	return nil
}
func (mc *MockConnector) GetDerivedMaxSessionTime(*engine.CDR, *float64) error    { return nil }
func (mc *MockConnector) GetSessionRuns(*engine.CDR, *[]*engine.SessionRun) error { return nil }
func (mc *MockConnector) ProcessCdr(*engine.CDR, *string) error                   { return nil }
func (mc *MockConnector) LogCallCost(*engine.CallCostLog, *string) error          { return nil }
func (mc *MockConnector) GetLCR(*engine.AttrGetLcr, *engine.LCRCost) error        { return nil }
func (mc *MockConnector) GetTimeout(int, *time.Duration) error                    { return nil }

func TestSessionRefund(t *testing.T) {
	mc := &MockConnector{}
	s := &Session{sessionManager: &FSSessionManager{rater: mc}}
	ts := &engine.TimeSpan{
		TimeStart: time.Date(2015, 6, 10, 14, 7, 0, 0, time.UTC),
		TimeEnd:   time.Date(2015, 6, 10, 14, 7, 30, 0, time.UTC),
	}
	// add increments
	for i := 0; i < 30; i++ {
		ts.AddIncrement(&engine.Increment{Duration: time.Second, Cost: 1.0})
	}

	cc := &engine.CallCost{Timespans: engine.TimeSpans{ts}}
	hangupTime := time.Date(2015, 6, 10, 14, 7, 20, 0, time.UTC)
	s.Refund(cc, hangupTime)
	if len(mc.refundCd.Increments) != 1 || mc.refundCd.Increments[0].GetCompressFactor() != 10 || len(cc.Timespans) != 1 || cc.Timespans[0].TimeEnd != hangupTime {
		t.Errorf("Error refunding: %+v, %+v", mc.refundCd.Increments, cc.Timespans[0])
	}
}

func TestSessionRefundAll(t *testing.T) {
	mc := &MockConnector{}
	s := &Session{sessionManager: &FSSessionManager{rater: mc}}
	ts := &engine.TimeSpan{
		TimeStart: time.Date(2015, 6, 10, 14, 7, 0, 0, time.UTC),
		TimeEnd:   time.Date(2015, 6, 10, 14, 7, 30, 0, time.UTC),
	}
	// add increments
	for i := 0; i < 30; i++ {
		ts.AddIncrement(&engine.Increment{Duration: time.Second, Cost: 1.0})
	}

	cc := &engine.CallCost{Timespans: engine.TimeSpans{ts}}
	hangupTime := time.Date(2015, 6, 10, 14, 7, 0, 0, time.UTC)
	s.Refund(cc, hangupTime)
	if len(mc.refundCd.Increments) != 1 || mc.refundCd.Increments[0].GetCompressFactor() != 30 || len(cc.Timespans) != 0 {
		t.Errorf("Error refunding: %+v, %+v", len(mc.refundCd.Increments), cc.Timespans)
	}
}

func TestSessionRefundManyAll(t *testing.T) {
	mc := &MockConnector{}
	s := &Session{sessionManager: &FSSessionManager{rater: mc}}
	ts1 := &engine.TimeSpan{
		TimeStart: time.Date(2015, 6, 10, 14, 7, 0, 0, time.UTC),
		TimeEnd:   time.Date(2015, 6, 10, 14, 7, 30, 0, time.UTC),
	}
	// add increments
	for i := 0; i < int(ts1.GetDuration().Seconds()); i++ {
		ts1.AddIncrement(&engine.Increment{Duration: time.Second, Cost: 1.0})
	}

	ts2 := &engine.TimeSpan{
		TimeStart: time.Date(2015, 6, 10, 14, 7, 30, 0, time.UTC),
		TimeEnd:   time.Date(2015, 6, 10, 14, 8, 0, 0, time.UTC),
	}
	// add increments
	for i := 0; i < int(ts2.GetDuration().Seconds()); i++ {
		ts2.AddIncrement(&engine.Increment{Duration: time.Second, Cost: 1.0})
	}

	cc := &engine.CallCost{Timespans: engine.TimeSpans{ts1, ts2}}
	hangupTime := time.Date(2015, 6, 10, 14, 07, 0, 0, time.UTC)
	s.Refund(cc, hangupTime)
	if len(mc.refundCd.Increments) != 1 || mc.refundCd.Increments[0].GetCompressFactor() != 60 || len(cc.Timespans) != 0 {
		t.Errorf("Error refunding: %+v, %+v", len(mc.refundCd.Increments), cc.Timespans)
	}
}