mirror of
https://github.com/cgrates/cgrates.git
synced 2026-02-11 18:16:24 +05:00
78 lines
2.2 KiB
Go
78 lines
2.2 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 utils
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/mitchellh/mapstructure"
|
|
)
|
|
|
|
type RpcStruct struct{}
|
|
|
|
type Attr struct {
|
|
Name string
|
|
Surname string
|
|
Age float64
|
|
}
|
|
|
|
func (rpc *RpcStruct) Method1(normal Attr, out *float64) error {
|
|
return nil
|
|
}
|
|
|
|
func (rpc *RpcStruct) Method2(pointer *Attr, out *float64) error {
|
|
return nil
|
|
}
|
|
|
|
func (rpc *RpcStruct) Call(string, any, any) error {
|
|
return nil
|
|
}
|
|
|
|
func TestRPCObjectPointer(t *testing.T) {
|
|
RegisterRpcParams("", &RpcStruct{})
|
|
if len(rpcParamsMap) != 2 {
|
|
t.Errorf("error registering rpc object: %v", rpcParamsMap)
|
|
}
|
|
x, found := rpcParamsMap["RpcStruct.Method1"]
|
|
if !found {
|
|
t.Errorf("error getting rpcobject: %v (%+v)", rpcParamsMap, x)
|
|
}
|
|
a := x.InParam
|
|
if err := mapstructure.Decode(map[string]any{"Name": "a", "Surname": "b", "Age": 10.2}, a); err != nil || a.(*Attr).Name != "a" || a.(*Attr).Surname != "b" || a.(*Attr).Age != 10.2 {
|
|
t.Errorf("error converting to struct: %+v (%v)", a, err)
|
|
}
|
|
}
|
|
|
|
func TestGetRpcParamsError(t *testing.T) {
|
|
_, err := GetRpcParams("exampleTest")
|
|
if err == nil || err.Error() != "NOT_FOUND" {
|
|
t.Errorf("Expected <NOT_FOUND>, received <%+v>", err)
|
|
}
|
|
}
|
|
|
|
func TestGetRpcParams(t *testing.T) {
|
|
testStruct := &Attr{"", "", 0}
|
|
RegisterRpcParams("", &RpcStruct{})
|
|
if result, err := GetRpcParams("RpcStruct.Method1"); err != nil {
|
|
t.Errorf("Expected <nil>, received <%+v>", err)
|
|
} else if !reflect.DeepEqual(result.InParam, testStruct) {
|
|
t.Errorf("Expected <%+v>, received <%+v>", testStruct, result.InParam)
|
|
}
|
|
}
|