[VOL-2235] Mocks and interfaces for rw-core
This update consists of mocks that are used by the rw-core
during unit testing. It also includes interfaces used for unit
tests.
Change-Id: I20ca1455c358113c3aa897acc6355e0ddbc614b7
diff --git a/rw_core/mocks/adapter_olt.go b/rw_core/mocks/adapter_olt.go
new file mode 100644
index 0000000..89f6c20
--- /dev/null
+++ b/rw_core/mocks/adapter_olt.go
@@ -0,0 +1,221 @@
+/*
+ * Copyright 2019-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 mocks
+
+import (
+ "context"
+ "fmt"
+ "github.com/gogo/protobuf/proto"
+ "github.com/opencord/voltha-lib-go/v2/pkg/adapters/adapterif"
+ com "github.com/opencord/voltha-lib-go/v2/pkg/adapters/common"
+ "github.com/opencord/voltha-lib-go/v2/pkg/log"
+ ic "github.com/opencord/voltha-protos/v2/go/inter_container"
+ of "github.com/opencord/voltha-protos/v2/go/openflow_13"
+ "github.com/opencord/voltha-protos/v2/go/voltha"
+ "strings"
+)
+
+const (
+ numONUPerOLT = 4
+)
+
+type OLTAdapter struct {
+ Adapter
+}
+
+func NewOLTAdapter(cp adapterif.CoreProxy) *OLTAdapter {
+ a := &OLTAdapter{}
+ a.coreProxy = cp
+ return a
+}
+
+func (oltA *OLTAdapter) Adopt_device(device *voltha.Device) error {
+ go func() {
+ d := proto.Clone(device).(*voltha.Device)
+ d.Root = true
+ d.Vendor = "olt_adapter_mock"
+ d.Model = "go-mock"
+ d.SerialNumber = com.GetRandomSerialNumber()
+ d.MacAddress = strings.ToUpper(com.GetRandomMacAddress())
+ oltA.storeDevice(d)
+ if res := oltA.coreProxy.DeviceUpdate(context.TODO(), d); res != nil {
+ log.Fatalf("deviceUpdate-failed-%s", res)
+ }
+ nniPort := &voltha.Port{
+ PortNo: 2,
+ Label: fmt.Sprintf("nni-%d", 2),
+ Type: voltha.Port_ETHERNET_NNI,
+ OperStatus: voltha.OperStatus_ACTIVE,
+ }
+ var err error
+ if err = oltA.coreProxy.PortCreated(context.TODO(), d.Id, nniPort); err != nil {
+ log.Fatalf("PortCreated-failed-%s", err)
+ }
+
+ ponPort := &voltha.Port{
+ PortNo: 1,
+ Label: fmt.Sprintf("pon-%d", 1),
+ Type: voltha.Port_PON_OLT,
+ OperStatus: voltha.OperStatus_ACTIVE,
+ }
+ if err = oltA.coreProxy.PortCreated(context.TODO(), d.Id, ponPort); err != nil {
+ log.Fatalf("PortCreated-failed-%s", err)
+ }
+
+ d.ConnectStatus = voltha.ConnectStatus_REACHABLE
+ d.OperStatus = voltha.OperStatus_ACTIVE
+
+ if err = oltA.coreProxy.DeviceStateUpdate(context.TODO(), d.Id, d.ConnectStatus, d.OperStatus); err != nil {
+ log.Fatalf("Device-state-update-failed-%s", err)
+ }
+
+ //Get the latest device data from the Core
+ if d, err = oltA.coreProxy.GetDevice(context.TODO(), d.Id, d.Id); err != nil {
+ log.Fatalf("getting-device-failed-%s", err)
+ }
+
+ if err = oltA.updateDevice(d); err != nil {
+ log.Fatalf("saving-device-failed-%s", err)
+ }
+
+ // Register Child devices
+ initialUniPortNo := 100
+ for i := 0; i < numONUPerOLT; i++ {
+ go func(seqNo int) {
+ if _, err := oltA.coreProxy.ChildDeviceDetected(
+ context.TODO(),
+ d.Id,
+ 1,
+ "onu_adapter_mock",
+ initialUniPortNo+seqNo,
+ "onu_adapter_mock",
+ com.GetRandomSerialNumber(),
+ int64(seqNo)); err != nil {
+ log.Fatalf("failure-sending-child-device-%s", err)
+ }
+ }(i)
+ }
+ }()
+ return nil
+}
+
+func (oltA *OLTAdapter) Get_ofp_device_info(device *voltha.Device) (*ic.SwitchCapability, error) {
+ if d := oltA.getDevice(device.Id); d == nil {
+ log.Fatalf("device-not-found-%s", device.Id)
+ }
+ return &ic.SwitchCapability{
+ Desc: &of.OfpDesc{
+ HwDesc: "olt_adapter_mock",
+ SwDesc: "olt_adapter_mock",
+ SerialNum: "12345678",
+ },
+ SwitchFeatures: &of.OfpSwitchFeatures{
+ NBuffers: 256,
+ NTables: 2,
+ Capabilities: uint32(of.OfpCapabilities_OFPC_FLOW_STATS |
+ of.OfpCapabilities_OFPC_TABLE_STATS |
+ of.OfpCapabilities_OFPC_PORT_STATS |
+ of.OfpCapabilities_OFPC_GROUP_STATS),
+ },
+ }, nil
+}
+
+func (oltA *OLTAdapter) Get_ofp_port_info(device *voltha.Device, port_no int64) (*ic.PortCapability, error) {
+ if d := oltA.getDevice(device.Id); d == nil {
+ log.Fatalf("device-not-found-%s", device.Id)
+ }
+ capability := uint32(of.OfpPortFeatures_OFPPF_1GB_FD | of.OfpPortFeatures_OFPPF_FIBER)
+ return &ic.PortCapability{
+ Port: &voltha.LogicalPort{
+ OfpPort: &of.OfpPort{
+ HwAddr: macAddressToUint32Array("11:22:33:44:55:66"),
+ Config: 0,
+ State: uint32(of.OfpPortState_OFPPS_LIVE),
+ Curr: capability,
+ Advertised: capability,
+ Peer: capability,
+ CurrSpeed: uint32(of.OfpPortFeatures_OFPPF_1GB_FD),
+ MaxSpeed: uint32(of.OfpPortFeatures_OFPPF_1GB_FD),
+ },
+ DeviceId: device.Id,
+ DevicePortNo: uint32(port_no),
+ },
+ }, nil
+}
+
+func (oltA *OLTAdapter) GetNumONUPerOLT() int {
+ return numONUPerOLT
+}
+
+func (oltA *OLTAdapter) Disable_device(device *voltha.Device) error {
+ go func() {
+ if d := oltA.getDevice(device.Id); d == nil {
+ log.Fatalf("device-not-found-%s", device.Id)
+ }
+
+ cloned := proto.Clone(device).(*voltha.Device)
+ // Update the all ports state on that device to disable
+ if err := oltA.coreProxy.PortsStateUpdate(context.TODO(), cloned.Id, voltha.OperStatus_UNKNOWN); err != nil {
+ log.Fatalf("updating-ports-failed", log.Fields{"deviceId": device.Id, "error": err})
+ }
+
+ //Update the device state
+ cloned.ConnectStatus = voltha.ConnectStatus_UNREACHABLE
+ cloned.OperStatus = voltha.OperStatus_UNKNOWN
+
+ if err := oltA.coreProxy.DeviceStateUpdate(context.TODO(), cloned.Id, cloned.ConnectStatus, cloned.OperStatus); err != nil {
+ log.Fatalf("device-state-update-failed", log.Fields{"deviceId": device.Id, "error": err})
+ }
+
+ if err := oltA.updateDevice(cloned); err != nil {
+ log.Fatalf("saving-device-failed-%s", err)
+ }
+
+ // Tell the Core that all child devices have been disabled (by default it's an action already taken by the Core
+ if err := oltA.coreProxy.ChildDevicesLost(context.TODO(), cloned.Id); err != nil {
+ log.Fatalf("lost-notif-of-child-devices-failed", log.Fields{"deviceId": device.Id, "error": err})
+ }
+ }()
+ return nil
+}
+
+func (oltA *OLTAdapter) Reenable_device(device *voltha.Device) error {
+ go func() {
+ if d := oltA.getDevice(device.Id); d == nil {
+ log.Fatalf("device-not-found-%s", device.Id)
+ }
+
+ cloned := proto.Clone(device).(*voltha.Device)
+ // Update the all ports state on that device to enable
+ if err := oltA.coreProxy.PortsStateUpdate(context.TODO(), cloned.Id, voltha.OperStatus_ACTIVE); err != nil {
+ log.Fatalf("updating-ports-failed", log.Fields{"deviceId": device.Id, "error": err})
+ }
+
+ //Update the device state
+ cloned.ConnectStatus = voltha.ConnectStatus_REACHABLE
+ cloned.OperStatus = voltha.OperStatus_ACTIVE
+
+ if err := oltA.coreProxy.DeviceStateUpdate(context.TODO(), cloned.Id, cloned.ConnectStatus, cloned.OperStatus); err != nil {
+ log.Fatalf("device-state-update-failed", log.Fields{"deviceId": device.Id, "error": err})
+ }
+
+ // Tell the Core that all child devices have been enabled
+ if err := oltA.coreProxy.ChildDevicesDetected(context.TODO(), cloned.Id); err != nil {
+ log.Fatalf("detection-notif-of-child-devices-failed", log.Fields{"deviceId": device.Id, "error": err})
+ }
+ }()
+ return nil
+}