def __init__(
self,
cmd: AzCliCommand,
app_id: str,
token: str,
consumer_group: str,
central_handler_args: CentralHandlerArguments,
central_dns_suffix: str,
):
central_device_provider = CentralDeviceProvider(cmd=cmd,
app_id=app_id,
token=token)
central_template_provider = CentralDeviceTemplateProvider(
cmd=cmd, app_id=app_id, token=token)
self._targets = self._build_targets(
cmd=cmd,
app_id=app_id,
token=token,
consumer_group=consumer_group,
central_dns_suffix=central_dns_suffix,
)
self._handler = self._build_handler(
central_device_provider=central_device_provider,
central_template_provider=central_template_provider,
central_handler_args=central_handler_args,
)
def registration_summary(
cmd, app_id: str, token=None, central_dns_suffix=CENTRAL_ENDPOINT,
):
provider = CentralDeviceProvider(cmd=cmd, app_id=app_id, token=token,)
return provider.get_device_registration_summary(
central_dns_suffix=central_dns_suffix,
)
def registration_info(
cmd, app_id: str, device_id, token=None, central_dns_suffix=CENTRAL_ENDPOINT,
):
provider = CentralDeviceProvider(cmd=cmd, app_id=app_id, token=token,)
return provider.get_device_registration_info(
device_id=device_id, central_dns_suffix=central_dns_suffix, device_status=None,
)
def _create_parser(self, device_template: dict, message: Message,
args: CommonParserArguments):
provider = CentralDeviceProvider(cmd=None, app_id=None)
provider.get_device_template_by_device_id = mock.MagicMock(
return_value=device_template)
return central_parser.CentralParser(message=message,
central_device_provider=provider,
common_parser_args=args)
def delete_device(
cmd,
app_id: str,
device_id: str,
token=None,
central_dns_suffix="azureiotcentral.com",
):
provider = CentralDeviceProvider(cmd=cmd, app_id=app_id, token=token)
return provider.delete_device(device_id)
def delete_device(
cmd,
app_id: str,
device_id: str,
token=None,
central_dns_suffix=CENTRAL_ENDPOINT,
):
provider = CentralDeviceProvider(cmd=cmd, app_id=app_id, token=token)
return provider.delete_device(device_id)
def registration_summary(
cmd,
app_id: str,
token=None,
central_dns_suffix="azureiotcentral.com",
):
provider = CentralDeviceProvider(
cmd=cmd,
app_id=app_id,
token=token,
)
return provider.get_device_registration_summary(
central_dns_suffix=central_dns_suffix, )
def _create_parser(self, device_template: Template, message: Message,
args: CommonParserArguments):
device_provider = CentralDeviceProvider(cmd=None, app_id=None)
template_provider = CentralDeviceTemplateProvider(cmd=None,
app_id=None)
device_provider.get_device = mock.MagicMock(return_value=Device({}))
template_provider.get_device_template = mock.MagicMock(
return_value=device_template)
return central_parser.CentralParser(
message=message,
central_device_provider=device_provider,
central_template_provider=template_provider,
common_parser_args=args,
)
def get_command_history(
cmd,
app_id: str,
device_id: str,
interface_id: str,
command_name: str,
token=None,
central_dns_suffix=CENTRAL_ENDPOINT,
):
provider = CentralDeviceProvider(cmd=cmd, app_id=app_id, token=token)
return provider.get_component_command_history(
device_id=device_id,
interface_id=interface_id,
command_name=command_name,
)
def get_credentials(
cmd,
app_id: str,
device_id,
token=None,
central_dns_suffix=CENTRAL_ENDPOINT,
):
provider = CentralDeviceProvider(
cmd=cmd,
app_id=app_id,
token=token,
)
return provider.get_device_credentials(
device_id=device_id,
central_dns_suffix=central_dns_suffix,
)
def test_should_return_device(self, mock_device_svc, mock_device_template_svc):
# setup
provider = CentralDeviceProvider(cmd=None, app_id=app_id)
mock_device_svc.get_device.return_value = self._device
mock_device_template_svc.get_device_template.return_value = (
self._device_template
)
# act
device = provider.get_device("someDeviceId")
# check that caching is working
device = provider.get_device("someDeviceId")
# verify
# call counts should be at most 1 since the provider has a cache
assert mock_device_svc.get_device.call_count == 1
assert mock_device_svc.get_device_template.call_count == 0
assert device == self._device
def registration_info(
cmd,
app_id: str,
device_id,
token=None,
central_dns_suffix="azureiotcentral.com",
):
provider = CentralDeviceProvider(
cmd=cmd,
app_id=app_id,
token=token,
)
return provider.get_device_registration_info(
device_id=device_id,
central_dns_suffix=central_dns_suffix,
device_status=None,
)
def run_command(
cmd,
app_id: str,
device_id: str,
interface_id: str,
command_name: str,
content: str,
token=None,
central_dns_suffix=CENTRAL_ENDPOINT,
):
if not isinstance(content, str):
raise CLIError("content must be a string: {}".format(content))
payload = utility.process_json_arg(content, argument_name="content")
provider = CentralDeviceProvider(cmd=cmd, app_id=app_id, token=token)
return provider.run_component_command(
device_id=device_id,
interface_id=interface_id,
command_name=command_name,
payload=payload,
)
def create_device(
cmd,
app_id: str,
device_id: str,
device_name=None,
instance_of=None,
simulated=False,
token=None,
central_dns_suffix=CENTRAL_ENDPOINT,
):
if simulated and not instance_of:
raise CLIError(
"Error: if you supply --simulated you must also specify --instance-of"
)
provider = CentralDeviceProvider(cmd=cmd, app_id=app_id, token=token)
return provider.create_device(
device_id=device_id,
device_name=device_name,
instance_of=instance_of,
simulated=simulated,
central_dns_suffix=central_dns_suffix,
)
def __init__(
self,
cmd,
app_id: str,
device_id: str,
token: str,
central_dns_suffix=CENTRAL_ENDPOINT,
):
self._cmd = cmd
self._app_id = app_id
self._device_id = device_id
self._token = token
self._central_dns_suffix = central_dns_suffix
self._device_twin_provider = CentralDeviceTwinProvider(
cmd=self._cmd,
app_id=self._app_id,
token=self._token,
device_id=self._device_id,
)
self._central_device_provider = CentralDeviceProvider(
cmd=self._cmd, app_id=self._app_id, token=self._token)
self._central_template_provider = CentralDeviceTemplateProvider(
cmd=self._cmd, app_id=self._app_id, token=self._token)
self._template = self._get_device_template()
def list_devices(cmd,
app_id: str,
token=None,
central_dns_suffix=CENTRAL_ENDPOINT):
provider = CentralDeviceProvider(cmd=cmd, app_id=app_id, token=token)
return provider.list_devices()
def list_devices(cmd,
app_id: str,
token=None,
central_dns_suffix="azureiotcentral.com"):
provider = CentralDeviceProvider(cmd=cmd, app_id=app_id, token=token)
return provider.list_devices()