Files
cgrates/sessionmanager/session_test.go
Radu Ioan Fericean 0c521b180c working
2012-05-03 16:45:58 +03:00

77 lines
2.4 KiB
Go

/*
Rating system designed to be used in VoIP Carriers World
Copyright (C) 2012 Radu Ioan Fericean
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"
)
func TestSessionDurationSingle(t *testing.T) {
s := &Session{}
s.AddCallToSession("", time.Now())
twoSeconds, _ := time.ParseDuration("2s")
if d := s.GetSessionDurationFrom(time.Now().Add(twoSeconds)); d.Seconds() < 2 || d.Seconds() > 3 {
t.Errorf("Wrong duration %v", d)
}
}
func TestSessionDurationMultiple(t *testing.T) {
s := &Session{}
s.AddCallToSession("", time.Now())
s.AddCallToSession("", time.Now())
s.AddCallToSession("", time.Now())
twoSeconds, _ := time.ParseDuration("2s")
if d := s.GetSessionDurationFrom(time.Now().Add(twoSeconds)); d.Seconds() < 6 || d.Seconds() > 7 {
t.Errorf("Wrong duration %v", d)
}
}
func TestSessionCostSingle(t *testing.T) {
s := &Session{customer: "vdf", subject: "rif"}
s.AddCallToSession("0723", time.Now())
twoSeconds, _ := time.ParseDuration("60s")
if ccs, err := s.GetSessionCostFrom(time.Now().Add(twoSeconds)); err != nil {
t.Errorf("Get cost returned error %v", err)
} else {
if len(ccs) != 1 || ccs[0].Cost < 1 || ccs[0].Cost > 1.1 {
t.Errorf("Expected %v got %v", "between 1 and 1.1", ccs[0].Cost)
}
}
}
func TestSessionCostMultiple(t *testing.T) {
s := &Session{customer: "vdf", subject: "rif"}
s.AddCallToSession("0723", time.Now())
s.AddCallToSession("0257", time.Now())
s.AddCallToSession("0256", time.Now())
twoSeconds, _ := time.ParseDuration("60s")
if ccs, err := s.GetSessionCostFrom(time.Now().Add(twoSeconds)); err != nil {
t.Errorf("Get cost returned error %v", err)
} else {
sum := 0.0
for _, cc := range ccs {
sum += cc.Cost
}
if len(ccs) != 3 || sum < 23 || sum > 23.1 {
t.Errorf("Expected %v got %v", "between 23 and 23.1", sum)
}
}
}