mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
121 lines
3.7 KiB
Go
121 lines
3.7 KiB
Go
//go:build integration
|
|
// +build integration
|
|
|
|
/*
|
|
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 cores
|
|
|
|
import (
|
|
"net/rpc/jsonrpc"
|
|
"path"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/cgrates/birpc/context"
|
|
"github.com/cgrates/cgrates/config"
|
|
"github.com/cgrates/cgrates/engine"
|
|
"github.com/cgrates/cgrates/utils"
|
|
)
|
|
|
|
func TestCAPsStatusAllocated(t *testing.T) {
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
cfgPath := path.Join("/usr/share/cgrates", "conf", "samples", "caps_queue")
|
|
cfg, err := config.NewCGRConfigFromPath(ctx, cfgPath)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if _, err := engine.StopStartEngine(cfgPath, 100); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer engine.KillEngine(100)
|
|
|
|
client, err := jsonrpc.Dial(utils.TCP, cfg.ListenCfg().RPCJSONListen)
|
|
if err != nil {
|
|
t.Fatalf("could not establish connection to engine: %v", err)
|
|
}
|
|
|
|
cfgStr := `{"cores":{"caps":2,"caps_stats_interval":"0","caps_strategy":"*queue","ees_conns":[],"shutdown_timeout":"1s"}}`
|
|
|
|
var rpl string
|
|
if err := client.Call(utils.ConfigSv1GetConfigAsJSON, &config.SectionWithAPIOpts{
|
|
Tenant: "cgrates.org",
|
|
Sections: []string{config.CoreSJSON},
|
|
}, &rpl); err != nil {
|
|
t.Error(err)
|
|
} else if cfgStr != rpl {
|
|
t.Errorf("Expected %q , \nreceived: %q", cfgStr, rpl)
|
|
}
|
|
|
|
var reply map[string]any
|
|
if err := client.Call(utils.CoreSv1Status, utils.TenantWithAPIOpts{},
|
|
&reply); err != nil {
|
|
t.Fatal(err)
|
|
} else if reply[utils.NodeID] != "ConcurrentQueueEngine" {
|
|
t.Errorf("Expected %+v , received: %+v ", "ConcurrentQueueEngine", reply)
|
|
} else if _, has := reply[utils.MetricCapsAllocated]; !has {
|
|
t.Errorf("Expected reply to contain CAPSAllocated , received <%+v>", reply)
|
|
}
|
|
|
|
}
|
|
func TestCAPsStatusPeak(t *testing.T) {
|
|
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
cfgPath := path.Join("/usr/share/cgrates", "conf", "samples", "caps_peak")
|
|
cfg, err := config.NewCGRConfigFromPath(ctx, cfgPath)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if _, err := engine.StopStartEngine(cfgPath, 100); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
defer engine.KillEngine(100)
|
|
|
|
client, err := jsonrpc.Dial(utils.TCP, cfg.ListenCfg().RPCJSONListen)
|
|
if err != nil {
|
|
t.Fatalf("could not establish connection to engine: %v", err)
|
|
}
|
|
cfgStr := `{"cores":{"caps":2,"caps_stats_interval":"100ms","caps_strategy":"*queue","ees_conns":[],"shutdown_timeout":"1s"}}`
|
|
|
|
var rpl string
|
|
if err := client.Call(utils.ConfigSv1GetConfigAsJSON, &config.SectionWithAPIOpts{
|
|
Tenant: "cgrates.org",
|
|
Sections: []string{config.CoreSJSON},
|
|
}, &rpl); err != nil {
|
|
t.Error(err)
|
|
} else if cfgStr != rpl {
|
|
t.Errorf("Expected %q , \nreceived: %q", cfgStr, rpl)
|
|
}
|
|
|
|
var reply map[string]any
|
|
if err := client.Call(utils.CoreSv1Status, utils.TenantWithAPIOpts{},
|
|
&reply); err != nil {
|
|
t.Fatal(err)
|
|
} else if reply[utils.NodeID] != "CAPSPeakEngine" {
|
|
t.Errorf("Expected %+v , received: %+v ", "CAPSPeakEngine", reply)
|
|
} else if _, has := reply[utils.MetricCapsPeak]; !has {
|
|
t.Errorf("Expected reply to contain CAPSPeak , received <%+v>", reply)
|
|
}
|
|
|
|
}
|