X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=pkg%2Fteststub%2Frmrenv.go;fp=pkg%2Fteststub%2Frmrenv.go;h=5a57b5649469beddf3a2acd29c23196587e2f1a1;hb=cd78aeeae3212fa722be4ba9768cac711305badf;hp=0000000000000000000000000000000000000000;hpb=aada64566a3a77cf9a20a98f9ddd7cd6f37529ae;p=ric-plt%2Fsubmgr.git

diff --git a/pkg/teststub/rmrenv.go b/pkg/teststub/rmrenv.go
new file mode 100644
index 0000000..5a57b56
--- /dev/null
+++ b/pkg/teststub/rmrenv.go
@@ -0,0 +1,148 @@
+/*
+==================================================================================
+  Copyright (c) 2019 AT&T Intellectual Property.
+  Copyright (c) 2019 Nokia
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.
+==================================================================================
+*/
+
+package teststub
+
+import (
+	"gerrit.o-ran-sc.org/r/ric-plt/submgr/pkg/xapptweaks"
+	"gerrit.o-ran-sc.org/r/ric-plt/xapp-frame/pkg/xapp"
+	"os"
+	"strconv"
+)
+
+//-----------------------------------------------------------------------------
+//
+//-----------------------------------------------------------------------------
+
+type RmrRouteTable struct {
+	tmpfile string
+	routes  []string
+	meids   []string
+}
+
+func (rrt *RmrRouteTable) AddRoute(mtype int, src string, subid int, trg string) {
+
+	line := "mse|"
+	line += strconv.FormatInt(int64(mtype), 10)
+	if len(src) > 0 {
+		line += "," + src
+	}
+	line += "|"
+	line += strconv.FormatInt(int64(subid), 10)
+	line += "|"
+	line += trg
+	rrt.routes = append(rrt.routes, line)
+}
+
+func (rrt *RmrRouteTable) AddMeid(trg string, meids []string) {
+
+	line := "mme_ar"
+	line += "|"
+	line += trg
+	line += "|"
+	for _, str := range meids {
+		line += " " + str
+	}
+	rrt.meids = append(rrt.meids, line)
+}
+
+func (rrt *RmrRouteTable) DelMeid(meids []string) {
+
+	line := "mme_del"
+	line += "|"
+	for _, str := range meids {
+		line += " " + str
+	}
+	rrt.meids = append(rrt.meids, line)
+}
+
+func (rrt *RmrRouteTable) FileName() string {
+	return rrt.tmpfile
+}
+
+func (rrt *RmrRouteTable) Table() string {
+	allrt := "newrt|start\n"
+	for _, val := range rrt.routes {
+		allrt += val + "\n"
+	}
+	allrt += "newrt|end\n"
+	allrt += "meid_map | start\n"
+	for _, val := range rrt.meids {
+		allrt += val + "\n"
+	}
+	allrt += "meid_map | end | " + strconv.FormatInt(int64(len(rrt.meids)), 10) + "\n"
+	return allrt
+}
+
+func (rrt *RmrRouteTable) Enable() {
+	if len(rrt.tmpfile) > 0 {
+		os.Remove(rrt.tmpfile)
+	}
+	rrt.tmpfile, _ = CreateTmpFile(rrt.Table())
+	os.Setenv("RMR_SEED_RT", rrt.tmpfile)
+	xapp.Logger.Info("Using rt file %s", os.Getenv("RMR_SEED_RT"))
+}
+
+func (rrt *RmrRouteTable) Disable() {
+	if len(rrt.tmpfile) > 0 {
+		os.Remove(rrt.tmpfile)
+		os.Unsetenv("RMR_SEED_RT")
+		rrt.tmpfile = ""
+		xapp.Logger.Info("Not using rt file ")
+	}
+}
+
+//-----------------------------------------------------------------------------
+//
+//-----------------------------------------------------------------------------
+
+type RmrSrcId struct {
+	xapptweaks.RmrEndpoint
+}
+
+func (rsi *RmrSrcId) Enable() {
+	if rsi.Port > 0 {
+		os.Setenv("RMR_SRC_ID", rsi.String())
+		xapp.Logger.Info("Using src id  %s", os.Getenv("RMR_SRC_ID"))
+	}
+}
+
+func (rsi *RmrSrcId) Disable() {
+	os.Unsetenv("RMR_SRC_ID")
+	xapp.Logger.Info("Not using Using src id")
+}
+
+//-----------------------------------------------------------------------------
+//
+//-----------------------------------------------------------------------------
+type RmrRtgSvc struct {
+	xapptweaks.RmrEndpoint
+}
+
+func (rrs *RmrRtgSvc) Enable() {
+	if rrs.Port > 0 {
+		os.Setenv("RMR_RTG_SVC", rrs.String())
+		xapp.Logger.Info("Using rtg svc  %s", os.Getenv("RMR_SRC_ID"))
+	}
+}
+
+func (rrs *RmrRtgSvc) Disable() {
+	os.Unsetenv("RMR_RTG_SVC")
+	xapp.Logger.Info("Not using Using rtg svc")
+}