X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=pkg%2Fcontrol%2Fsdl_restSubsDb_test.go;fp=pkg%2Fcontrol%2Fsdl_restSubsDb_test.go;h=ac3c8e3a558036859524af030afc66658a8acd48;hb=268d715e3bceab8f7955d89945141efdb2c3b368;hp=0000000000000000000000000000000000000000;hpb=f0835e8ca1ca3e526132d317a9ae849103a0d6db;p=ric-plt%2Fsubmgr.git

diff --git a/pkg/control/sdl_restSubsDb_test.go b/pkg/control/sdl_restSubsDb_test.go
new file mode 100644
index 0000000..ac3c8e3
--- /dev/null
+++ b/pkg/control/sdl_restSubsDb_test.go
@@ -0,0 +1,486 @@
+/*
+   ==================================================================================
+  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 control
+
+import (
+	"encoding/json"
+	"fmt"
+	"reflect"
+	"strings"
+	"sync"
+	"testing"
+
+	"github.com/segmentio/ksuid"
+	"github.com/stretchr/testify/assert"
+)
+
+var sdlRestShouldReturnError bool = false
+
+const sdlRestTestErrorString string = "Test sdl REST returned error on purpose"
+
+type RestSubsDbMock struct {
+	restSubsDb             map[string]string // Store information as a string like real db does.
+	restSubscriptions      map[string]*RESTSubscription
+	lastAllocatedRestSubId string
+	restSubIdsInDb         []string
+	marshalLock            sync.Mutex
+}
+
+var restSubsDbMock *RestSubsDbMock
+
+func CreateRestSubsDbMock() *RestSubsDbMock {
+	fmt.Println("Test CreateRestSubsDbMock()")
+	restSubsDbMock = new(RestSubsDbMock)
+	restSubsDbMock.ResetTestSettings()
+	restSubsDbMock.lastAllocatedRestSubId = ""
+	return restSubsDbMock
+}
+
+func (m *RestSubsDbMock) ResetTestSettings() {
+	m.restSubsDb = make(map[string]string)
+	m.restSubscriptions = make(map[string]*RESTSubscription)
+}
+
+func (m *RestSubsDbMock) AllocNextRestSubId() string {
+	m.lastAllocatedRestSubId = ksuid.New().String()
+	return m.lastAllocatedRestSubId
+}
+
+func (m *RestSubsDbMock) GetLastAllocatedRestSubId() string {
+	return m.lastAllocatedRestSubId
+}
+
+func (m *RestSubsDbMock) AddRestSubIdsInDb(restSubId string) {
+	m.restSubIdsInDb = append(m.restSubIdsInDb, restSubId)
+}
+
+func (m *RestSubsDbMock) DeleteRestSubIdsFromDb(restSubId string) {
+	newrestSubIdsInDb := []string{}
+	for _, i := range m.restSubIdsInDb {
+		if i != restSubId {
+			newrestSubIdsInDb = append(newrestSubIdsInDb, i)
+		}
+	}
+	m.restSubIdsInDb = newrestSubIdsInDb
+}
+
+func (m *RestSubsDbMock) EmptyRestSubIdsFromDb() {
+	m.restSubIdsInDb = nil
+}
+
+func CreateRESTSubscription(t *testing.T) *RESTSubscription {
+	t.Log("TEST: Creating REST subscription")
+
+	restSubscription := &RESTSubscription{}
+	restSubscription.xAppRmrEndPoint = "localhost:13560"
+	restSubscription.Meid = "RAN_NAME_1"
+	restSubscription.SubReqOngoing = true
+	restSubscription.SubDelReqOngoing = false
+	restSubscription.xAppIdToE2Id = make(map[int64]int64)
+	restSubscription.Md5sum = "856e9546f6f7b65a13a86996f2d1686a"
+	return restSubscription
+}
+
+func PrintRESTSubscriptionData(t *testing.T, restSubs *RESTSubscription) {
+	t.Log("TEST: RESTSubscription data")
+	t.Logf("TEST: restSubs.xAppRmrEndPoint = %v", restSubs.xAppRmrEndPoint)
+	t.Logf("TEST: restSubs.Meid = %v", restSubs.Meid)
+	t.Logf("TEST: restSubs.InstanceIds = %v", restSubs.InstanceIds)
+	t.Logf("TEST: restSubs.xAppIdToE2Id = %v", restSubs.xAppIdToE2Id)
+	t.Logf("TEST: restSubs.SubReqOngoing = %v", restSubs.SubReqOngoing)
+	t.Logf("TEST: restSubs.SubDelReqOngoing = %v", restSubs.SubDelReqOngoing)
+}
+
+func TestWriteRESTSubscriptionToSdl(t *testing.T) {
+
+	// Write one subscription
+	restSubId := restSubsDbMock.AllocNextRestSubId()
+	restSubs := CreateRESTSubscription(t)
+	PrintRESTSubscriptionData(t, restSubs)
+	t.Logf("TEST: Writing subId = %v\n", restSubId)
+	err := mainCtrl.c.WriteRESTSubscriptionToSdl(restSubId, restSubs)
+	if err != nil {
+		t.Errorf("TEST: %s", err.Error())
+	}
+	restSubsDbMock.AddRestSubIdsInDb(restSubId)
+}
+
+func TestReadRESTSubscriptionFromSdl(t *testing.T) {
+
+	restSubId := restSubsDbMock.GetLastAllocatedRestSubId()
+	t.Logf("Reading restSubId = %v\n", restSubId)
+	restSubs, err := mainCtrl.c.ReadRESTSubscriptionFromSdl(restSubId)
+	if err != nil {
+		t.Errorf("TEST: %s", err.Error())
+		return
+	}
+	PrintRESTSubscriptionData(t, restSubs)
+	assert.Equal(t, restSubsDbMock.restSubscriptions[restSubId], restSubs)
+}
+
+func TestRemoveRESTSubscriptionFromSdl(t *testing.T) {
+
+	restSubId := restSubsDbMock.GetLastAllocatedRestSubId()
+	err := mainCtrl.c.RemoveRESTSubscriptionFromSdl(restSubId)
+	if err != nil {
+		t.Errorf("TEST: %s", err.Error())
+		return
+	}
+	delete(restSubsDbMock.restSubscriptions, restSubId)
+	t.Logf("TEST: REST subscription removed from db. subId = %v", restSubId)
+	restSubsDbMock.DeleteRestSubIdsFromDb(restSubId)
+}
+
+func TestReadNotExistingRESTSubscriptionFromSdl(t *testing.T) {
+
+	restSubId := ""
+	restSubs, err := mainCtrl.c.ReadRESTSubscriptionFromSdl(restSubId)
+	if err != nil {
+		t.Logf("TEST: REST subscription not found from db. restSubId = %v", restSubId)
+		return
+	}
+	t.Errorf("TEST: REST subscription read from db. %v", restSubs)
+	PrintRESTSubscriptionData(t, restSubs)
+}
+
+func TestReadNotExistingRESTSubscriptionFromSdl2(t *testing.T) {
+
+	restSubId := "NotExistingSubsId"
+	restSubs, err := mainCtrl.c.ReadRESTSubscriptionFromSdl(restSubId)
+	if err != nil {
+		t.Logf("TEST: REST subscription not found from db. restSubId = %v", restSubId)
+		return
+	}
+	t.Errorf("TEST: REST subscription read from db. %v", restSubs)
+	PrintRESTSubscriptionData(t, restSubs)
+}
+
+func TestRemoveNotExistingRESTSubscriptionFromSdl(t *testing.T) {
+
+	restSubId := ""
+	err := mainCtrl.c.RemoveRESTSubscriptionFromSdl(restSubId)
+	if err != nil {
+		t.Logf("TEST: %s", err.Error())
+		return
+	}
+	t.Logf("TEST: REST subscription removed from db. subId = %v", restSubId)
+}
+
+func TestWriteRESTSubscriptionsToSdl(t *testing.T) {
+
+	// Write 1st subscription
+	restSubId := restSubsDbMock.AllocNextRestSubId()
+	t.Logf("TEST: Writing restSubId = %v\n", restSubId)
+	restSubs := CreateRESTSubscription(t)
+	PrintRESTSubscriptionData(t, restSubs)
+	err := mainCtrl.c.WriteRESTSubscriptionToSdl(restSubId, restSubs)
+	if err != nil {
+		t.Errorf("TEST: %s", err.Error())
+		return
+	}
+	restSubsDbMock.AddRestSubIdsInDb(restSubId)
+	t.Logf("TEST: REST subscription written in db = %v", restSubs)
+
+	// Write 2nd subscription
+	restSubId = restSubsDbMock.AllocNextRestSubId()
+	t.Logf("TEST:Writing restSubId = %v\n", restSubId)
+	restSubs = CreateRESTSubscription(t)
+	PrintRESTSubscriptionData(t, restSubs)
+	err = mainCtrl.c.WriteRESTSubscriptionToSdl(restSubId, restSubs)
+	if err != nil {
+		t.Errorf("TEST: %s", err.Error())
+		return
+	}
+	restSubsDbMock.AddRestSubIdsInDb(restSubId)
+	t.Logf("TEST: REST subscription written in db = %v", restSubs)
+
+	// Write 3rd subscription
+	restSubId = restSubsDbMock.AllocNextRestSubId()
+	t.Logf("TEST: Writing restSubId = %v\n", restSubId)
+	restSubs = CreateRESTSubscription(t)
+	PrintRESTSubscriptionData(t, restSubs)
+	err = mainCtrl.c.WriteRESTSubscriptionToSdl(restSubId, restSubs)
+	if err != nil {
+		t.Errorf("TEST: %s", err.Error())
+		return
+	}
+	restSubsDbMock.AddRestSubIdsInDb(restSubId)
+	t.Logf("TEST: REST subscription written in db = %v", restSubs)
+}
+
+func TestReadRESTSubscriptionsFromSdl(t *testing.T) {
+
+	for _, restSubId := range restSubsDbMock.restSubIdsInDb {
+		restSubs, err := mainCtrl.c.ReadRESTSubscriptionFromSdl(restSubId)
+		if err != nil {
+			t.Errorf("TEST: %s", err.Error())
+			return
+		}
+		PrintRESTSubscriptionData(t, restSubs)
+	}
+}
+
+func TestReadAllRESTSubscriptionsFromSdl(t *testing.T) {
+
+	register, err := mainCtrl.c.ReadAllRESTSubscriptionsFromSdl()
+	if err != nil {
+		t.Errorf("TEST: %s", err.Error())
+		return
+	}
+
+	for _, restSubs := range register {
+		PrintRESTSubscriptionData(t, restSubs)
+	}
+
+	assert.Equal(t, len(register), 3)
+}
+
+func TestRemoveAllRESTSubscriptionsFromSdl(t *testing.T) {
+
+	err := mainCtrl.c.RemoveAllRESTSubscriptionsFromSdl()
+	if err != nil {
+		t.Errorf("TEST: %s", err.Error())
+		return
+	}
+	t.Log("TEST: All subscription removed from db")
+	restSubsDbMock.EmptyRestSubIdsFromDb()
+}
+
+func TestReadAllRESTSubscriptionsFromSdl2(t *testing.T) {
+
+	register, err := mainCtrl.c.ReadAllRESTSubscriptionsFromSdl()
+	if err != nil {
+		t.Errorf("TEST: %s", err.Error())
+		return
+	}
+	for _, restSubs := range restSubsDbMock.restSubscriptions {
+		PrintRESTSubscriptionData(t, restSubs)
+	}
+	assert.Equal(t, len(register), 0)
+}
+
+func TestWriteRESTSubscriptionToSdlFail(t *testing.T) {
+
+	// Try to write one subscription.
+	// Test db should return test error string
+	MakeNextSdlRestCallFail()
+	restsubId := restSubsDbMock.AllocNextRestSubId()
+	restSubs := CreateRESTSubscription(t)
+	PrintRESTSubscriptionData(t, restSubs)
+	t.Logf("TEST: Writing subId = %v\n", restsubId)
+	err := mainCtrl.c.WriteRESTSubscriptionToSdl(restsubId, restSubs)
+	if err != nil {
+		if !strings.Contains(fmt.Sprintf("%s", err), sdlRestTestErrorString) {
+			t.Errorf("TEST: %s", err.Error())
+		}
+	} else {
+		t.Errorf("TEST: This test case should return error")
+	}
+}
+
+func TestReadRESTSubscriptionFromSdlFail(t *testing.T) {
+
+	// Try to read one subscription.
+	// Test db should return test error string
+	MakeNextSdlRestCallFail()
+	restSubId := restSubsDbMock.GetLastAllocatedRestSubId()
+	t.Logf("Reading restSubId = %v\n", restSubId)
+	restSubs, err := mainCtrl.c.ReadRESTSubscriptionFromSdl(restSubId)
+	if err != nil {
+		if !strings.Contains(fmt.Sprintf("%s", err), sdlRestTestErrorString) {
+			t.Errorf("TEST: %s", err.Error())
+		}
+		return
+	} else {
+		t.Errorf("TEST: This test case should return error")
+	}
+	PrintRESTSubscriptionData(t, restSubs)
+}
+
+func TestRemoveRESTSubscriptionFromSdlFail(t *testing.T) {
+
+	// Try to remove one subscription.
+	// Test db should return test error string
+	MakeNextSdlRestCallFail()
+	restSubId := restSubsDbMock.GetLastAllocatedRestSubId()
+	err := mainCtrl.c.RemoveRESTSubscriptionFromSdl(restSubId)
+	if err != nil {
+		if !strings.Contains(fmt.Sprintf("%s", err), sdlRestTestErrorString) {
+			t.Errorf("TEST: %s", err.Error())
+		}
+		return
+	} else {
+		t.Errorf("TEST: This test case should return error")
+	}
+	t.Logf("TEST: subscription removed from db. subId = %v", restSubId)
+}
+
+func TestReadAllRESTSubscriptionsFromSdlFail(t *testing.T) {
+
+	// Try to read all subscriptions.
+	// Test db should return test error string
+	MakeNextSdlRestCallFail()
+	register, err := mainCtrl.c.ReadAllRESTSubscriptionsFromSdl()
+	if err != nil {
+		if !strings.Contains(fmt.Sprintf("%s", err), sdlRestTestErrorString) {
+			t.Errorf("TEST: %s", err.Error())
+		}
+		return
+	} else {
+		t.Errorf("TEST: This test case should return error")
+	}
+
+	for _, restSubs := range register {
+		PrintRESTSubscriptionData(t, restSubs)
+	}
+}
+
+func TestRemoveAllRESTSubscriptionsFromSdlFail(t *testing.T) {
+
+	// Try to remove all subscriptions.
+	// Test db should return test error string
+	MakeNextSdlRestCallFail()
+	err := mainCtrl.c.RemoveAllRESTSubscriptionsFromSdl()
+	if err != nil {
+		if !strings.Contains(fmt.Sprintf("%s", err), sdlRestTestErrorString) {
+			t.Errorf("TEST: %s", err.Error())
+		}
+		return
+	} else {
+		t.Errorf("TEST: This test case should return error")
+	}
+	t.Log("TEST: All subscription removed from db")
+}
+
+func (m *RestSubsDbMock) Set(pairs ...interface{}) error {
+	var key string
+	var val string
+
+	m.marshalLock.Lock()
+	defer m.marshalLock.Unlock()
+
+	if sdlRestShouldReturnError == true {
+		return GetSdlRestError()
+	}
+
+	for _, v := range pairs {
+		reflectType := reflect.TypeOf(v)
+		switch reflectType.Kind() {
+		case reflect.Slice:
+			val = fmt.Sprintf("%s", v.([]uint8))
+		default:
+			switch v.(type) {
+			case string:
+				key = v.(string)
+			default:
+				return fmt.Errorf("Set() error: Unexpected type\n")
+			}
+		}
+	}
+
+	if key != "" {
+		m.restSubsDb[key] = val
+		restSubId := key
+		restSubscriptionInfo := &RESTSubscriptionInfo{}
+		err := json.Unmarshal([]byte(val), restSubscriptionInfo)
+		if err != nil {
+			return fmt.Errorf("Set() json.unmarshal error: %s\n", err.Error())
+		}
+
+		restSubs := mainCtrl.c.CreateRESTSubscription(restSubscriptionInfo, &val)
+		m.restSubscriptions[restSubId] = restSubs
+	} else {
+		return fmt.Errorf("Set() error: key == ''\n")
+	}
+	return nil
+}
+
+func (m *RestSubsDbMock) Get(keys []string) (map[string]interface{}, error) {
+	retMap := make(map[string]interface{})
+	if len(keys) == 0 {
+		return nil, fmt.Errorf("Get() error: len(key) == 0\n")
+	}
+
+	if sdlRestShouldReturnError == true {
+		return nil, GetSdlRestError()
+	}
+
+	for _, key := range keys {
+		if key != "" {
+			retMap[key] = m.restSubsDb[key]
+		} else {
+			return nil, fmt.Errorf("Get() error: key == ''\n")
+		}
+	}
+	return retMap, nil
+}
+
+func (m *RestSubsDbMock) GetAll() ([]string, error) {
+
+	if sdlRestShouldReturnError == true {
+		return nil, GetSdlRestError()
+	}
+
+	keys := []string{}
+	for key, _ := range m.restSubsDb {
+		keys = append(keys, key)
+	}
+	return keys, nil
+}
+
+func (m *RestSubsDbMock) Remove(keys []string) error {
+	if len(keys) == 0 {
+		return fmt.Errorf("Remove() error: len(key) == 0\n")
+	}
+
+	if sdlRestShouldReturnError == true {
+		return GetSdlRestError()
+	}
+
+	restSubId := keys[0]
+	delete(m.restSubsDb, restSubId)
+	delete(m.restSubscriptions, restSubId)
+	return nil
+}
+
+func (m *RestSubsDbMock) RemoveAll() error {
+
+	for key := range m.restSubsDb {
+
+		restSubId := key
+		delete(m.restSubsDb, restSubId)
+		delete(m.restSubscriptions, restSubId)
+	}
+
+	if sdlRestShouldReturnError == true {
+		return GetSdlRestError()
+	}
+
+	return nil
+}
+
+func MakeNextSdlRestCallFail() {
+	sdlRestShouldReturnError = true
+}
+
+func GetSdlRestError() error {
+	sdlRestShouldReturnError = false
+	return fmt.Errorf(sdlRestTestErrorString)
+}