X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=pkg%2Fsbi%2Fsbi_test.go;h=aa12d60a82525d2547157eddb0aafb276ffa0e46;hb=765b273b4444673d4e45a5f363dea28cdb6ca0bf;hp=52436ed3a15e849b3b4e16a8cf9ae2edd8efe59c;hpb=427929cf789725d9362da7afd960ec01f1054112;p=ric-plt%2Frtmgr.git

diff --git a/pkg/sbi/sbi_test.go b/pkg/sbi/sbi_test.go
index 52436ed..aa12d60 100644
--- 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)
 }