Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [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 | |
| 15 | import unittest |
| 16 | from mock import patch, call, Mock, PropertyMock |
| 17 | import requests_mock |
| 18 | |
| 19 | import os, sys |
| 20 | |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 21 | test_path=os.path.abspath(os.path.dirname(os.path.realpath(__file__))) |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 22 | |
| 23 | class TestSyncOLTDevice(unittest.TestCase): |
| 24 | |
| 25 | def setUp(self): |
| 26 | global DeferredException |
| 27 | |
| 28 | self.sys_path_save = sys.path |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 29 | |
| 30 | # Setting up the config module |
| 31 | from xosconfig import Config |
Matteo Scandolo | f7ebb11 | 2018-09-18 16:17:22 -0700 | [diff] [blame] | 32 | config = os.path.join(test_path, "../test_config.yaml") |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 33 | Config.clear() |
| 34 | Config.init(config, "synchronizer-config-schema.yaml") |
| 35 | # END Setting up the config module |
| 36 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 37 | from xossynchronizer.mock_modelaccessor_build import mock_modelaccessor_config |
| 38 | mock_modelaccessor_config(test_path, [("olt-service", "volt.xproto"), |
| 39 | ("vsg", "vsg.xproto"), |
Matteo Scandolo | e04a488 | 2019-02-14 10:22:24 -0800 | [diff] [blame] | 40 | ("rcord", "rcord.xproto"),]) |
Matteo Scandolo | 19466a0 | 2018-05-16 17:43:39 -0700 | [diff] [blame] | 41 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 42 | import xossynchronizer.modelaccessor |
| 43 | reload(xossynchronizer.modelaccessor) # in case nose2 loaded it in a previous test |
| 44 | |
| 45 | from xossynchronizer.modelaccessor import model_accessor |
| 46 | self.model_accessor = model_accessor |
| 47 | |
| 48 | from pull_olts import OLTDevicePullStep |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 49 | |
| 50 | # import all class names to globals |
| 51 | for (k, v) in model_accessor.all_model_classes.items(): |
| 52 | globals()[k] = v |
| 53 | |
| 54 | self.sync_step = OLTDevicePullStep |
| 55 | |
| 56 | # mock volt service |
| 57 | self.volt_service = Mock() |
| 58 | self.volt_service.id = "volt_service_id" |
| 59 | self.volt_service.voltha_url = "voltha_url" |
| 60 | self.volt_service.voltha_user = "voltha_user" |
| 61 | self.volt_service.voltha_pass = "voltha_pass" |
Matteo Scandolo | 19466a0 | 2018-05-16 17:43:39 -0700 | [diff] [blame] | 62 | self.volt_service.voltha_port = 1234 |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 63 | |
| 64 | # mock voltha responses |
| 65 | self.devices = { |
| 66 | "items": [ |
| 67 | { |
| 68 | "id": "test_id", |
| 69 | "type": "simulated_olt", |
| 70 | "host_and_port": "172.17.0.1:50060", |
| 71 | "admin_state": "ENABLED", |
Matteo Scandolo | d6fce51 | 2018-10-16 10:35:29 -0700 | [diff] [blame] | 72 | "oper_status": "ACTIVE", |
| 73 | "serial_number": "serial_number", |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 74 | } |
| 75 | ] |
| 76 | } |
| 77 | |
| 78 | self.logical_devices = { |
| 79 | "items": [ |
| 80 | { |
| 81 | "root_device_id": "test_id", |
| 82 | "id": "of_id", |
| 83 | "datapath_id": "55334486016" |
| 84 | } |
| 85 | ] |
| 86 | } |
| 87 | |
Matteo Scandolo | 6be6ee9 | 2018-05-24 15:07:51 -0700 | [diff] [blame] | 88 | self.ports = { |
| 89 | "items": [ |
| 90 | { |
| 91 | "label": "PON port", |
| 92 | "port_no": 1, |
| 93 | "type": "PON_OLT", |
| 94 | "admin_state": "ENABLED", |
| 95 | "oper_status": "ACTIVE" |
| 96 | }, |
| 97 | { |
| 98 | "label": "NNI facing Ethernet port", |
| 99 | "port_no": 2, |
| 100 | "type": "ETHERNET_NNI", |
| 101 | "admin_state": "ENABLED", |
| 102 | "oper_status": "ACTIVE" |
| 103 | } |
| 104 | ] |
| 105 | } |
| 106 | |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 107 | def tearDown(self): |
| 108 | sys.path = self.sys_path_save |
| 109 | |
| 110 | @requests_mock.Mocker() |
| 111 | def test_missing_volt_service(self, m): |
| 112 | self.assertFalse(m.called) |
| 113 | |
| 114 | @requests_mock.Mocker() |
Matteo Scandolo | 778fef7 | 2019-04-22 13:32:36 -0700 | [diff] [blame^] | 115 | def test_pull_host_and_port(self, m): |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 116 | |
| 117 | with patch.object(VOLTService.objects, "all") as olt_service_mock, \ |
Matteo Scandolo | 778fef7 | 2019-04-22 13:32:36 -0700 | [diff] [blame^] | 118 | patch.object(OLTDevice, "save", autospec=True) as mock_olt_save, \ |
Matteo Scandolo | 6be6ee9 | 2018-05-24 15:07:51 -0700 | [diff] [blame] | 119 | patch.object(PONPort, "save") as mock_pon_save, \ |
| 120 | patch.object(NNIPort, "save") as mock_nni_save: |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 121 | olt_service_mock.return_value = [self.volt_service] |
| 122 | |
Matteo Scandolo | 19466a0 | 2018-05-16 17:43:39 -0700 | [diff] [blame] | 123 | m.get("http://voltha_url:1234/api/v1/devices", status_code=200, json=self.devices) |
Matteo Scandolo | 6be6ee9 | 2018-05-24 15:07:51 -0700 | [diff] [blame] | 124 | m.get("http://voltha_url:1234/api/v1/devices/test_id/ports", status_code=200, json=self.ports) |
Matteo Scandolo | 19466a0 | 2018-05-16 17:43:39 -0700 | [diff] [blame] | 125 | m.get("http://voltha_url:1234/api/v1/logical_devices", status_code=200, json=self.logical_devices) |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 126 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 127 | self.sync_step(model_accessor=self.model_accessor).pull_records() |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 128 | |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 129 | |
Matteo Scandolo | 778fef7 | 2019-04-22 13:32:36 -0700 | [diff] [blame^] | 130 | saved_olts = mock_olt_save.call_args_list |
| 131 | simulated_olt = saved_olts[0][0][0] |
| 132 | self.assertEqual(len(saved_olts), 1) |
| 133 | |
| 134 | self.assertEqual(simulated_olt.admin_state, "ENABLED") |
| 135 | self.assertEqual(simulated_olt.oper_status, "ACTIVE") |
| 136 | self.assertEqual(simulated_olt.volt_service_id, "volt_service_id") |
| 137 | self.assertEqual(simulated_olt.device_id, "test_id") |
| 138 | self.assertEqual(simulated_olt.of_id, "of_id") |
| 139 | self.assertEqual(simulated_olt.dp_id, "of:0000000ce2314000") |
| 140 | |
| 141 | |
Matteo Scandolo | 6be6ee9 | 2018-05-24 15:07:51 -0700 | [diff] [blame] | 142 | mock_pon_save.assert_called() |
| 143 | mock_nni_save.assert_called() |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 144 | |
| 145 | @requests_mock.Mocker() |
Matteo Scandolo | 778fef7 | 2019-04-22 13:32:36 -0700 | [diff] [blame^] | 146 | def test_pull_mac_address(self, m): |
| 147 | devices = { |
| 148 | "items": [ |
| 149 | { |
| 150 | 'id': 'tibit_id', |
| 151 | 'type': 'tibit_olt', |
| 152 | 'mac_address': '70:b3:d5:52:30:6f', |
| 153 | 'admin_state': 'ENABLED', |
| 154 | 'oper_status': 'ACTIVE', |
| 155 | 'serial_number': 'OLT-70b3d552306f', |
| 156 | } |
| 157 | ] |
| 158 | } |
| 159 | |
| 160 | logical_devices = { |
| 161 | "items": [ |
| 162 | { |
| 163 | "root_device_id": "tibit_id", |
| 164 | "id": "of_id", |
| 165 | "datapath_id": "55334486017" |
| 166 | } |
| 167 | ] |
| 168 | } |
| 169 | |
| 170 | with patch.object(VOLTService.objects, "all") as olt_service_mock, \ |
| 171 | patch.object(OLTDevice, "save", autospec=True) as mock_olt_save, \ |
| 172 | patch.object(PONPort, "save") as mock_pon_save, \ |
| 173 | patch.object(NNIPort, "save") as mock_nni_save: |
| 174 | olt_service_mock.return_value = [self.volt_service] |
| 175 | |
| 176 | m.get("http://voltha_url:1234/api/v1/devices", status_code=200, json=devices) |
| 177 | m.get("http://voltha_url:1234/api/v1/devices/tibit_id/ports", status_code=200, json=self.ports) |
| 178 | m.get("http://voltha_url:1234/api/v1/logical_devices", status_code=200, json=logical_devices) |
| 179 | |
| 180 | |
| 181 | self.sync_step(model_accessor=self.model_accessor).pull_records() |
| 182 | |
| 183 | |
| 184 | saved_olts = mock_olt_save.call_args_list |
| 185 | tibit_olt = saved_olts[0][0][0] |
| 186 | self.assertEqual(len(saved_olts), 1) |
| 187 | |
| 188 | self.assertEqual(tibit_olt.admin_state, "ENABLED") |
| 189 | self.assertEqual(tibit_olt.oper_status, "ACTIVE") |
| 190 | self.assertEqual(tibit_olt.volt_service_id, "volt_service_id") |
| 191 | self.assertEqual(tibit_olt.device_id, "tibit_id") |
| 192 | self.assertEqual(tibit_olt.of_id, "of_id") |
| 193 | self.assertEqual(tibit_olt.dp_id, "of:0000000ce2314001") |
| 194 | |
| 195 | mock_pon_save.assert_called() |
| 196 | mock_nni_save.assert_called() |
| 197 | |
| 198 | |
| 199 | @requests_mock.Mocker() |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 200 | def test_pull_existing(self, m): |
| 201 | |
| 202 | existing_olt = Mock() |
Scott Baker | 09798d8 | 2019-01-17 08:34:59 -0800 | [diff] [blame] | 203 | existing_olt.admin_state = "ENABLED" |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 204 | existing_olt.enacted = 2 |
| 205 | existing_olt.updated = 1 |
| 206 | |
| 207 | with patch.object(VOLTService.objects, "all") as olt_service_mock, \ |
Matteo Scandolo | 6be6ee9 | 2018-05-24 15:07:51 -0700 | [diff] [blame] | 208 | patch.object(OLTDevice.objects, "filter") as mock_get, \ |
| 209 | patch.object(PONPort, "save") as mock_pon_save, \ |
| 210 | patch.object(NNIPort, "save") as mock_nni_save, \ |
| 211 | patch.object(existing_olt, "save") as mock_olt_save: |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 212 | olt_service_mock.return_value = [self.volt_service] |
| 213 | mock_get.return_value = [existing_olt] |
| 214 | |
Matteo Scandolo | 19466a0 | 2018-05-16 17:43:39 -0700 | [diff] [blame] | 215 | m.get("http://voltha_url:1234/api/v1/devices", status_code=200, json=self.devices) |
Matteo Scandolo | 6be6ee9 | 2018-05-24 15:07:51 -0700 | [diff] [blame] | 216 | m.get("http://voltha_url:1234/api/v1/devices/test_id/ports", status_code=200, json=self.ports) |
Matteo Scandolo | 19466a0 | 2018-05-16 17:43:39 -0700 | [diff] [blame] | 217 | m.get("http://voltha_url:1234/api/v1/logical_devices", status_code=200, json=self.logical_devices) |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 218 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 219 | self.sync_step(model_accessor=self.model_accessor).pull_records() |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 220 | |
| 221 | self.assertEqual(existing_olt.admin_state, "ENABLED") |
| 222 | self.assertEqual(existing_olt.oper_status, "ACTIVE") |
| 223 | self.assertEqual(existing_olt.volt_service_id, "volt_service_id") |
| 224 | self.assertEqual(existing_olt.device_id, "test_id") |
| 225 | self.assertEqual(existing_olt.of_id, "of_id") |
| 226 | self.assertEqual(existing_olt.dp_id, "of:0000000ce2314000") |
| 227 | |
Andy Bavier | 00c573c | 2019-02-08 16:19:11 -0700 | [diff] [blame] | 228 | # mock_olt_save.assert_called() |
Matteo Scandolo | 6be6ee9 | 2018-05-24 15:07:51 -0700 | [diff] [blame] | 229 | mock_pon_save.assert_called() |
| 230 | mock_nni_save.assert_called() |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 231 | |
| 232 | @requests_mock.Mocker() |
| 233 | def test_pull_existing_do_not_sync(self, m): |
| 234 | existing_olt = Mock() |
| 235 | existing_olt.enacted = 1 |
| 236 | existing_olt.updated = 2 |
Matteo Scandolo | 6be6ee9 | 2018-05-24 15:07:51 -0700 | [diff] [blame] | 237 | existing_olt.device_id = "test_id" |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 238 | |
| 239 | with patch.object(VOLTService.objects, "all") as olt_service_mock, \ |
Matteo Scandolo | 6be6ee9 | 2018-05-24 15:07:51 -0700 | [diff] [blame] | 240 | patch.object(OLTDevice.objects, "filter") as mock_get, \ |
| 241 | patch.object(PONPort, "save") as mock_pon_save, \ |
| 242 | patch.object(NNIPort, "save") as mock_nni_save, \ |
| 243 | patch.object(existing_olt, "save") as mock_olt_save: |
| 244 | |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 245 | olt_service_mock.return_value = [self.volt_service] |
Matteo Scandolo | 6be6ee9 | 2018-05-24 15:07:51 -0700 | [diff] [blame] | 246 | mock_get.return_value = [existing_olt] |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 247 | |
Matteo Scandolo | 19466a0 | 2018-05-16 17:43:39 -0700 | [diff] [blame] | 248 | m.get("http://voltha_url:1234/api/v1/devices", status_code=200, json=self.devices) |
Matteo Scandolo | 6be6ee9 | 2018-05-24 15:07:51 -0700 | [diff] [blame] | 249 | m.get("http://voltha_url:1234/api/v1/devices/test_id/ports", status_code=200, json=self.ports) |
Matteo Scandolo | 19466a0 | 2018-05-16 17:43:39 -0700 | [diff] [blame] | 250 | m.get("http://voltha_url:1234/api/v1/logical_devices", status_code=200, json=self.logical_devices) |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 251 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 252 | self.sync_step(model_accessor=self.model_accessor).pull_records() |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 253 | |
Matteo Scandolo | 6be6ee9 | 2018-05-24 15:07:51 -0700 | [diff] [blame] | 254 | mock_olt_save.assert_not_called() |
| 255 | mock_pon_save.assert_called() |
| 256 | mock_nni_save.assert_called() |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 257 | |
Matteo Scandolo | a4fb032 | 2018-08-20 17:35:29 -0700 | [diff] [blame] | 258 | @requests_mock.Mocker() |
| 259 | def test_pull_deleted_object(self, m): |
| 260 | existing_olt = Mock() |
| 261 | existing_olt.enacted = 2 |
| 262 | existing_olt.updated = 1 |
| 263 | existing_olt.device_id = "test_id" |
| 264 | |
| 265 | m.get("http://voltha_url:1234/api/v1/devices", status_code=200, json={"items": []}) |
| 266 | |
| 267 | with patch.object(VOLTService.objects, "all") as olt_service_mock, \ |
| 268 | patch.object(OLTDevice.objects, "get_items") as mock_get, \ |
| 269 | patch.object(existing_olt, "delete") as mock_olt_delete: |
| 270 | |
| 271 | olt_service_mock.return_value = [self.volt_service] |
| 272 | mock_get.return_value = [existing_olt] |
| 273 | |
Scott Baker | 47b4730 | 2019-01-30 16:55:07 -0800 | [diff] [blame] | 274 | self.sync_step(model_accessor=self.model_accessor).pull_records() |
Matteo Scandolo | a4fb032 | 2018-08-20 17:35:29 -0700 | [diff] [blame] | 275 | |
| 276 | mock_olt_delete.assert_called() |
| 277 | |
Vijaykumar Kushwaha | 377d1da | 2019-03-20 10:29:42 +0530 | [diff] [blame] | 278 | #[SEBA-367] Unit test for blank response recieved from Voltha |
| 279 | |
| 280 | @requests_mock.Mocker() |
| 281 | def test_blank_response_received(self, m): |
| 282 | |
| 283 | m.get("http://voltha_url:1234/api/v1/devices", status_code=200, text="") |
| 284 | with patch.object(VOLTService.objects, "all") as olt_service_mock, \ |
| 285 | patch.object(PONPort, "save") as mock_pon_save, \ |
| 286 | patch.object(NNIPort, "save") as mock_nni_save, \ |
| 287 | patch.object(OLTDevice.objects, "get_items") as mock_get: |
| 288 | |
| 289 | olt_service_mock.return_value = [self.volt_service] |
| 290 | |
| 291 | self.sync_step(model_accessor=self.model_accessor).pull_records() |
| 292 | |
| 293 | olt_service_mock.assert_called() |
| 294 | mock_pon_save.assert_not_called() |
| 295 | mock_nni_save.assert_not_called() |
| 296 | mock_get.assert_not_called() |
| 297 | |
| 298 | #[SEBA-367] Unit test for invalid json received from Voltha |
| 299 | |
| 300 | @requests_mock.Mocker() |
| 301 | def test_invalid_json(self, m): |
| 302 | |
| 303 | m.get("http://voltha_url:1234/api/v1/devices", status_code=200, text="{\"items\" : [host_and_port }") |
| 304 | with patch.object(VOLTService.objects, "all") as olt_service_mock, \ |
| 305 | patch.object(PONPort, "save") as mock_pon_save, \ |
| 306 | patch.object(NNIPort, "save") as mock_nni_save, \ |
| 307 | patch.object(OLTDevice.objects, "get_items") as mock_get: |
| 308 | |
| 309 | olt_service_mock.return_value = [self.volt_service] |
| 310 | |
| 311 | self.sync_step(model_accessor=self.model_accessor).pull_records() |
| 312 | |
| 313 | olt_service_mock.assert_called() |
| 314 | mock_pon_save.assert_not_called() |
| 315 | mock_nni_save.assert_not_called() |
| 316 | mock_get.assert_not_called() |
Matteo Scandolo | a4fb032 | 2018-08-20 17:35:29 -0700 | [diff] [blame] | 317 | |
Matteo Scandolo | 3352341 | 2018-04-12 15:21:13 -0700 | [diff] [blame] | 318 | if __name__ == "__main__": |
Vijaykumar Kushwaha | 377d1da | 2019-03-20 10:29:42 +0530 | [diff] [blame] | 319 | unittest.main() |
| 320 | |