Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 1 | # Copyright 2017-present Open Networking Foundation |
| 2 | # |
| 3 | # Licensed under the Apache License, Version 2.0 (the "License"); |
| 4 | # you may not use this file except in compliance with the License. |
| 5 | # You may obtain a copy of the License at |
| 6 | # |
| 7 | # http://www.apache.org/licenses/LICENSE-2.0 |
| 8 | # |
| 9 | # Unless required by applicable law or agreed to in writing, software |
| 10 | # distributed under the License is distributed on an "AS IS" BASIS, |
| 11 | # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| 12 | # See the License for the specific language governing permissions and |
| 13 | # limitations under the License. |
| 14 | |
Scott Baker | 22b46c5 | 2018-11-15 15:15:29 -0800 | [diff] [blame] | 15 | from requests import ConnectionError |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 16 | import unittest |
Matteo Scandolo | 2ed64b9 | 2018-06-18 10:32:56 -0700 | [diff] [blame] | 17 | import functools |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 18 | from mock import patch, call, Mock, PropertyMock |
| 19 | import requests_mock |
| 20 | |
| 21 | import os, sys |
| 22 | |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 23 | test_path=os.path.abspath(os.path.dirname(os.path.realpath(__file__))) |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 24 | |
Matteo Scandolo | 2ed64b9 | 2018-06-18 10:32:56 -0700 | [diff] [blame] | 25 | def match_json(desired, req): |
| 26 | if desired!=req.json(): |
| 27 | raise Exception("Got request %s, but body is not matching" % req.url) |
| 28 | return False |
| 29 | return True |
| 30 | |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 31 | class TestSyncOLTDevice(unittest.TestCase): |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 32 | def setUp(self): |
Matteo Scandolo | 6be6ee9 | 2018-05-24 15:07:51 -0700 | [diff] [blame] | 33 | global DeferredException |
Matteo Scandolo | ce27e9c | 2018-04-06 10:06:53 -0700 | [diff] [blame] | 34 | self.sys_path_save = sys.path |
Matteo Scandolo | ce27e9c | 2018-04-06 10:06:53 -0700 | [diff] [blame] | 35 | |
| 36 | # Setting up the config module |
| 37 | from xosconfig import Config |
Matteo Scandolo | f7ebb11 | 2018-09-18 16:17:22 -0700 | [diff] [blame] | 38 | config = os.path.join(test_path, "../test_config.yaml") |
Matteo Scandolo | ce27e9c | 2018-04-06 10:06:53 -0700 | [diff] [blame] | 39 | Config.clear() |
| 40 | Config.init(config, "synchronizer-config-schema.yaml") |
Luca Prete | ca974c8 | 2018-05-01 18:06:16 -0700 | [diff] [blame] | 41 | # END setting up the config module |
Matteo Scandolo | ce27e9c | 2018-04-06 10:06:53 -0700 | [diff] [blame] | 42 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 43 | from xossynchronizer.mock_modelaccessor_build import mock_modelaccessor_config |
| 44 | mock_modelaccessor_config(test_path, [("olt-service", "volt.xproto"), |
Matteo Scandolo | 35207b7 | 2019-05-10 08:46:48 -0700 | [diff] [blame] | 45 | ("rcord", "rcord.xproto")]) |
Matteo Scandolo | 19466a0 | 2018-05-16 17:43:39 -0700 | [diff] [blame] | 46 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 47 | import xossynchronizer.modelaccessor |
| 48 | reload(xossynchronizer.modelaccessor) # in case nose2 loaded it in a previous test |
Matteo Scandolo | 19466a0 | 2018-05-16 17:43:39 -0700 | [diff] [blame] | 49 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 50 | from xossynchronizer.modelaccessor import model_accessor |
| 51 | self.model_accessor = model_accessor |
| 52 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 53 | # import all class names to globals |
| 54 | for (k, v) in model_accessor.all_model_classes.items(): |
| 55 | globals()[k] = v |
| 56 | |
Matteo Scandolo | 6be6ee9 | 2018-05-24 15:07:51 -0700 | [diff] [blame] | 57 | from sync_olt_device import SyncOLTDevice, DeferredException |
Matteo Scandolo | ce27e9c | 2018-04-06 10:06:53 -0700 | [diff] [blame] | 58 | self.sync_step = SyncOLTDevice |
| 59 | |
Matteo Scandolo | b8621cd | 2018-04-04 17:12:37 -0700 | [diff] [blame] | 60 | pon_port = Mock() |
| 61 | pon_port.port_id = "00ff00" |
Matteo Scandolo | b8621cd | 2018-04-04 17:12:37 -0700 | [diff] [blame] | 62 | |
Matteo Scandolo | 0d756f2 | 2019-06-10 14:55:55 -0700 | [diff] [blame] | 63 | # create a mock ONOS Service |
| 64 | onos = Mock() |
| 65 | onos.name = "ONOS" |
| 66 | onos.leaf_model.rest_hostname = "onos" |
| 67 | onos.leaf_model.rest_port = 4321 |
| 68 | onos.leaf_model.rest_username = "karaf" |
| 69 | onos.leaf_model.rest_password = "karaf" |
| 70 | |
Matteo Scandolo | 2ed64b9 | 2018-06-18 10:32:56 -0700 | [diff] [blame] | 71 | # Create a mock OLTDevice |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 72 | o = Mock() |
| 73 | o.volt_service.voltha_url = "voltha_url" |
Luca Prete | ca974c8 | 2018-05-01 18:06:16 -0700 | [diff] [blame] | 74 | o.volt_service.voltha_port = 1234 |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 75 | o.volt_service.voltha_user = "voltha_user" |
| 76 | o.volt_service.voltha_pass = "voltha_pass" |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 77 | |
Matteo Scandolo | 0d756f2 | 2019-06-10 14:55:55 -0700 | [diff] [blame] | 78 | o.volt_service.provider_services = [onos] |
Matteo Scandolo | a79395f | 2018-10-08 13:34:49 -0700 | [diff] [blame] | 79 | |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 80 | o.device_type = "ponsim_olt" |
| 81 | o.host = "172.17.0.1" |
| 82 | o.port = "50060" |
Scott Baker | 0bbbfd1 | 2018-12-11 07:01:06 +0000 | [diff] [blame] | 83 | o.uplink = "129" |
Luca Prete | 244e6ec | 2018-07-02 14:30:24 +0200 | [diff] [blame] | 84 | o.driver = "voltha" |
Matteo Scandolo | a79395f | 2018-10-08 13:34:49 -0700 | [diff] [blame] | 85 | o.name = "Test Device" |
Scott Baker | 09798d8 | 2019-01-17 08:34:59 -0800 | [diff] [blame] | 86 | o.admin_state = "ENABLED" |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 87 | |
Matteo Scandolo | b8621cd | 2018-04-04 17:12:37 -0700 | [diff] [blame] | 88 | # feedback state |
| 89 | o.device_id = None |
Matteo Scandolo | b8621cd | 2018-04-04 17:12:37 -0700 | [diff] [blame] | 90 | o.oper_status = None |
Scott Baker | 3581f64 | 2019-06-26 14:24:20 -0700 | [diff] [blame] | 91 | o.serial_number= None |
Matteo Scandolo | b8621cd | 2018-04-04 17:12:37 -0700 | [diff] [blame] | 92 | o.of_id = None |
Matteo Scandolo | 096a3cf | 2018-06-20 13:56:13 -0700 | [diff] [blame] | 93 | o.id = 1 |
Matteo Scandolo | b8621cd | 2018-04-04 17:12:37 -0700 | [diff] [blame] | 94 | |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 95 | o.tologdict.return_value = {'name': "Mock VOLTServiceInstance"} |
| 96 | |
Andy Bavier | 00c573c | 2019-02-08 16:19:11 -0700 | [diff] [blame] | 97 | o.save_changed_fields.return_value = "Saved" |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 98 | |
Matteo Scandolo | 6be6ee9 | 2018-05-24 15:07:51 -0700 | [diff] [blame] | 99 | o.pon_ports.all.return_value = [pon_port] |
Matteo Scandolo | b8621cd | 2018-04-04 17:12:37 -0700 | [diff] [blame] | 100 | |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 101 | self.o = o |
| 102 | |
Matteo Scandolo | d6fce51 | 2018-10-16 10:35:29 -0700 | [diff] [blame] | 103 | self.voltha_devices_response = {"id": "123", "serial_number": "foobar"} |
| 104 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 105 | self.tp = TechnologyProfile( |
| 106 | technology="xgspon", |
| 107 | profile_id=64, |
| 108 | profile_value="{}" |
| 109 | ) |
| 110 | |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 111 | def tearDown(self): |
| 112 | self.o = None |
Matteo Scandolo | ce27e9c | 2018-04-06 10:06:53 -0700 | [diff] [blame] | 113 | sys.path = self.sys_path_save |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 114 | |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 115 | @requests_mock.Mocker() |
| 116 | def test_get_of_id_from_device(self, m): |
| 117 | logical_devices = { |
| 118 | "items": [ |
Matteo Scandolo | f6337eb | 2018-04-05 15:58:37 -0700 | [diff] [blame] | 119 | {"root_device_id": "123", "id": "0001000ce2314000", "datapath_id": "55334486016"}, |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 120 | {"root_device_id": "0001cc4974a62b87", "id": "0001000000000001"} |
| 121 | ] |
| 122 | } |
Luca Prete | ca974c8 | 2018-05-01 18:06:16 -0700 | [diff] [blame] | 123 | m.get("http://voltha_url:1234/api/v1/logical_devices", status_code=200, json=logical_devices) |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 124 | self.o.device_id = "123" |
Matteo Scandolo | f6337eb | 2018-04-05 15:58:37 -0700 | [diff] [blame] | 125 | self.o = self.sync_step.get_ids_from_logical_device(self.o) |
| 126 | self.assertEqual(self.o.of_id, "0001000ce2314000") |
| 127 | self.assertEqual(self.o.dp_id, "of:0000000ce2314000") |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 128 | |
| 129 | with self.assertRaises(Exception) as e: |
| 130 | self.o.device_id = "idonotexist" |
Matteo Scandolo | f6337eb | 2018-04-05 15:58:37 -0700 | [diff] [blame] | 131 | self.sync_step.get_ids_from_logical_device(self.o) |
Matteo Scandolo | 2c14493 | 2018-05-04 14:06:24 -0700 | [diff] [blame] | 132 | self.assertEqual(e.exception.message, "Can't find a logical_device for OLT device id: idonotexist") |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 133 | |
| 134 | @requests_mock.Mocker() |
| 135 | def test_sync_record_fail_add(self, m): |
| 136 | """ |
| 137 | Should print an error if we can't add the device in VOLTHA |
| 138 | """ |
Luca Prete | ca974c8 | 2018-05-01 18:06:16 -0700 | [diff] [blame] | 139 | m.post("http://voltha_url:1234/api/v1/devices", status_code=500, text="MockError") |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 140 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 141 | with self.assertRaises(Exception) as e, \ |
| 142 | patch.object(TechnologyProfile.objects, "get") as tp_mock: |
| 143 | tp_mock.return_value = self.tp |
| 144 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 145 | self.sync_step(model_accessor=self.model_accessor).sync_record(self.o) |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 146 | |
Matteo Scandolo | 2c14493 | 2018-05-04 14:06:24 -0700 | [diff] [blame] | 147 | self.assertEqual(e.exception.message, "Failed to add OLT device: MockError") |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 148 | |
| 149 | @requests_mock.Mocker() |
| 150 | def test_sync_record_fail_no_id(self, m): |
| 151 | """ |
| 152 | Should print an error if VOLTHA does not return the device id |
| 153 | """ |
Luca Prete | ca974c8 | 2018-05-01 18:06:16 -0700 | [diff] [blame] | 154 | m.post("http://voltha_url:1234/api/v1/devices", status_code=200, json={"id": ""}) |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 155 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 156 | with self.assertRaises(Exception) as e, \ |
| 157 | patch.object(TechnologyProfile.objects, "get") as tp_mock: |
| 158 | tp_mock.return_value = self.tp |
| 159 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 160 | self.sync_step(model_accessor=self.model_accessor).sync_record(self.o) |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 161 | |
Matteo Scandolo | 2c14493 | 2018-05-04 14:06:24 -0700 | [diff] [blame] | 162 | self.assertEqual(e.exception.message, "VOLTHA Device Id is empty. This probably means that the OLT device is already provisioned in VOLTHA") |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 163 | |
| 164 | @requests_mock.Mocker() |
| 165 | def test_sync_record_fail_enable(self, m): |
| 166 | """ |
| 167 | Should print an error if device.enable fails |
| 168 | """ |
Matteo Scandolo | d6fce51 | 2018-10-16 10:35:29 -0700 | [diff] [blame] | 169 | m.post("http://voltha_url:1234/api/v1/devices", status_code=200, json=self.voltha_devices_response) |
Luca Prete | ca974c8 | 2018-05-01 18:06:16 -0700 | [diff] [blame] | 170 | m.post("http://voltha_url:1234/api/v1/devices/123/enable", status_code=500, text="EnableError") |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 171 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 172 | with self.assertRaises(Exception) as e, \ |
| 173 | patch.object(TechnologyProfile.objects, "get") as tp_mock: |
| 174 | tp_mock.return_value = self.tp |
| 175 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 176 | self.sync_step(model_accessor=self.model_accessor).sync_record(self.o) |
Matteo Scandolo | 2c14493 | 2018-05-04 14:06:24 -0700 | [diff] [blame] | 177 | |
| 178 | self.assertEqual(e.exception.message, "Failed to enable OLT device: EnableError") |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 179 | |
| 180 | @requests_mock.Mocker() |
| 181 | def test_sync_record_success(self, m): |
| 182 | """ |
| 183 | If device.enable succed should fetch the state, retrieve the of_id and push it to ONOS |
| 184 | """ |
Matteo Scandolo | 2ed64b9 | 2018-06-18 10:32:56 -0700 | [diff] [blame] | 185 | |
| 186 | expected_conf = { |
| 187 | "type": self.o.device_type, |
| 188 | "host_and_port": "%s:%s" % (self.o.host, self.o.port) |
| 189 | } |
| 190 | |
Matteo Scandolo | d6fce51 | 2018-10-16 10:35:29 -0700 | [diff] [blame] | 191 | m.post("http://voltha_url:1234/api/v1/devices", status_code=200, json=self.voltha_devices_response, additional_matcher=functools.partial(match_json, expected_conf)) |
Luca Prete | ca974c8 | 2018-05-01 18:06:16 -0700 | [diff] [blame] | 192 | m.post("http://voltha_url:1234/api/v1/devices/123/enable", status_code=200) |
Matteo Scandolo | 4587665 | 2018-10-16 16:03:17 -0700 | [diff] [blame] | 193 | m.get("http://voltha_url:1234/api/v1/devices/123", json={"oper_status": "ACTIVE", "admin_state": "ENABLED", "serial_number": "foobar"}) |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 194 | logical_devices = { |
| 195 | "items": [ |
Matteo Scandolo | f6337eb | 2018-04-05 15:58:37 -0700 | [diff] [blame] | 196 | {"root_device_id": "123", "id": "0001000ce2314000", "datapath_id": "55334486016"}, |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 197 | {"root_device_id": "0001cc4974a62b87", "id": "0001000000000001"} |
| 198 | ] |
| 199 | } |
Luca Prete | ca974c8 | 2018-05-01 18:06:16 -0700 | [diff] [blame] | 200 | m.get("http://voltha_url:1234/api/v1/logical_devices", status_code=200, json=logical_devices) |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 201 | |
Matteo Scandolo | a79395f | 2018-10-08 13:34:49 -0700 | [diff] [blame] | 202 | onos_expected_conf = { |
| 203 | "devices": { |
| 204 | "of:0000000ce2314000": { |
| 205 | "basic": { |
| 206 | "name": self.o.name |
| 207 | } |
| 208 | } |
| 209 | } |
| 210 | } |
| 211 | m.post("http://onos:4321/onos/v1/network/configuration/", status_code=200, json=onos_expected_conf, |
| 212 | additional_matcher=functools.partial(match_json, onos_expected_conf)) |
| 213 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 214 | with patch.object(TechnologyProfile.objects, "get") as tp_mock: |
| 215 | tp_mock.return_value = self.tp |
Scott Baker | 09798d8 | 2019-01-17 08:34:59 -0800 | [diff] [blame] | 216 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 217 | self.sync_step(model_accessor=self.model_accessor).sync_record(self.o) |
| 218 | self.assertEqual(self.o.admin_state, "ENABLED") |
| 219 | self.assertEqual(self.o.oper_status, "ACTIVE") |
| 220 | self.assertEqual(self.o.serial_number, "foobar") |
| 221 | self.assertEqual(self.o.of_id, "0001000ce2314000") |
| 222 | |
| 223 | # One save during preprovision |
| 224 | # One save during activation to set backend_status to "Waiting for device to activate" |
| 225 | # One save after activation has succeeded |
| 226 | self.assertEqual(self.o.save_changed_fields.call_count, 3) |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 227 | |
| 228 | @requests_mock.Mocker() |
Matteo Scandolo | 2ed64b9 | 2018-06-18 10:32:56 -0700 | [diff] [blame] | 229 | def test_sync_record_success_mac_address(self, m): |
| 230 | """ |
| 231 | A device should be pre-provisioned via mac_address, the the process is the same |
| 232 | """ |
| 233 | |
| 234 | del self.o.host |
| 235 | del self.o.port |
| 236 | self.o.mac_address = "00:0c:e2:31:40:00" |
| 237 | |
| 238 | expected_conf = { |
| 239 | "type": self.o.device_type, |
| 240 | "mac_address": self.o.mac_address |
| 241 | } |
| 242 | |
Matteo Scandolo | a79395f | 2018-10-08 13:34:49 -0700 | [diff] [blame] | 243 | onos_expected_conf = { |
| 244 | "devices": { |
| 245 | "of:0000000ce2314000": { |
| 246 | "basic": { |
| 247 | "name": self.o.name |
| 248 | } |
| 249 | } |
| 250 | } |
| 251 | } |
| 252 | m.post("http://onos:4321/onos/v1/network/configuration/", status_code=200, json=onos_expected_conf, |
| 253 | additional_matcher=functools.partial(match_json, onos_expected_conf)) |
| 254 | |
Matteo Scandolo | d6fce51 | 2018-10-16 10:35:29 -0700 | [diff] [blame] | 255 | m.post("http://voltha_url:1234/api/v1/devices", status_code=200, json=self.voltha_devices_response, |
Matteo Scandolo | 2ed64b9 | 2018-06-18 10:32:56 -0700 | [diff] [blame] | 256 | additional_matcher=functools.partial(match_json, expected_conf)) |
| 257 | m.post("http://voltha_url:1234/api/v1/devices/123/enable", status_code=200) |
Matteo Scandolo | 4587665 | 2018-10-16 16:03:17 -0700 | [diff] [blame] | 258 | m.get("http://voltha_url:1234/api/v1/devices/123", json={"oper_status": "ACTIVE", "admin_state": "ENABLED", "serial_number": "foobar"}) |
Matteo Scandolo | 2ed64b9 | 2018-06-18 10:32:56 -0700 | [diff] [blame] | 259 | logical_devices = { |
| 260 | "items": [ |
| 261 | {"root_device_id": "123", "id": "0001000ce2314000", "datapath_id": "55334486016"}, |
| 262 | {"root_device_id": "0001cc4974a62b87", "id": "0001000000000001"} |
| 263 | ] |
| 264 | } |
| 265 | m.get("http://voltha_url:1234/api/v1/logical_devices", status_code=200, json=logical_devices) |
| 266 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 267 | with patch.object(TechnologyProfile.objects, "get") as tp_mock: |
| 268 | tp_mock.return_value = self.tp |
Scott Baker | 09798d8 | 2019-01-17 08:34:59 -0800 | [diff] [blame] | 269 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 270 | self.sync_step(model_accessor=self.model_accessor).sync_record(self.o) |
| 271 | self.assertEqual(self.o.admin_state, "ENABLED") |
| 272 | self.assertEqual(self.o.oper_status, "ACTIVE") |
| 273 | self.assertEqual(self.o.of_id, "0001000ce2314000") |
| 274 | |
| 275 | # One save during preprovision |
| 276 | # One save during activation to set backend_status to "Waiting for device to activate" |
| 277 | # One save after activation has succeeded |
| 278 | self.assertEqual(self.o.save_changed_fields.call_count, 3) |
Matteo Scandolo | 096a3cf | 2018-06-20 13:56:13 -0700 | [diff] [blame] | 279 | |
| 280 | @requests_mock.Mocker() |
| 281 | def test_sync_record_enable_timeout(self, m): |
| 282 | """ |
Scott Baker | 09798d8 | 2019-01-17 08:34:59 -0800 | [diff] [blame] | 283 | If device activation fails we need to tell the user. |
| 284 | |
| 285 | OLT will be preprovisioned. |
| 286 | OLT will return "ERROR" for oper_status during activate and will eventually exceed retries.s |
Matteo Scandolo | 096a3cf | 2018-06-20 13:56:13 -0700 | [diff] [blame] | 287 | """ |
| 288 | |
| 289 | expected_conf = { |
| 290 | "type": self.o.device_type, |
| 291 | "host_and_port": "%s:%s" % (self.o.host, self.o.port) |
| 292 | } |
| 293 | |
Matteo Scandolo | d6fce51 | 2018-10-16 10:35:29 -0700 | [diff] [blame] | 294 | m.post("http://voltha_url:1234/api/v1/devices", status_code=200, json=self.voltha_devices_response, |
Matteo Scandolo | 096a3cf | 2018-06-20 13:56:13 -0700 | [diff] [blame] | 295 | additional_matcher=functools.partial(match_json, expected_conf)) |
| 296 | m.post("http://voltha_url:1234/api/v1/devices/123/enable", status_code=200) |
| 297 | m.get("http://voltha_url:1234/api/v1/devices/123", [ |
Matteo Scandolo | 4587665 | 2018-10-16 16:03:17 -0700 | [diff] [blame] | 298 | {"json": {"oper_status": "ACTIVATING", "admin_state": "ENABLED", "serial_number": "foobar"}, "status_code": 200}, |
Scott Baker | 09798d8 | 2019-01-17 08:34:59 -0800 | [diff] [blame] | 299 | {"json": {"oper_status": "ERROR", "admin_state": "ENABLED", "serial_number": "foobar"}, "status_code": 200} |
Matteo Scandolo | 096a3cf | 2018-06-20 13:56:13 -0700 | [diff] [blame] | 300 | ]) |
| 301 | |
| 302 | logical_devices = { |
| 303 | "items": [ |
| 304 | {"root_device_id": "123", "id": "0001000ce2314000", "datapath_id": "55334486016"}, |
| 305 | {"root_device_id": "0001cc4974a62b87", "id": "0001000000000001"} |
| 306 | ] |
| 307 | } |
| 308 | m.get("http://voltha_url:1234/api/v1/logical_devices", status_code=200, json=logical_devices) |
| 309 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 310 | with self.assertRaises(Exception) as e, \ |
| 311 | patch.object(TechnologyProfile.objects, "get") as tp_mock: |
| 312 | tp_mock.return_value = self.tp |
| 313 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 314 | self.sync_step(model_accessor=self.model_accessor).sync_record(self.o) |
Matteo Scandolo | 096a3cf | 2018-06-20 13:56:13 -0700 | [diff] [blame] | 315 | |
| 316 | self.assertEqual(e.exception.message, "It was not possible to activate OLTDevice with id 1") |
| 317 | self.assertEqual(self.o.oper_status, "ERROR") |
Scott Baker | 09798d8 | 2019-01-17 08:34:59 -0800 | [diff] [blame] | 318 | self.assertEqual(self.o.admin_state, "ENABLED") |
| 319 | self.assertEqual(self.o.device_id, "123") |
| 320 | self.assertEqual(self.o.serial_number, "foobar") |
| 321 | |
| 322 | # One save from preprovision to set device_id, serial_number |
| 323 | # One save from activate to set backend_status to "Waiting for device to be activated" |
Andy Bavier | 00c573c | 2019-02-08 16:19:11 -0700 | [diff] [blame] | 324 | self.assertEqual(self.o.save_changed_fields.call_count, 2) |
Matteo Scandolo | 2ed64b9 | 2018-06-18 10:32:56 -0700 | [diff] [blame] | 325 | |
| 326 | @requests_mock.Mocker() |
Matteo Scandolo | b8621cd | 2018-04-04 17:12:37 -0700 | [diff] [blame] | 327 | def test_sync_record_already_existing_in_voltha(self, m): |
Scott Baker | 09798d8 | 2019-01-17 08:34:59 -0800 | [diff] [blame] | 328 | """ |
| 329 | If device.admin_state == "ENABLED" and oper_status == "ACTIVE", then the OLT should not be reactivated. |
| 330 | """ |
| 331 | |
Matteo Scandolo | b8621cd | 2018-04-04 17:12:37 -0700 | [diff] [blame] | 332 | # mock device feedback state |
| 333 | self.o.device_id = "123" |
Matteo Scandolo | 096a3cf | 2018-06-20 13:56:13 -0700 | [diff] [blame] | 334 | self.o.admin_state = "ENABLED" |
| 335 | self.o.oper_status = "ACTIVE" |
Matteo Scandolo | f6337eb | 2018-04-05 15:58:37 -0700 | [diff] [blame] | 336 | self.o.dp_id = "of:0000000ce2314000" |
Matteo Scandolo | 2c14493 | 2018-05-04 14:06:24 -0700 | [diff] [blame] | 337 | self.o.of_id = "0001000ce2314000" |
Matteo Scandolo | b8621cd | 2018-04-04 17:12:37 -0700 | [diff] [blame] | 338 | |
Matteo Scandolo | a79395f | 2018-10-08 13:34:49 -0700 | [diff] [blame] | 339 | expected_conf = { |
| 340 | "devices": { |
| 341 | self.o.dp_id: { |
| 342 | "basic": { |
| 343 | "name": self.o.name |
| 344 | } |
| 345 | } |
| 346 | } |
| 347 | } |
| 348 | m.post("http://onos:4321/onos/v1/network/configuration/", status_code=200, json=expected_conf, |
| 349 | additional_matcher=functools.partial(match_json, expected_conf)) |
| 350 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 351 | with patch.object(TechnologyProfile.objects, "get") as tp_mock: |
| 352 | tp_mock.return_value = self.tp |
Andy Bavier | 00c573c | 2019-02-08 16:19:11 -0700 | [diff] [blame] | 353 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 354 | self.sync_step(model_accessor=self.model_accessor).sync_record(self.o) |
| 355 | self.o.save.assert_not_called() |
| 356 | self.o.save_changed_fields.assert_not_called() |
Matteo Scandolo | b8621cd | 2018-04-04 17:12:37 -0700 | [diff] [blame] | 357 | |
Matteo Scandolo | b8621cd | 2018-04-04 17:12:37 -0700 | [diff] [blame] | 358 | @requests_mock.Mocker() |
Scott Baker | 09798d8 | 2019-01-17 08:34:59 -0800 | [diff] [blame] | 359 | def test_sync_record_deactivate(self, m): |
| 360 | """ |
| 361 | If device.admin_state == "DISABLED" and oper_status == "ACTIVE", then OLT should be deactivated. |
| 362 | """ |
| 363 | |
| 364 | expected_conf = { |
| 365 | "type": self.o.device_type, |
| 366 | "host_and_port": "%s:%s" % (self.o.host, self.o.port) |
| 367 | } |
| 368 | |
| 369 | # Make it look like we have an active OLT that we are deactivating. |
| 370 | self.o.admin_state = "DISABLED" |
| 371 | self.o.oper_status = "ACTIVE" |
| 372 | self.o.serial_number = "foobar" |
| 373 | self.o.device_id = "123" |
| 374 | self.o.of_id = "0001000ce2314000" |
| 375 | |
| 376 | m.post("http://voltha_url:1234/api/v1/devices", status_code=200, json=self.voltha_devices_response, additional_matcher=functools.partial(match_json, expected_conf)) |
| 377 | m.post("http://voltha_url:1234/api/v1/devices/123/disable", status_code=200) |
| 378 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 379 | with patch.object(TechnologyProfile.objects, "get") as tp_mock: |
| 380 | tp_mock.return_value = self.tp |
Scott Baker | 09798d8 | 2019-01-17 08:34:59 -0800 | [diff] [blame] | 381 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 382 | self.sync_step(model_accessor=self.model_accessor).sync_record(self.o) |
Andy Bavier | 00c573c | 2019-02-08 16:19:11 -0700 | [diff] [blame] | 383 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 384 | # No saves as state has not changed (will eventually be saved |
| 385 | # by the synchronizer framework to update backend_status) |
| 386 | self.assertEqual(self.o.save.call_count, 0) |
| 387 | self.assertEqual(self.o.save_changed_fields.call_count, 0) |
Scott Baker | 09798d8 | 2019-01-17 08:34:59 -0800 | [diff] [blame] | 388 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 389 | # Make sure disable was called |
| 390 | urls = [x.url for x in m.request_history] |
| 391 | self.assertIn("http://voltha_url:1234/api/v1/devices/123/disable", urls) |
Scott Baker | 09798d8 | 2019-01-17 08:34:59 -0800 | [diff] [blame] | 392 | |
| 393 | @requests_mock.Mocker() |
| 394 | def test_sync_record_deactivate_already_inactive(self, m): |
| 395 | """ |
| 396 | If device.admin_state == "DISABLED" and device.oper_status == "UNKNOWN", then the device is already deactivated |
| 397 | and VOLTHA should not be called. |
| 398 | """ |
| 399 | |
| 400 | expected_conf = { |
| 401 | "type": self.o.device_type, |
| 402 | "host_and_port": "%s:%s" % (self.o.host, self.o.port) |
| 403 | } |
| 404 | |
| 405 | # Make it look like we have an active OLT that we are deactivating. |
| 406 | self.o.admin_state = "DISABLED" |
| 407 | self.o.oper_status = "UNKNOWN" |
| 408 | self.o.serial_number = "foobar" |
| 409 | self.o.device_id = "123" |
| 410 | self.o.of_id = "0001000ce2314000" |
| 411 | |
| 412 | m.post("http://voltha_url:1234/api/v1/devices", status_code=200, json=self.voltha_devices_response, additional_matcher=functools.partial(match_json, expected_conf)) |
| 413 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 414 | with patch.object(TechnologyProfile.objects, "get") as tp_mock: |
| 415 | tp_mock.return_value = self.tp |
Scott Baker | 09798d8 | 2019-01-17 08:34:59 -0800 | [diff] [blame] | 416 | |
Matteo Scandolo | 2b4c847 | 2019-06-26 18:06:47 -0700 | [diff] [blame] | 417 | self.sync_step(model_accessor=self.model_accessor).sync_record(self.o) |
| 418 | |
| 419 | # No saves as state has not changed (will eventually be saved by synchronizer framework |
| 420 | # to update backend_status) |
| 421 | self.assertEqual(self.o.save.call_count, 0) |
| 422 | self.assertEqual(self.o.save_changed_fields.call_count, 0) |
| 423 | |
| 424 | def test_do_not_sync_without_tech_profile(self): |
| 425 | self.o.technology = "xgspon" |
| 426 | with self.assertRaises(DeferredException) as e: |
| 427 | |
| 428 | self.sync_step(model_accessor=self.model_accessor).sync_record(self.o) |
| 429 | |
| 430 | self.assertEqual(e.exception.message, "Waiting for a TechnologyProfile (technology=xgspon) to be synchronized") |
Scott Baker | 09798d8 | 2019-01-17 08:34:59 -0800 | [diff] [blame] | 431 | |
| 432 | @requests_mock.Mocker() |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 433 | def test_delete_record(self, m): |
Matteo Scandolo | f6337eb | 2018-04-05 15:58:37 -0700 | [diff] [blame] | 434 | self.o.of_id = "0001000ce2314000" |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 435 | self.o.device_id = "123" |
| 436 | |
Luca Prete | ca974c8 | 2018-05-01 18:06:16 -0700 | [diff] [blame] | 437 | m.post("http://voltha_url:1234/api/v1/devices/123/disable", status_code=200) |
| 438 | m.delete("http://voltha_url:1234/api/v1/devices/123/delete", status_code=200) |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 439 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 440 | self.sync_step(model_accessor=self.model_accessor).delete_record(self.o) |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 441 | |
Matteo Scandolo | 563891c | 2018-08-21 11:56:32 -0700 | [diff] [blame] | 442 | self.assertEqual(m.call_count, 2) |
| 443 | |
Scott Baker | 22b46c5 | 2018-11-15 15:15:29 -0800 | [diff] [blame] | 444 | @patch('requests.post') |
| 445 | def test_delete_record_connectionerror(self, m): |
| 446 | self.o.of_id = "0001000ce2314000" |
| 447 | self.o.device_id = "123" |
| 448 | |
| 449 | m.side_effect = ConnectionError() |
| 450 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 451 | self.sync_step(model_accessor=self.model_accessor).delete_record(self.o) |
Scott Baker | 22b46c5 | 2018-11-15 15:15:29 -0800 | [diff] [blame] | 452 | |
| 453 | # No exception thrown, as ConnectionError will be caught |
| 454 | |
| 455 | |
Matteo Scandolo | 563891c | 2018-08-21 11:56:32 -0700 | [diff] [blame] | 456 | @requests_mock.Mocker() |
| 457 | def test_delete_unsynced_record(self, m): |
| 458 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 459 | self.sync_step(model_accessor=self.model_accessor).delete_record(self.o) |
Matteo Scandolo | 563891c | 2018-08-21 11:56:32 -0700 | [diff] [blame] | 460 | |
| 461 | self.assertEqual(m.call_count, 0) |
Matteo Scandolo | 4a8b4d6 | 2018-03-06 17:18:46 -0800 | [diff] [blame] | 462 | |
| 463 | if __name__ == "__main__": |
Luca Prete | ca974c8 | 2018-05-01 18:06:16 -0700 | [diff] [blame] | 464 | unittest.main() |