RMR updated to v3.6.0 with support for E2 Setup message types

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

diff --git a/pkg/sbi/sbi_test.go b/pkg/sbi/sbi_test.go
index ee8671f..7cb350c 100644 (file)
--- a/pkg/sbi/sbi_test.go
+++ b/pkg/sbi/sbi_test.go
@@ -30,6 +30,7 @@ package sbi
 import (
        "errors"
        "reflect"
+       "routing-manager/pkg/rtmgr"
        "testing"
 )
 
@@ -53,3 +54,35 @@ func TestGetSbi(t *testing.T) {
                }
        }
 }
+
+func TestUpdateE2TendPoint(t *testing.T) {
+       var err error
+       var sbi = Sbi{}
+       sbii, err := GetSbi("nngpush")
+       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"},
+       }
+
+       sbi.updateE2TEndpoints(&E2map, sbii)
+       t.Log(err)
+}
+
+func TestPruneEndpointList(t *testing.T) {
+       var sbi = Sbi{}
+       var err error
+       sbii, err := GetSbi("nngpush")
+       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: "10.20.30.40", Port: 1234, TxMessages: []string{"", ""}, RxMessages: []string{"", ""}, Socket: nil, IsReady: true, Keepalive: false}
+       rtmgr.Eps = EP
+
+       var nngpush = NngPush{}
+       nngpush.NewSocket = createNewStubPushSocket
+       nngpush.AddEndpoint(rtmgr.Eps["10.0.0.1:0"])
+
+       sbi.pruneEndpointList(sbii)
+       t.Log(err)
+}