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

diff --git a/pkg/teststub/controlRmrStub.go b/pkg/teststub/controlRmrStub.go
new file mode 100644
index 0000000..a8c861f
--- /dev/null
+++ b/pkg/teststub/controlRmrStub.go
@@ -0,0 +1,175 @@
+/*
+==================================================================================
+  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"
+	"strings"
+	"testing"
+	"time"
+)
+
+//-----------------------------------------------------------------------------
+//
+//-----------------------------------------------------------------------------
+type RmrStubControl struct {
+	RmrControl
+	xapptweaks.RmrWrapper
+	RecvChan chan *xapptweaks.RMRParams
+	Active   bool
+	InitMsg  int
+	CheckXid bool
+}
+
+func (tc *RmrStubControl) SetActive() {
+	tc.Active = true
+}
+
+func (tc *RmrStubControl) IsActive() bool {
+	return tc.Active
+}
+
+func (tc *RmrStubControl) SetCheckXid(val bool) {
+	tc.CheckXid = val
+}
+
+func (tc *RmrStubControl) IsCheckXid() bool {
+	return tc.CheckXid
+}
+
+func (tc *RmrStubControl) IsChanEmpty() bool {
+	if len(tc.RecvChan) > 0 {
+		return false
+	}
+	return true
+}
+
+func (tc *RmrStubControl) TestMsgChanEmpty(t *testing.T) {
+	if tc.IsChanEmpty() == false {
+		tc.TestError(t, "message channel not empty")
+	}
+}
+
+func (tc *RmrStubControl) Init(desc string, rtfile string, port string, stat string, initMsg int) {
+	tc.InitMsg = initMsg
+	tc.Active = false
+	tc.RecvChan = make(chan *xapptweaks.RMRParams)
+	tc.RmrControl.Init(desc, rtfile, port)
+	tc.RmrWrapper.Init()
+
+	tc.Rmr = xapp.NewRMRClientWithParams("tcp:"+port, 4096, 1, stat)
+	tc.Rmr.SetReadyCB(tc.ReadyCB, nil)
+	go tc.Rmr.Start(tc)
+
+	tc.WaitCB()
+	allRmrStubs = append(allRmrStubs, tc)
+}
+
+func (tc *RmrStubControl) Consume(params *xapp.RMRParams) (err error) {
+	defer tc.Rmr.Free(params.Mbuf)
+	msg := xapptweaks.NewParams(params)
+	tc.CntRecvMsg++
+
+	if msg.Mtype == tc.InitMsg {
+		tc.Logger.Info("Testing message ignore %s", msg.String())
+		tc.SetActive()
+		return
+	}
+
+	if tc.IsCheckXid() == true && strings.Contains(msg.Xid, tc.GetDesc()) == false {
+		tc.Logger.Info("Ignore %s", msg.String())
+		return
+	}
+
+	tc.Logger.Info("Consume %s", msg.String())
+	tc.PushMsg(msg)
+	return
+}
+
+func (tc *RmrStubControl) PushMsg(msg *xapptweaks.RMRParams) {
+	tc.Logger.Debug("RmrStubControl PushMsg ... msg(%d) waiting", msg.Mtype)
+	tc.RecvChan <- msg
+	tc.Logger.Debug("RmrStubControl PushMsg ... done")
+}
+
+func (tc *RmrStubControl) WaitMsg(secs time.Duration) *xapptweaks.RMRParams {
+	tc.Logger.Debug("RmrStubControl WaitMsg ... waiting")
+	if secs == 0 {
+		msg := <-tc.RecvChan
+		tc.Logger.Debug("RmrStubControl WaitMsg ... msg(%d) done", msg.Mtype)
+		return msg
+	}
+	select {
+	case msg := <-tc.RecvChan:
+		tc.Logger.Debug("RmrStubControl WaitMsg ... msg(%d) done", msg.Mtype)
+		return msg
+	case <-time.After(secs * time.Second):
+		tc.Logger.Debug("RmrStubControl WaitMsg ... timeout")
+		return nil
+	}
+	tc.Logger.Debug("RmrStubControl WaitMsg ... error")
+	return nil
+}
+
+var allRmrStubs []*RmrStubControl
+
+//-----------------------------------------------------------------------------
+//
+//-----------------------------------------------------------------------------
+
+func RmrStubControlWaitAlive(seconds int, mtype int, rmr xapptweaks.XAppWrapperIf) bool {
+
+	var dummyBuf []byte = make([]byte, 100)
+
+	params := xapptweaks.NewParams(nil)
+	params.Mtype = mtype
+	params.SubId = -1
+	params.Payload = dummyBuf
+	params.PayloadLen = 100
+	params.Meid = &xapp.RMRMeid{RanName: "TESTPING"}
+	params.Xid = "TESTPING"
+	params.Mbuf = nil
+
+	status := false
+	i := 1
+	for ; i <= seconds*2 && status == false; i++ {
+
+		rmr.GetLogger().Info("SEND TESTPING: %s", params.String())
+		rmr.RmrSend(params)
+
+		status = true
+		for _, val := range allRmrStubs {
+			if val.IsActive() == false {
+				status = false
+				break
+			}
+		}
+		if status == true {
+			break
+		}
+		rmr.GetLogger().Info("Sleep 0.5 secs and try routes again")
+		time.Sleep(500 * time.Millisecond)
+	}
+
+	if status == false {
+		rmr.GetLogger().Error("Could not initialize routes")
+	}
+	return status
+}