# coding: utf-8
# -------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for
# license information.
# --------------------------------------------------------------------------
import functools
from devtools_testutils import AzureRecordedTestCase, PowerShellPreparer
from azure.iot.deviceupdate import DeviceUpdateClient


class DeviceUpdateTest(AzureRecordedTestCase):
    def create_client(self, endpoint, instance_id):
        credential = self.get_credential(DeviceUpdateClient)
        return self.create_client_from_credential(
            DeviceUpdateClient,
            endpoint=endpoint,
            instance_id=instance_id,
            credential=credential
        )


DeviceUpdatePowerShellPreparer = functools.partial(
    PowerShellPreparer,
    "deviceupdate",
    deviceupdate_endpoint="foo.api.adu.microsoft.com",
    deviceupdate_instance_id="bar",
    deviceupdate_update_provider="foo",
    deviceupdate_update_name="bar",
    deviceupdate_update_version="1.2",
    deviceupdate_device_group="foo"
)