/*
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
*/
package utils
import (
"reflect"
"testing"
"github.com/cgrates/birpc/context"
"github.com/mitchellh/mapstructure"
)
type RpcStruct struct{}
type Attr struct {
Name string
Surname string
Age float64
}
func (rpc *RpcStruct) Method1(ctx *context.Context, normal Attr, out *float64) error {
return nil
}
func (rpc *RpcStruct) Method2(ctx *context.Context, pointer *Attr, out *float64) 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 , received <%+v>", err)
}
}
func TestGetRpcParams(t *testing.T) {
testStruct := Attr{"", "", 0}
RegisterRpcParams("", &RpcStruct{})
if result, err := GetRpcParams("RpcStruct.Method1"); err != nil {
t.Errorf("Expected , received <%+v>", err)
} else if !reflect.DeepEqual(result.InParam, testStruct) {
t.Errorf("Expected <%+v>, received <%+v>", testStruct, result.InParam)
}
}