RIC-851: Updated rtmgr to create xapp routes

[ric-plt/rtmgr.git] / pkg / sbi / sbi_test.go

diff --git a/pkg/sbi/sbi_test.go b/pkg/sbi/sbi_test.go
index 52436ed..aa12d60 100644 (file)
--- a/pkg/sbi/sbi_test.go
+++ b/pkg/sbi/sbi_test.go
@@ -30,21 +30,21 @@ package sbi
 import (
        "errors"
        "reflect"
-       "testing"
        "routing-manager/pkg/rtmgr"
+       "testing"
 )
 
 func TestGetSbi(t *testing.T) {
        var errtype = errors.New("")
-       var sbitype = new(NngPush)
-       var invalids = []string{"nngpus", ""}
+       var sbitype = new(RmrPush)
+       var invalids = []string{"rmrpus", ""}
 
-       sbii, err := GetSbi("nngpush")
+       sbii, err := GetSbi("rmrpush")
        if err != nil {
-               t.Errorf("GetSbi(nngpub) was incorrect, got: %v, want: %v.", reflect.TypeOf(err), nil)
+               t.Errorf("GetSbi(rmrpub) was incorrect, got: %v, want: %v.", reflect.TypeOf(err), nil)
        }
        if reflect.TypeOf(sbii) != reflect.TypeOf(sbitype) {
-               t.Errorf("GetSbi(nngpub) was incorrect, got: %v, want: %v.", reflect.TypeOf(sbii), reflect.TypeOf(sbitype))
+               t.Errorf("GetSbi(rmrpub) was incorrect, got: %v, want: %v.", reflect.TypeOf(sbii), reflect.TypeOf(sbitype))
        }
 
        for _, arg := range invalids {
@@ -58,27 +58,38 @@ func TestGetSbi(t *testing.T) {
 func TestUpdateE2TendPoint(t *testing.T) {
        var err error
        var sbi = Sbi{}
-       sbii, err := GetSbi("nngpush")
+       sbii, err := GetSbi("rmrpush")
+
+       var EP = make(map[string]*rtmgr.Endpoint)
+       EP["127.0.0.2"] = &rtmgr.Endpoint{Uuid: "127.0.0.2", Name: "E2TERM", XAppType: "app1", Ip: "127.0.0.2", Port: 4562, TxMessages: []string{"", ""}, RxMessages: []string{"", ""}, Socket: nil, IsReady: true, Keepalive: false}
+       rtmgr.Eps = EP
+
+       var rmrpush = RmrPush{}
+       rmrpush.AddEndpoint(rtmgr.Eps["127.0.0.2"])
+
        var E2map = make(map[string]rtmgr.E2TInstance)
 
-       E2map["1.2.3.4:100"] = rtmgr.E2TInstance{
-                Name: "E2Tinstance1",
-               Fqdn: "1.2.3.4:100",
-               Ranlist: []string{"1","2"},
+       E2map["127.0.0.2:4562"] = rtmgr.E2TInstance{
+               Name:    "E2Tinstance1",
+               Fqdn:    "127.0.0.2:4562",
+               Ranlist: []string{"1", "2"},
        }
 
-       sbi.updateE2TEndpoints(&E2map,sbii)
+       sbi.updateE2TEndpoints(&E2map, sbii)
        t.Log(err)
 }
 
-func TestpruneEndpointList(t *testing.T) {
+func TestPruneEndpointList(t *testing.T) {
        var sbi = Sbi{}
        var err error
-        sbii, err := GetSbi("nngpush")
-
+       sbii, err := GetSbi("rmrpush")
        var EP = make(map[string]*rtmgr.Endpoint)
-       EP["10.0.0.1:0"] = &rtmgr.Endpoint{Uuid: "10.0.0.1:0", Name: "E2TERM", XAppType: "app1", Ip: "", Port: 0, TxMessages: []string{"", ""}, RxMessages: []string{"", ""}, Socket: nil, IsReady: true, Keepalive: true}
+       EP["127.0.0.2"] = &rtmgr.Endpoint{Uuid: "127.0.0.2", Name: "E2TERM", XAppType: "app1", Ip: "127.0.0.1", Port: 4562, TxMessages: []string{"", ""}, RxMessages: []string{"", ""}, Socket: nil, IsReady: true, Keepalive: false}
        rtmgr.Eps = EP
+
+       var rmrpush = RmrPush{}
+       rmrpush.AddEndpoint(rtmgr.Eps["127.0.0.2"])
+
        sbi.pruneEndpointList(sbii)
        t.Log(err)
 }