db/model/proxy_test.go

/*
 * Copyright 2018-present Open Networking Foundation

 * 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 model

import (
	"github.com/opencord/voltha-go/protos/voltha"
	"testing"
	"github.com/opencord/voltha-go/common/log"
	"strconv"
	"reflect"
	"github.com/google/uuid"
	"encoding/hex"
	"encoding/json"
	"fmt"
)

type proxyTest struct {
	Root      *Root
	Backend   *Backend
	Proxy     *Proxy
	DbPrefix  string
	DbType    string
	DbHost    string
	DbPort    int
	DbTimeout int
}

var (
	pt = &proxyTest{
		DbPrefix: "service/voltha/data/core/0001",
		DbType:   "etcd",
		//DbHost:    "10.102.58.0",
		DbHost:    "localhost",
		DbPort:    2379,
		DbTimeout: 5,
	}
	devId          string
	targetDeviceId string
)

func init() {
	if _, err := log.SetLogger(log.CONSOLE, 0, log.Fields{"instanceId": "proxy_test"}); err != nil {
		log.With(log.Fields{"error": err}).Fatal("cannot setup logging")
	}
	defer log.CleanUp()

	pt.Backend = NewBackend(pt.DbType, pt.DbHost, pt.DbPort, pt.DbTimeout, pt.DbPrefix)

	msgClass := &voltha.Voltha{}
	root := NewRoot(msgClass, pt.Backend, nil)
	pt.Root = root.Load(msgClass)

	GetProfiling().Report()

	pt.Proxy = pt.Root.Node.GetProxy("/", false)
}

//func Test_Proxy_0_GetRootProxy(t *testing.T) {
//	pt.Backend = NewBackend(pt.DbType, pt.DbHost, pt.DbPort, pt.DbTimeout, pt.DbPrefix)
//
//	msgClass := &voltha.Voltha{}
//	root := NewRoot(msgClass, pt.Backend, nil)
//	pt.Root = root.Load(msgClass)
//
//	GetProfiling().Report()
//
//	pt.Proxy = pt.Root.Node.GetProxy("/", false)
//}

func Test_Proxy_1_GetDevices(t *testing.T) {
	devices := pt.Proxy.Get("/devices", 1, false, "")

	if len(devices.([]interface{})) == 0 {
		t.Error("there are no available devices to retrieve")
	} else {
		// Save the target device id for later tests
		targetDeviceId = devices.([]interface{})[0].(*voltha.Device).Id
		t.Logf("retrieved devices: %+v", devices)
	}
}

func Test_Proxy_2_GetDevice(t *testing.T) {
	basePath := "/devices/" + targetDeviceId
	device1 := pt.Proxy.Get(basePath+"/ports", 1, false, "")
	t.Logf("retrieved device with ports: %+v", device1)

	device2 := pt.Proxy.Get(basePath, 0, false, "")

	t.Logf("retrieved device: %+v", device2)
}

//func Test_Proxy_3_AddDevice(t *testing.T) {
//	//ports := []*voltha.Port{
//	//	{
//	//		PortNo:     123,
//	//		Label:      "test-port-0",
//	//		Type:       voltha.Port_PON_OLT,
//	//		AdminState: common.AdminState_ENABLED,
//	//		OperStatus: common.OperStatus_ACTIVE,
//	//		DeviceId:   "etcd_port-0-device-id",
//	//		Peers:      []*voltha.Port_PeerPort{},
//	//	},
//	//}
//	devIdBin, _ := uuid.New().MarshalBinary()
//	devId := hex.EncodeToString(devIdBin)[:12]
//
//	device := &voltha.Device{
//		Id:                  devId,
//		Type:                "simulated_olt",
//		//Root:                true,
//		//ParentId:            "",
//		//ParentPortNo:        0,
//		//Vendor:              "voltha-test",
//		//Model:               "latest-voltha-simulated-olt",
//		//HardwareVersion:     "1.0.0",
//		//FirmwareVersion:     "1.0.0",
//		//Images:              &voltha.Images{},
//		//SerialNumber:        "abcdef-123456",
//		//VendorId:            "DEADBEEF-INC",
//		//Adapter:             "simulated_olt",
//		//Vlan:                1234,
//		Address:             &voltha.Device_HostAndPort{HostAndPort: "1.2.3.4:5555"},
//		//ExtraArgs:           "",
//		//ProxyAddress:        &voltha.Device_ProxyAddress{},
//		AdminState:          voltha.AdminState_PREPROVISIONED,
//		//OperStatus:          common.OperStatus_ACTIVE,
//		//Reason:              "",
//		//ConnectStatus:       common.ConnectStatus_REACHABLE,
//		//Custom:              &any.Any{},
//		//Ports:               ports,
//		//Flows:               &openflow_13.Flows{},
//		//FlowGroups:          &openflow_13.FlowGroups{},
//		//PmConfigs:           &voltha.PmConfigs{},
//		//ImageDownloads:      []*voltha.ImageDownload{},
//	}
//
//	//if retrieved := pt.Proxy.Get("/devices/00019b09a90bbe17", 0, false, ""); retrieved == nil {
//	//	t.Error("Failed to get device")
//	//} else {
//	//	devIdBin, _ := uuid.New().MarshalBinary()
//	//	devId = "0001" + hex.EncodeToString(devIdBin)[:12]
//	//	newDevice := Clone(de\).(*voltha.Device)
//	//	newDevice.Id = devId
//
//		if added := pt.Proxy.Add("/devices", device, ""); added == nil {
//			t.Error("Failed to add device")
//		} else {
//			t.Logf("Added device : %+v", added)
//		}
//	//}
//
//}
func Test_Proxy_3_AddDevice(t *testing.T) {
	devIdBin, _ := uuid.New().MarshalBinary()
	devId = "0001" + hex.EncodeToString(devIdBin)[:12]

	device := &voltha.Device{
		Id:         devId,
		Type:       "simulated_olt",
		Address:    &voltha.Device_HostAndPort{HostAndPort: "1.2.3.4:5555"},
		AdminState: voltha.AdminState_PREPROVISIONED,
	}

	if added := pt.Proxy.Add("/devices", device, ""); added == nil {
		t.Error("Failed to add device")
	} else {
		t.Logf("Added device : %+v", added)
	}
}

func Test_Proxy_4_CheckAddedDevice(t *testing.T) {
	if d := pt.Proxy.Get("/devices/"+devId, 0, false, ""); !reflect.ValueOf(d).IsValid() {
		t.Error("Failed to find added device")
	} else {
		djson, _ := json.Marshal(d)

		t.Logf("Found device: count: %s", djson)
	}
}

func Test_Proxy_5_UpdateDevice(t *testing.T) {
	if retrieved := pt.Proxy.Get("/devices/"+targetDeviceId, 1, false, ""); retrieved == nil {
		t.Error("Failed to get device")
	} else {
		var fwVersion int
		if retrieved.(*voltha.Device).FirmwareVersion == "n/a" {
			fwVersion = 0
		} else {
			fwVersion, _ = strconv.Atoi(retrieved.(*voltha.Device).FirmwareVersion)
			fwVersion += 1
		}

		cloned := reflect.ValueOf(retrieved).Elem().Interface().(voltha.Device)
		cloned.FirmwareVersion = strconv.Itoa(fwVersion)
		t.Logf("Before update : %+v", cloned)

		if afterUpdate := pt.Proxy.Update("/devices/"+targetDeviceId, &cloned, false, ""); afterUpdate == nil {
			t.Error("Failed to update device")
		} else {
			t.Logf("Updated device : %+v", afterUpdate.(Revision).GetData())
		}
	}
}

func Test_Proxy_6_CheckUpdatedDevice(t *testing.T) {
	device := pt.Proxy.Get("/devices/"+targetDeviceId, 0, false, "")

	t.Logf("content of updated device: %+v", device)
}

func Test_Proxy_7_RemoveDevice(t *testing.T) {
	if removed := pt.Proxy.Remove("/devices/"+devId, ""); removed == nil {
		t.Error("Failed to remove device")
	} else {
		t.Logf("Removed device : %+v", removed)
	}
}

func Test_Proxy_8_CheckRemovedDevice(t *testing.T) {
	if d := pt.Proxy.Get("/devices/"+devId, 0, false, ""); reflect.ValueOf(d).IsValid() {
		djson, _ := json.Marshal(d)
		t.Errorf("Device was not removed - %s", djson)
	} else {
		t.Logf("Device was removed: %s", devId)
	}
}

// -----------------------------
// Callback tests
// -----------------------------

func firstCallback(args ...interface{}) interface{} {
	name := args[0]
	id := args[1]
	fmt.Printf("Running first callback - name: %s, id: %s\n", name, id)
	return nil
}
func secondCallback(args ...interface{}) interface{} {
	name := args[0].(map[string]string)
	id := args[1]
	fmt.Printf("Running second callback - name: %s, id: %f\n", name["name"], id)
	panic("Generating a panic in second callback")
	return nil
}
func thirdCallback(args ...interface{}) interface{} {
	name := args[0]
	id := args[1].(*voltha.Device)
	fmt.Printf("Running third callback - name: %+v, id: %s\n", name, id.Id)
	return nil
}

func Test_Proxy_Callbacks_1_Register(t *testing.T) {
	pt.Proxy.RegisterCallback(PRE_ADD, firstCallback, "abcde", "12345")

	m := make(map[string]string)
	m["name"] = "fghij"
	pt.Proxy.RegisterCallback(PRE_ADD, secondCallback, m, 1.2345)

	d := &voltha.Device{Id: "12345"}
	pt.Proxy.RegisterCallback(PRE_ADD, thirdCallback, "klmno", d)
}

func Test_Proxy_Callbacks_2_Invoke_WithNoInterruption(t *testing.T) {
	pt.Proxy.InvokeCallbacks(PRE_ADD, nil, true)
}
func Test_Proxy_Callbacks_3_Invoke_WithInterruption(t *testing.T) {
	pt.Proxy.InvokeCallbacks(PRE_ADD, nil, false)
}

func Test_Proxy_Callbacks_4_Unregister(t *testing.T) {
	pt.Proxy.UnregisterCallback(PRE_ADD, firstCallback)
	pt.Proxy.UnregisterCallback(PRE_ADD, secondCallback)
	pt.Proxy.UnregisterCallback(PRE_ADD, thirdCallback)
}