def create_devices(self): """ Create the devices in Dojot. Returns a list with device IDs. """ try: template_id = DojotAPI.create_template(self.jwt) DojotAPI.create_devices(self.jwt, template_id, self.parser_args.devices, self.parser_args.batch) except Exception as exception: LOGGER.error(str(exception))
def test_successfully_create_one_device_one_batch(self, mock_requests): """ Should successfully create one device in one batch. """ DojotAPI.divide_loads.return_value = [1] self.args["data"] = json.dumps({ "templates": [self.template_id], "attrs": {}, "label": "CargoContainer_0" }) DojotAPI.create_devices(self.jwt, self.template_id, 1, 1) DojotAPI.divide_loads.assert_called_once_with(1, 1) self.assertEqual(DojotAPI.divide_loads.return_value, [1]) DojotAPI.call_api.assert_called_once_with(mock_requests.post, self.args, False)
def test_successfully_create_two_devices_two_batches(self, mock_requests): """ Should successfully create two devices in two batches. """ DojotAPI.divide_loads.return_value = [1, 1] self.args["data"] = json.dumps({ "templates": [self.template_id], "attrs": {}, "label": "CargoContainer_1" }) DojotAPI.create_devices(self.jwt, self.template_id, 2, 2) DojotAPI.divide_loads.assert_called_once_with(2, 2) self.assertEqual(DojotAPI.divide_loads.return_value, [1, 1]) self.assertEqual(DojotAPI.call_api.call_count, 2) DojotAPI.call_api.assert_has_calls([ mock.call(mock_requests.post, self.args, False), mock.call(mock_requests.post, self.args, False) ])