Files
cgrates/ers/ers_test.go
2019-12-03 06:44:36 -05:00

71 lines
2.4 KiB
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 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 ers
import (
"reflect"
"testing"
"time"
"github.com/cgrates/cgrates/config"
"github.com/cgrates/cgrates/engine"
"github.com/cgrates/cgrates/utils"
)
func TestERsNewERService(t *testing.T) {
cfg, _ := config.NewDefaultCGRConfig()
fltrS := &engine.FilterS{}
expected := &ERService{cfg: cfg,
filterS: fltrS,
rdrs: make(map[string]EventReader),
rdrPaths: make(map[string]string),
stopLsn: make(map[string]chan struct{}),
rdrEvents: make(chan *erEvent),
rdrErr: make(chan error),
stopChan: nil}
rcv := NewERService(cfg, fltrS, nil, nil)
if !reflect.DeepEqual(expected.cfg, rcv.cfg) {
t.Errorf("Expecting: <%+v>, received: <%+v>", expected.cfg, rcv.cfg)
} else if !reflect.DeepEqual(expected.filterS, rcv.filterS) {
t.Errorf("Expecting: <%+v>, received: <%+v>", expected.filterS, rcv.filterS)
}
}
func TestERsAddReader(t *testing.T) {
cfg, _ := config.NewDefaultCGRConfig()
fltrS := &engine.FilterS{}
erS := NewERService(cfg, fltrS, nil, nil)
reader := cfg.ERsCfg().Readers[0]
reader.Type = utils.MetaFileCSV
reader.ID = "file_reader"
reader.RunDelay = time.Duration(0)
cfg.ERsCfg().Readers = append(cfg.ERsCfg().Readers, reader)
if len(cfg.ERsCfg().Readers) != 2 {
t.Errorf("Expecting: <2>, received: <%+v>", len(cfg.ERsCfg().Readers))
}
if err := erS.addReader("file_reader", 1); err != nil {
t.Errorf("Expecting: <nil>, received: <%+v>", len(cfg.ERsCfg().Readers))
} else if len(erS.rdrs) != 1 {
t.Errorf("Expecting: <2>, received: <%+v>", len(erS.rdrs))
} else if !reflect.DeepEqual(erS.rdrs["file_reader"].Config(), reader) {
t.Errorf("Expecting: <%+v>, received: <%+v>", reader, erS.rdrs["file_reader"].Config())
}
}