Adopt Black and isort (SC-700) (#1157)

Applied Black and isort, fixed any linting issues, updated tox.ini
and CI.

1 files changed, 653 insertions, 485 deletions

diff --git a/tests/unittests/sources/test_azure_helper.py b/tests/unittests/sources/test_azure_helper.py
index 24c582c2..6f7f2890 100644
--- a/tests/unittests/sources/test_azure_helper.py
+++ b/tests/unittests/sources/test_azure_helper.py
@@ -9,10 +9,9 @@ from xml.etree import ElementTree
 from xml.sax.saxutils import escape, unescape
 
 from cloudinit.sources.helpers import azure as azure_helper
-from tests.unittests.helpers import CiTestCase, ExitStack, mock, populate_dir
-
-from cloudinit.util import load_file
 from cloudinit.sources.helpers.azure import WALinuxAgentShim as wa_shim
+from cloudinit.util import load_file
+from tests.unittests.helpers import CiTestCase, ExitStack, mock, populate_dir
 
 GOAL_STATE_TEMPLATE = """\
 <?xml version="1.0" encoding="utf-8"?>
@@ -52,7 +51,7 @@ GOAL_STATE_TEMPLATE = """\
 </GoalState>
 """
 
-HEALTH_REPORT_XML_TEMPLATE = '''\
+HEALTH_REPORT_XML_TEMPLATE = """\
 <?xml version="1.0" encoding="utf-8"?>
 <Health xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
  xmlns:xsd="http://www.w3.org/2001/XMLSchema">
@@ -70,14 +69,16 @@ HEALTH_REPORT_XML_TEMPLATE = '''\
     </RoleInstanceList>
   </Container>
 </Health>
-'''
+"""
 
-HEALTH_DETAIL_SUBSECTION_XML_TEMPLATE = dedent('''\
+HEALTH_DETAIL_SUBSECTION_XML_TEMPLATE = dedent(
+    """\
     <Details>
       <SubStatus>{health_substatus}</SubStatus>
       <Description>{health_description}</Description>
     </Details>
-    ''')
+    """
+)
 
 HEALTH_REPORT_DESCRIPTION_TRIM_LEN = 512
 
@@ -87,24 +88,27 @@ class SentinelException(Exception):
 
 
 class TestFindEndpoint(CiTestCase):
-
     def setUp(self):
         super(TestFindEndpoint, self).setUp()
         patches = ExitStack()
         self.addCleanup(patches.close)
 
         self.load_file = patches.enter_context(
-            mock.patch.object(azure_helper.util, 'load_file'))
+            mock.patch.object(azure_helper.util, "load_file")
+        )
 
         self.dhcp_options = patches.enter_context(
-            mock.patch.object(wa_shim, '_load_dhclient_json'))
+            mock.patch.object(wa_shim, "_load_dhclient_json")
+        )
 
         self.networkd_leases = patches.enter_context(
-            mock.patch.object(wa_shim, '_networkd_get_value_from_leases'))
+            mock.patch.object(wa_shim, "_networkd_get_value_from_leases")
+        )
         self.networkd_leases.return_value = None
 
     def test_missing_file(self):
-        """wa_shim find_endpoint uses default endpoint if leasefile not found
+        """wa_shim find_endpoint uses default endpoint if
+        leasefile not found
         """
         self.assertEqual(wa_shim.find_endpoint(), "168.63.129.16")
 
@@ -112,80 +116,93 @@ class TestFindEndpoint(CiTestCase):
         """wa_shim find_endpoint uses default endpoint if leasefile is found
         but does not contain DHCP Option 245 (whose value is the endpoint)
         """
-        self.load_file.return_value = ''
-        self.dhcp_options.return_value = {'eth0': {'key': 'value'}}
+        self.load_file.return_value = ""
+        self.dhcp_options.return_value = {"eth0": {"key": "value"}}
         self.assertEqual(wa_shim.find_endpoint(), "168.63.129.16")
 
     @staticmethod
     def _build_lease_content(encoded_address):
         endpoint = azure_helper._get_dhcp_endpoint_option_name()
-        return '\n'.join([
-            'lease {',
-            ' interface "eth0";',
-            ' option {0} {1};'.format(endpoint, encoded_address),
-            '}'])
+        return "\n".join(
+            [
+                "lease {",
+                ' interface "eth0";',
+                " option {0} {1};".format(endpoint, encoded_address),
+                "}",
+            ]
+        )
 
     def test_from_dhcp_client(self):
         self.dhcp_options.return_value = {"eth0": {"unknown_245": "5:4:3:2"}}
-        self.assertEqual('5.4.3.2', wa_shim.find_endpoint(None))
+        self.assertEqual("5.4.3.2", wa_shim.find_endpoint(None))
 
     def test_latest_lease_used(self):
-        encoded_addresses = ['5:4:3:2', '4:3:2:1']
-        file_content = '\n'.join([self._build_lease_content(encoded_address)
-                                  for encoded_address in encoded_addresses])
+        encoded_addresses = ["5:4:3:2", "4:3:2:1"]
+        file_content = "\n".join(
+            [
+                self._build_lease_content(encoded_address)
+                for encoded_address in encoded_addresses
+            ]
+        )
         self.load_file.return_value = file_content
-        self.assertEqual(encoded_addresses[-1].replace(':', '.'),
-                         wa_shim.find_endpoint("foobar"))
+        self.assertEqual(
+            encoded_addresses[-1].replace(":", "."),
+            wa_shim.find_endpoint("foobar"),
+        )
 
 
 class TestExtractIpAddressFromLeaseValue(CiTestCase):
-
     def test_hex_string(self):
-        ip_address, encoded_address = '98.76.54.32', '62:4c:36:20'
+        ip_address, encoded_address = "98.76.54.32", "62:4c:36:20"
         self.assertEqual(
-            ip_address, wa_shim.get_ip_from_lease_value(encoded_address))
+            ip_address, wa_shim.get_ip_from_lease_value(encoded_address)
+        )
 
     def test_hex_string_with_single_character_part(self):
-        ip_address, encoded_address = '4.3.2.1', '4:3:2:1'
+        ip_address, encoded_address = "4.3.2.1", "4:3:2:1"
         self.assertEqual(
-            ip_address, wa_shim.get_ip_from_lease_value(encoded_address))
+            ip_address, wa_shim.get_ip_from_lease_value(encoded_address)
+        )
 
     def test_packed_string(self):
-        ip_address, encoded_address = '98.76.54.32', 'bL6 '
+        ip_address, encoded_address = "98.76.54.32", "bL6 "
         self.assertEqual(
-            ip_address, wa_shim.get_ip_from_lease_value(encoded_address))
+            ip_address, wa_shim.get_ip_from_lease_value(encoded_address)
+        )
 
     def test_packed_string_with_escaped_quote(self):
-        ip_address, encoded_address = '100.72.34.108', 'dH\\"l'
+        ip_address, encoded_address = "100.72.34.108", 'dH\\"l'
         self.assertEqual(
-            ip_address, wa_shim.get_ip_from_lease_value(encoded_address))
+            ip_address, wa_shim.get_ip_from_lease_value(encoded_address)
+        )
 
     def test_packed_string_containing_a_colon(self):
-        ip_address, encoded_address = '100.72.58.108', 'dH:l'
+        ip_address, encoded_address = "100.72.58.108", "dH:l"
         self.assertEqual(
-            ip_address, wa_shim.get_ip_from_lease_value(encoded_address))
+            ip_address, wa_shim.get_ip_from_lease_value(encoded_address)
+        )
 
 
 class TestGoalStateParsing(CiTestCase):
 
     default_parameters = {
-        'incarnation': 1,
-        'container_id': 'MyContainerId',
-        'instance_id': 'MyInstanceId',
-        'certificates_url': 'MyCertificatesUrl',
+        "incarnation": 1,
+        "container_id": "MyContainerId",
+        "instance_id": "MyInstanceId",
+        "certificates_url": "MyCertificatesUrl",
     }
 
     def _get_formatted_goal_state_xml_string(self, **kwargs):
         parameters = self.default_parameters.copy()
         parameters.update(kwargs)
         xml = GOAL_STATE_TEMPLATE.format(**parameters)
-        if parameters['certificates_url'] is None:
+        if parameters["certificates_url"] is None:
             new_xml_lines = []
             for line in xml.splitlines():
-                if 'Certificates' in line:
+                if "Certificates" in line:
                     continue
                 new_xml_lines.append(line)
-            xml = '\n'.join(new_xml_lines)
+            xml = "\n".join(new_xml_lines)
         return xml
 
     def _get_goal_state(self, m_azure_endpoint_client=None, **kwargs):
@@ -195,17 +212,17 @@ class TestGoalStateParsing(CiTestCase):
         return azure_helper.GoalState(xml, m_azure_endpoint_client)
 
     def test_incarnation_parsed_correctly(self):
-        incarnation = '123'
+        incarnation = "123"
         goal_state = self._get_goal_state(incarnation=incarnation)
         self.assertEqual(incarnation, goal_state.incarnation)
 
     def test_container_id_parsed_correctly(self):
-        container_id = 'TestContainerId'
+        container_id = "TestContainerId"
         goal_state = self._get_goal_state(container_id=container_id)
         self.assertEqual(container_id, goal_state.container_id)
 
     def test_instance_id_parsed_correctly(self):
-        instance_id = 'TestInstanceId'
+        instance_id = "TestInstanceId"
         goal_state = self._get_goal_state(instance_id=instance_id)
         self.assertEqual(instance_id, goal_state.instance_id)
 
@@ -214,67 +231,72 @@ class TestGoalStateParsing(CiTestCase):
         previous_iid = "D0DF4C54-4ECB-4A4B-9954-5BDF3ED5C3B8"
         current_iid = "544CDFD0-CB4E-4B4A-9954-5BDF3ED5C3B8"
         self.assertTrue(
-            azure_helper.is_byte_swapped(previous_iid, current_iid))
+            azure_helper.is_byte_swapped(previous_iid, current_iid)
+        )
 
     def test_instance_id_no_byte_swap_same_instance_id(self):
         previous_iid = "D0DF4C54-4ECB-4A4B-9954-5BDF3ED5C3B8"
         current_iid = "D0DF4C54-4ECB-4A4B-9954-5BDF3ED5C3B8"
         self.assertFalse(
-            azure_helper.is_byte_swapped(previous_iid, current_iid))
+            azure_helper.is_byte_swapped(previous_iid, current_iid)
+        )
 
     def test_instance_id_no_byte_swap_diff_instance_id(self):
         previous_iid = "D0DF4C54-4ECB-4A4B-9954-5BDF3ED5C3B8"
         current_iid = "G0DF4C54-4ECB-4A4B-9954-5BDF3ED5C3B8"
         self.assertFalse(
-            azure_helper.is_byte_swapped(previous_iid, current_iid))
+            azure_helper.is_byte_swapped(previous_iid, current_iid)
+        )
 
     def test_certificates_xml_parsed_and_fetched_correctly(self):
         m_azure_endpoint_client = mock.MagicMock()
-        certificates_url = 'TestCertificatesUrl'
+        certificates_url = "TestCertificatesUrl"
         goal_state = self._get_goal_state(
             m_azure_endpoint_client=m_azure_endpoint_client,
-            certificates_url=certificates_url)
+            certificates_url=certificates_url,
+        )
         certificates_xml = goal_state.certificates_xml
         self.assertEqual(1, m_azure_endpoint_client.get.call_count)
         self.assertEqual(
-            certificates_url,
-            m_azure_endpoint_client.get.call_args[0][0])
+            certificates_url, m_azure_endpoint_client.get.call_args[0][0]
+        )
         self.assertTrue(
-            m_azure_endpoint_client.get.call_args[1].get(
-                'secure', False))
+            m_azure_endpoint_client.get.call_args[1].get("secure", False)
+        )
         self.assertEqual(
-            m_azure_endpoint_client.get.return_value.contents,
-            certificates_xml)
+            m_azure_endpoint_client.get.return_value.contents, certificates_xml
+        )
 
     def test_missing_certificates_skips_http_get(self):
         m_azure_endpoint_client = mock.MagicMock()
         goal_state = self._get_goal_state(
             m_azure_endpoint_client=m_azure_endpoint_client,
-            certificates_url=None)
+            certificates_url=None,
+        )
         certificates_xml = goal_state.certificates_xml
         self.assertEqual(0, m_azure_endpoint_client.get.call_count)
         self.assertIsNone(certificates_xml)
 
     def test_invalid_goal_state_xml_raises_parse_error(self):
-        xml = 'random non-xml data'
+        xml = "random non-xml data"
         with self.assertRaises(ElementTree.ParseError):
             azure_helper.GoalState(xml, mock.MagicMock())
 
     def test_missing_container_id_in_goal_state_xml_raises_exc(self):
         xml = self._get_formatted_goal_state_xml_string()
-        xml = re.sub('<ContainerId>.*</ContainerId>', '', xml)
+        xml = re.sub("<ContainerId>.*</ContainerId>", "", xml)
         with self.assertRaises(azure_helper.InvalidGoalStateXMLException):
             azure_helper.GoalState(xml, mock.MagicMock())
 
     def test_missing_instance_id_in_goal_state_xml_raises_exc(self):
         xml = self._get_formatted_goal_state_xml_string()
-        xml = re.sub('<InstanceId>.*</InstanceId>', '', xml)
+        xml = re.sub("<InstanceId>.*</InstanceId>", "", xml)
         with self.assertRaises(azure_helper.InvalidGoalStateXMLException):
             azure_helper.GoalState(xml, mock.MagicMock())
 
     def test_missing_incarnation_in_goal_state_xml_raises_exc(self):
         xml = self._get_formatted_goal_state_xml_string()
-        xml = re.sub('<Incarnation>.*</Incarnation>', '', xml)
+        xml = re.sub("<Incarnation>.*</Incarnation>", "", xml)
         with self.assertRaises(azure_helper.InvalidGoalStateXMLException):
             azure_helper.GoalState(xml, mock.MagicMock())
 
@@ -282,8 +304,8 @@ class TestGoalStateParsing(CiTestCase):
 class TestAzureEndpointHttpClient(CiTestCase):
 
     regular_headers = {
-        'x-ms-agent-name': 'WALinuxAgent',
-        'x-ms-version': '2012-11-30',
+        "x-ms-agent-name": "WALinuxAgent",
+        "x-ms-version": "2012-11-30",
     }
 
     def setUp(self):
@@ -291,43 +313,48 @@ class TestAzureEndpointHttpClient(CiTestCase):
         patches = ExitStack()
         self.addCleanup(patches.close)
         self.m_http_with_retries = patches.enter_context(
-            mock.patch.object(azure_helper, 'http_with_retries'))
+            mock.patch.object(azure_helper, "http_with_retries")
+        )
 
     def test_non_secure_get(self):
         client = azure_helper.AzureEndpointHttpClient(mock.MagicMock())
-        url = 'MyTestUrl'
+        url = "MyTestUrl"
         response = client.get(url, secure=False)
         self.assertEqual(1, self.m_http_with_retries.call_count)
         self.assertEqual(self.m_http_with_retries.return_value, response)
         self.assertEqual(
             mock.call(url, headers=self.regular_headers),
-            self.m_http_with_retries.call_args)
+            self.m_http_with_retries.call_args,
+        )
 
     def test_non_secure_get_raises_exception(self):
         client = azure_helper.AzureEndpointHttpClient(mock.MagicMock())
-        url = 'MyTestUrl'
+        url = "MyTestUrl"
         self.m_http_with_retries.side_effect = SentinelException
         self.assertRaises(SentinelException, client.get, url, secure=False)
         self.assertEqual(1, self.m_http_with_retries.call_count)
 
     def test_secure_get(self):
-        url = 'MyTestUrl'
+        url = "MyTestUrl"
         m_certificate = mock.MagicMock()
         expected_headers = self.regular_headers.copy()
-        expected_headers.update({
-            "x-ms-cipher-name": "DES_EDE3_CBC",
-            "x-ms-guest-agent-public-x509-cert": m_certificate,
-        })
+        expected_headers.update(
+            {
+                "x-ms-cipher-name": "DES_EDE3_CBC",
+                "x-ms-guest-agent-public-x509-cert": m_certificate,
+            }
+        )
         client = azure_helper.AzureEndpointHttpClient(m_certificate)
         response = client.get(url, secure=True)
         self.assertEqual(1, self.m_http_with_retries.call_count)
         self.assertEqual(self.m_http_with_retries.return_value, response)
         self.assertEqual(
             mock.call(url, headers=expected_headers),
-            self.m_http_with_retries.call_args)
+            self.m_http_with_retries.call_args,
+        )
 
     def test_secure_get_raises_exception(self):
-        url = 'MyTestUrl'
+        url = "MyTestUrl"
         client = azure_helper.AzureEndpointHttpClient(mock.MagicMock())
         self.m_http_with_retries.side_effect = SentinelException
         self.assertRaises(SentinelException, client.get, url, secure=True)
@@ -335,44 +362,50 @@ class TestAzureEndpointHttpClient(CiTestCase):
 
     def test_post(self):
         m_data = mock.MagicMock()
-        url = 'MyTestUrl'
+        url = "MyTestUrl"
         client = azure_helper.AzureEndpointHttpClient(mock.MagicMock())
         response = client.post(url, data=m_data)
         self.assertEqual(1, self.m_http_with_retries.call_count)
         self.assertEqual(self.m_http_with_retries.return_value, response)
         self.assertEqual(
             mock.call(url, data=m_data, headers=self.regular_headers),
-            self.m_http_with_retries.call_args)
+            self.m_http_with_retries.call_args,
+        )
 
     def test_post_raises_exception(self):
         m_data = mock.MagicMock()
-        url = 'MyTestUrl'
+        url = "MyTestUrl"
         client = azure_helper.AzureEndpointHttpClient(mock.MagicMock())
         self.m_http_with_retries.side_effect = SentinelException
         self.assertRaises(SentinelException, client.post, url, data=m_data)
         self.assertEqual(1, self.m_http_with_retries.call_count)
 
     def test_post_with_extra_headers(self):
-        url = 'MyTestUrl'
+        url = "MyTestUrl"
         client = azure_helper.AzureEndpointHttpClient(mock.MagicMock())
-        extra_headers = {'test': 'header'}
+        extra_headers = {"test": "header"}
         client.post(url, extra_headers=extra_headers)
         expected_headers = self.regular_headers.copy()
         expected_headers.update(extra_headers)
         self.assertEqual(1, self.m_http_with_retries.call_count)
         self.assertEqual(
             mock.call(url, data=mock.ANY, headers=expected_headers),
-            self.m_http_with_retries.call_args)
+            self.m_http_with_retries.call_args,
+        )
 
     def test_post_with_sleep_with_extra_headers_raises_exception(self):
         m_data = mock.MagicMock()
-        url = 'MyTestUrl'
-        extra_headers = {'test': 'header'}
+        url = "MyTestUrl"
+        extra_headers = {"test": "header"}
         client = azure_helper.AzureEndpointHttpClient(mock.MagicMock())
         self.m_http_with_retries.side_effect = SentinelException
         self.assertRaises(
-            SentinelException, client.post,
-            url, data=m_data, extra_headers=extra_headers)
+            SentinelException,
+            client.post,
+            url,
+            data=m_data,
+            extra_headers=extra_headers,
+        )
         self.assertEqual(1, self.m_http_with_retries.call_count)
 
 
@@ -392,128 +425,139 @@ class TestAzureHelperHttpWithRetries(CiTestCase):
 
         self.m_readurl = patches.enter_context(
             mock.patch.object(
-                azure_helper.url_helper, 'readurl', mock.MagicMock()))
+                azure_helper.url_helper, "readurl", mock.MagicMock()
+            )
+        )
         self.m_sleep = patches.enter_context(
-            mock.patch.object(azure_helper.time, 'sleep', autospec=True))
+            mock.patch.object(azure_helper.time, "sleep", autospec=True)
+        )
 
     def test_http_with_retries(self):
-        self.m_readurl.return_value = 'TestResp'
+        self.m_readurl.return_value = "TestResp"
         self.assertEqual(
-            azure_helper.http_with_retries('testurl'),
-            self.m_readurl.return_value)
+            azure_helper.http_with_retries("testurl"),
+            self.m_readurl.return_value,
+        )
         self.assertEqual(self.m_readurl.call_count, 1)
 
-    def test_http_with_retries_propagates_readurl_exc_and_logs_exc(
-            self):
+    def test_http_with_retries_propagates_readurl_exc_and_logs_exc(self):
         self.m_readurl.side_effect = SentinelException
 
         self.assertRaises(
-            SentinelException, azure_helper.http_with_retries, 'testurl')
+            SentinelException, azure_helper.http_with_retries, "testurl"
+        )
         self.assertEqual(self.m_readurl.call_count, self.max_readurl_attempts)
 
         self.assertIsNotNone(
             re.search(
-                r'Failed HTTP request with Azure endpoint \S* during '
-                r'attempt \d+ with exception: \S*',
-                self.logs.getvalue()))
+                r"Failed HTTP request with Azure endpoint \S* during "
+                r"attempt \d+ with exception: \S*",
+                self.logs.getvalue(),
+            )
+        )
         self.assertIsNone(
             re.search(
-                r'Successful HTTP request with Azure endpoint \S* after '
-                r'\d+ attempts',
-                self.logs.getvalue()))
+                r"Successful HTTP request with Azure endpoint \S* after "
+                r"\d+ attempts",
+                self.logs.getvalue(),
+            )
+        )
 
     def test_http_with_retries_delayed_success_due_to_temporary_readurl_exc(
-            self):
-        self.m_readurl.side_effect = \
-            [SentinelException] * self.periodic_logging_attempts + \
-            ['TestResp']
-        self.m_readurl.return_value = 'TestResp'
-
-        response = azure_helper.http_with_retries('testurl')
-        self.assertEqual(
-            response,
-            self.m_readurl.return_value)
+        self,
+    ):
+        self.m_readurl.side_effect = [
+            SentinelException
+        ] * self.periodic_logging_attempts + ["TestResp"]
+        self.m_readurl.return_value = "TestResp"
+
+        response = azure_helper.http_with_retries("testurl")
+        self.assertEqual(response, self.m_readurl.return_value)
         self.assertEqual(
-            self.m_readurl.call_count,
-            self.periodic_logging_attempts + 1)
+            self.m_readurl.call_count, self.periodic_logging_attempts + 1
+        )
 
         # Ensure that cloud-init did sleep between each failed request
         self.assertEqual(
-            self.m_sleep.call_count,
-            self.periodic_logging_attempts)
+            self.m_sleep.call_count, self.periodic_logging_attempts
+        )
         self.m_sleep.assert_called_with(self.sleep_duration_between_retries)
 
     def test_http_with_retries_long_delay_logs_periodic_failure_msg(self):
-        self.m_readurl.side_effect = \
-            [SentinelException] * self.periodic_logging_attempts + \
-            ['TestResp']
-        self.m_readurl.return_value = 'TestResp'
+        self.m_readurl.side_effect = [
+            SentinelException
+        ] * self.periodic_logging_attempts + ["TestResp"]
+        self.m_readurl.return_value = "TestResp"
 
-        azure_helper.http_with_retries('testurl')
+        azure_helper.http_with_retries("testurl")
 
         self.assertEqual(
-            self.m_readurl.call_count,
-            self.periodic_logging_attempts + 1)
+            self.m_readurl.call_count, self.periodic_logging_attempts + 1
+        )
         self.assertIsNotNone(
             re.search(
-                r'Failed HTTP request with Azure endpoint \S* during '
-                r'attempt \d+ with exception: \S*',
-                self.logs.getvalue()))
+                r"Failed HTTP request with Azure endpoint \S* during "
+                r"attempt \d+ with exception: \S*",
+                self.logs.getvalue(),
+            )
+        )
         self.assertIsNotNone(
             re.search(
-                r'Successful HTTP request with Azure endpoint \S* after '
-                r'\d+ attempts',
-                self.logs.getvalue()))
+                r"Successful HTTP request with Azure endpoint \S* after "
+                r"\d+ attempts",
+                self.logs.getvalue(),
+            )
+        )
 
     def test_http_with_retries_short_delay_does_not_log_periodic_failure_msg(
-            self):
-        self.m_readurl.side_effect = \
-            [SentinelException] * \
-            (self.periodic_logging_attempts - 1) + \
-            ['TestResp']
-        self.m_readurl.return_value = 'TestResp'
-
-        azure_helper.http_with_retries('testurl')
+        self,
+    ):
+        self.m_readurl.side_effect = [SentinelException] * (
+            self.periodic_logging_attempts - 1
+        ) + ["TestResp"]
+        self.m_readurl.return_value = "TestResp"
+
+        azure_helper.http_with_retries("testurl")
         self.assertEqual(
-            self.m_readurl.call_count,
-            self.periodic_logging_attempts)
+            self.m_readurl.call_count, self.periodic_logging_attempts
+        )
 
         self.assertIsNone(
             re.search(
-                r'Failed HTTP request with Azure endpoint \S* during '
-                r'attempt \d+ with exception: \S*',
-                self.logs.getvalue()))
+                r"Failed HTTP request with Azure endpoint \S* during "
+                r"attempt \d+ with exception: \S*",
+                self.logs.getvalue(),
+            )
+        )
         self.assertIsNotNone(
             re.search(
-                r'Successful HTTP request with Azure endpoint \S* after '
-                r'\d+ attempts',
-                self.logs.getvalue()))
+                r"Successful HTTP request with Azure endpoint \S* after "
+                r"\d+ attempts",
+                self.logs.getvalue(),
+            )
+        )
 
     def test_http_with_retries_calls_url_helper_readurl_with_args_kwargs(self):
         testurl = mock.MagicMock()
         kwargs = {
-            'headers': mock.MagicMock(),
-            'data': mock.MagicMock(),
+            "headers": mock.MagicMock(),
+            "data": mock.MagicMock(),
             # timeout kwarg should not be modified or deleted if present
-            'timeout': mock.MagicMock()
+            "timeout": mock.MagicMock(),
         }
         azure_helper.http_with_retries(testurl, **kwargs)
         self.m_readurl.assert_called_once_with(testurl, **kwargs)
 
     def test_http_with_retries_adds_timeout_kwarg_if_not_present(self):
         testurl = mock.MagicMock()
-        kwargs = {
-            'headers': mock.MagicMock(),
-            'data': mock.MagicMock()
-        }
+        kwargs = {"headers": mock.MagicMock(), "data": mock.MagicMock()}
         expected_kwargs = copy.deepcopy(kwargs)
-        expected_kwargs['timeout'] = self.default_readurl_timeout
+        expected_kwargs["timeout"] = self.default_readurl_timeout
 
         azure_helper.http_with_retries(testurl, **kwargs)
         self.m_readurl.assert_called_once_with(testurl, **expected_kwargs)
 
-    def test_http_with_retries_deletes_retries_kwargs_passed_in(
-            self):
+    def test_http_with_retries_deletes_retries_kwargs_passed_in(self):
         """http_with_retries already implements retry logic,
         so url_helper.readurl should not have retries.
         http_with_retries should delete kwargs that
@@ -521,44 +565,44 @@ class TestAzureHelperHttpWithRetries(CiTestCase):
         """
         testurl = mock.MagicMock()
         kwargs = {
-            'headers': mock.MagicMock(),
-            'data': mock.MagicMock(),
-            'timeout': mock.MagicMock(),
-            'retries': mock.MagicMock(),
-            'infinite': mock.MagicMock()
+            "headers": mock.MagicMock(),
+            "data": mock.MagicMock(),
+            "timeout": mock.MagicMock(),
+            "retries": mock.MagicMock(),
+            "infinite": mock.MagicMock(),
         }
         expected_kwargs = copy.deepcopy(kwargs)
-        expected_kwargs.pop('retries', None)
-        expected_kwargs.pop('infinite', None)
+        expected_kwargs.pop("retries", None)
+        expected_kwargs.pop("infinite", None)
 
         azure_helper.http_with_retries(testurl, **kwargs)
         self.m_readurl.assert_called_once_with(testurl, **expected_kwargs)
         self.assertIn(
-            'retries kwarg passed in for communication with Azure endpoint.',
-            self.logs.getvalue())
+            "retries kwarg passed in for communication with Azure endpoint.",
+            self.logs.getvalue(),
+        )
         self.assertIn(
-            'infinite kwarg passed in for communication with Azure endpoint.',
-            self.logs.getvalue())
+            "infinite kwarg passed in for communication with Azure endpoint.",
+            self.logs.getvalue(),
+        )
 
 
 class TestOpenSSLManager(CiTestCase):
-
     def setUp(self):
         super(TestOpenSSLManager, self).setUp()
         patches = ExitStack()
         self.addCleanup(patches.close)
 
         self.subp = patches.enter_context(
-            mock.patch.object(azure_helper.subp, 'subp'))
+            mock.patch.object(azure_helper.subp, "subp")
+        )
         try:
-            self.open = patches.enter_context(
-                mock.patch('__builtin__.open'))
+            self.open = patches.enter_context(mock.patch("__builtin__.open"))
         except ImportError:
-            self.open = patches.enter_context(
-                mock.patch('builtins.open'))
+            self.open = patches.enter_context(mock.patch("builtins.open"))
 
-    @mock.patch.object(azure_helper, 'cd', mock.MagicMock())
-    @mock.patch.object(azure_helper.temp_utils, 'mkdtemp')
+    @mock.patch.object(azure_helper, "cd", mock.MagicMock())
+    @mock.patch.object(azure_helper.temp_utils, "mkdtemp")
     def test_openssl_manager_creates_a_tmpdir(self, mkdtemp):
         manager = azure_helper.OpenSSLManager()
         self.assertEqual(mkdtemp.return_value, manager.tmpdir)
@@ -567,16 +611,16 @@ class TestOpenSSLManager(CiTestCase):
         subp_directory = {}
 
         def capture_directory(*args, **kwargs):
-            subp_directory['path'] = os.getcwd()
+            subp_directory["path"] = os.getcwd()
 
         self.subp.side_effect = capture_directory
         manager = azure_helper.OpenSSLManager()
-        self.assertEqual(manager.tmpdir, subp_directory['path'])
+        self.assertEqual(manager.tmpdir, subp_directory["path"])
         manager.clean_up()
 
-    @mock.patch.object(azure_helper, 'cd', mock.MagicMock())
-    @mock.patch.object(azure_helper.temp_utils, 'mkdtemp', mock.MagicMock())
-    @mock.patch.object(azure_helper.util, 'del_dir')
+    @mock.patch.object(azure_helper, "cd", mock.MagicMock())
+    @mock.patch.object(azure_helper.temp_utils, "mkdtemp", mock.MagicMock())
+    @mock.patch.object(azure_helper.util, "del_dir")
     def test_clean_up(self, del_dir):
         manager = azure_helper.OpenSSLManager()
         manager.clean_up()
@@ -584,43 +628,42 @@ class TestOpenSSLManager(CiTestCase):
 
 
 class TestOpenSSLManagerActions(CiTestCase):
-
     def setUp(self):
         super(TestOpenSSLManagerActions, self).setUp()
 
         self.allowed_subp = True
 
     def _data_file(self, name):
-        path = 'tests/data/azure'
+        path = "tests/data/azure"
         return os.path.join(path, name)
 
     @unittest.skip("todo move to cloud_test")
     def test_pubkey_extract(self):
-        cert = load_file(self._data_file('pubkey_extract_cert'))
-        good_key = load_file(self._data_file('pubkey_extract_ssh_key'))
+        cert = load_file(self._data_file("pubkey_extract_cert"))
+        good_key = load_file(self._data_file("pubkey_extract_ssh_key"))
         sslmgr = azure_helper.OpenSSLManager()
         key = sslmgr._get_ssh_key_from_cert(cert)
         self.assertEqual(good_key, key)
 
-        good_fingerprint = '073E19D14D1C799224C6A0FD8DDAB6A8BF27D473'
+        good_fingerprint = "073E19D14D1C799224C6A0FD8DDAB6A8BF27D473"
         fingerprint = sslmgr._get_fingerprint_from_cert(cert)
         self.assertEqual(good_fingerprint, fingerprint)
 
     @unittest.skip("todo move to cloud_test")
-    @mock.patch.object(azure_helper.OpenSSLManager, '_decrypt_certs_from_xml')
+    @mock.patch.object(azure_helper.OpenSSLManager, "_decrypt_certs_from_xml")
     def test_parse_certificates(self, mock_decrypt_certs):
         """Azure control plane puts private keys as well as certificates
-           into the Certificates XML object. Make sure only the public keys
-           from certs are extracted and that fingerprints are converted to
-           the form specified in the ovf-env.xml file.
+        into the Certificates XML object. Make sure only the public keys
+        from certs are extracted and that fingerprints are converted to
+        the form specified in the ovf-env.xml file.
         """
-        cert_contents = load_file(self._data_file('parse_certificates_pem'))
-        fingerprints = load_file(self._data_file(
-            'parse_certificates_fingerprints')
+        cert_contents = load_file(self._data_file("parse_certificates_pem"))
+        fingerprints = load_file(
+            self._data_file("parse_certificates_fingerprints")
         ).splitlines()
         mock_decrypt_certs.return_value = cert_contents
         sslmgr = azure_helper.OpenSSLManager()
-        keys_by_fp = sslmgr.parse_certificates('')
+        keys_by_fp = sslmgr.parse_certificates("")
         for fp in keys_by_fp.keys():
             self.assertIn(fp, fingerprints)
         for fp in fingerprints:
@@ -632,21 +675,23 @@ class TestGoalStateHealthReporter(CiTestCase):
     maxDiff = None
 
     default_parameters = {
-        'incarnation': 1634,
-        'container_id': 'MyContainerId',
-        'instance_id': 'MyInstanceId'
+        "incarnation": 1634,
+        "container_id": "MyContainerId",
+        "instance_id": "MyInstanceId",
     }
 
-    test_azure_endpoint = 'TestEndpoint'
-    test_health_report_url = 'http://{0}/machine?comp=health'.format(
-        test_azure_endpoint)
-    test_default_headers = {'Content-Type': 'text/xml; charset=utf-8'}
+    test_azure_endpoint = "TestEndpoint"
+    test_health_report_url = "http://{0}/machine?comp=health".format(
+        test_azure_endpoint
+    )
+    test_default_headers = {"Content-Type": "text/xml; charset=utf-8"}
 
-    provisioning_success_status = 'Ready'
-    provisioning_not_ready_status = 'NotReady'
-    provisioning_failure_substatus = 'ProvisioningFailed'
+    provisioning_success_status = "Ready"
+    provisioning_not_ready_status = "NotReady"
+    provisioning_failure_substatus = "ProvisioningFailed"
     provisioning_failure_err_description = (
-        'Test error message containing provisioning failure details')
+        "Test error message containing provisioning failure details"
+    )
 
     def setUp(self):
         super(TestGoalStateHealthReporter, self).setUp()
@@ -654,22 +699,28 @@ class TestGoalStateHealthReporter(CiTestCase):
         self.addCleanup(patches.close)
 
         patches.enter_context(
-            mock.patch.object(azure_helper.time, 'sleep', mock.MagicMock()))
+            mock.patch.object(azure_helper.time, "sleep", mock.MagicMock())
+        )
         self.read_file_or_url = patches.enter_context(
-            mock.patch.object(azure_helper.url_helper, 'read_file_or_url'))
+            mock.patch.object(azure_helper.url_helper, "read_file_or_url")
+        )
 
         self.post = patches.enter_context(
-            mock.patch.object(azure_helper.AzureEndpointHttpClient,
-                              'post'))
+            mock.patch.object(azure_helper.AzureEndpointHttpClient, "post")
+        )
 
         self.GoalState = patches.enter_context(
-            mock.patch.object(azure_helper, 'GoalState'))
-        self.GoalState.return_value.container_id = \
-            self.default_parameters['container_id']
-        self.GoalState.return_value.instance_id = \
-            self.default_parameters['instance_id']
-        self.GoalState.return_value.incarnation = \
-            self.default_parameters['incarnation']
+            mock.patch.object(azure_helper, "GoalState")
+        )
+        self.GoalState.return_value.container_id = self.default_parameters[
+            "container_id"
+        ]
+        self.GoalState.return_value.instance_id = self.default_parameters[
+            "instance_id"
+        ]
+        self.GoalState.return_value.incarnation = self.default_parameters[
+            "incarnation"
+        ]
 
     def _text_from_xpath_in_xroot(self, xroot, xpath):
         element = xroot.find(xpath)
@@ -685,34 +736,41 @@ class TestGoalStateHealthReporter(CiTestCase):
 
     def _get_report_ready_health_document(self):
         return self._get_formatted_health_report_xml_string(
-            incarnation=escape(str(self.default_parameters['incarnation'])),
-            container_id=escape(self.default_parameters['container_id']),
-            instance_id=escape(self.default_parameters['instance_id']),
+            incarnation=escape(str(self.default_parameters["incarnation"])),
+            container_id=escape(self.default_parameters["container_id"]),
+            instance_id=escape(self.default_parameters["instance_id"]),
             health_status=escape(self.provisioning_success_status),
-            health_detail_subsection='')
+            health_detail_subsection="",
+        )
 
     def _get_report_failure_health_document(self):
-        health_detail_subsection = \
+        health_detail_subsection = (
             self._get_formatted_health_detail_subsection_xml_string(
                 health_substatus=escape(self.provisioning_failure_substatus),
                 health_description=escape(
-                    self.provisioning_failure_err_description))
+                    self.provisioning_failure_err_description
+                ),
+            )
+        )
 
         return self._get_formatted_health_report_xml_string(
-            incarnation=escape(str(self.default_parameters['incarnation'])),
-            container_id=escape(self.default_parameters['container_id']),
-            instance_id=escape(self.default_parameters['instance_id']),
+            incarnation=escape(str(self.default_parameters["incarnation"])),
+            container_id=escape(self.default_parameters["container_id"]),
+            instance_id=escape(self.default_parameters["instance_id"]),
             health_status=escape(self.provisioning_not_ready_status),
-            health_detail_subsection=health_detail_subsection)
+            health_detail_subsection=health_detail_subsection,
+        )
 
     def test_send_ready_signal_sends_post_request(self):
         with mock.patch.object(
-                azure_helper.GoalStateHealthReporter,
-                'build_report') as m_build_report:
+            azure_helper.GoalStateHealthReporter, "build_report"
+        ) as m_build_report:
             client = azure_helper.AzureEndpointHttpClient(mock.MagicMock())
             reporter = azure_helper.GoalStateHealthReporter(
                 azure_helper.GoalState(mock.MagicMock(), mock.MagicMock()),
-                client, self.test_azure_endpoint)
+                client,
+                self.test_azure_endpoint,
+            )
             reporter.send_ready_signal()
 
             self.assertEqual(1, self.post.call_count)
@@ -720,73 +778,94 @@ class TestGoalStateHealthReporter(CiTestCase):
                 mock.call(
                     self.test_health_report_url,
                     data=m_build_report.return_value,
-                    extra_headers=self.test_default_headers),
-                self.post.call_args)
+                    extra_headers=self.test_default_headers,
+                ),
+                self.post.call_args,
+            )
 
     def test_send_failure_signal_sends_post_request(self):
         with mock.patch.object(
-                azure_helper.GoalStateHealthReporter,
-                'build_report') as m_build_report:
+            azure_helper.GoalStateHealthReporter, "build_report"
+        ) as m_build_report:
             client = azure_helper.AzureEndpointHttpClient(mock.MagicMock())
             reporter = azure_helper.GoalStateHealthReporter(
                 azure_helper.GoalState(mock.MagicMock(), mock.MagicMock()),
-                client, self.test_azure_endpoint)
+                client,
+                self.test_azure_endpoint,
+            )
             reporter.send_failure_signal(
-                description=self.provisioning_failure_err_description)
+                description=self.provisioning_failure_err_description
+            )
 
             self.assertEqual(1, self.post.call_count)
             self.assertEqual(
                 mock.call(
                     self.test_health_report_url,
                     data=m_build_report.return_value,
-                    extra_headers=self.test_default_headers),
-                self.post.call_args)
+                    extra_headers=self.test_default_headers,
+                ),
+                self.post.call_args,
+            )
 
     def test_build_report_for_ready_signal_health_document(self):
         health_document = self._get_report_ready_health_document()
         reporter = azure_helper.GoalStateHealthReporter(
             azure_helper.GoalState(mock.MagicMock(), mock.MagicMock()),
             azure_helper.AzureEndpointHttpClient(mock.MagicMock()),
-            self.test_azure_endpoint)
+            self.test_azure_endpoint,
+        )
         generated_health_document = reporter.build_report(
-            incarnation=self.default_parameters['incarnation'],
-            container_id=self.default_parameters['container_id'],
-            instance_id=self.default_parameters['instance_id'],
-            status=self.provisioning_success_status)
+            incarnation=self.default_parameters["incarnation"],
+            container_id=self.default_parameters["container_id"],
+            instance_id=self.default_parameters["instance_id"],
+            status=self.provisioning_success_status,
+        )
 
         self.assertEqual(health_document, generated_health_document)
 
         generated_xroot = ElementTree.fromstring(generated_health_document)
         self.assertEqual(
             self._text_from_xpath_in_xroot(
-                generated_xroot, './GoalStateIncarnation'),
-            str(self.default_parameters['incarnation']))
+                generated_xroot, "./GoalStateIncarnation"
+            ),
+            str(self.default_parameters["incarnation"]),
+        )
         self.assertEqual(
             self._text_from_xpath_in_xroot(
-                generated_xroot, './Container/ContainerId'),
-            str(self.default_parameters['container_id']))
+                generated_xroot, "./Container/ContainerId"
+            ),
+            str(self.default_parameters["container_id"]),
+        )
         self.assertEqual(
             self._text_from_xpath_in_xroot(
-                generated_xroot,
-                './Container/RoleInstanceList/Role/InstanceId'),
-            str(self.default_parameters['instance_id']))
+                generated_xroot, "./Container/RoleInstanceList/Role/InstanceId"
+            ),
+            str(self.default_parameters["instance_id"]),
+        )
         self.assertEqual(
             self._text_from_xpath_in_xroot(
                 generated_xroot,
-                './Container/RoleInstanceList/Role/Health/State'),
-            escape(self.provisioning_success_status))
+                "./Container/RoleInstanceList/Role/Health/State",
+            ),
+            escape(self.provisioning_success_status),
+        )
         self.assertIsNone(
             self._text_from_xpath_in_xroot(
                 generated_xroot,
-                './Container/RoleInstanceList/Role/Health/Details'))
+                "./Container/RoleInstanceList/Role/Health/Details",
+            )
+        )
         self.assertIsNone(
             self._text_from_xpath_in_xroot(
                 generated_xroot,
-                './Container/RoleInstanceList/Role/Health/Details/SubStatus'))
+                "./Container/RoleInstanceList/Role/Health/Details/SubStatus",
+            )
+        )
         self.assertIsNone(
             self._text_from_xpath_in_xroot(
                 generated_xroot,
-                './Container/RoleInstanceList/Role/Health/Details/Description')
+                "./Container/RoleInstanceList/Role/Health/Details/Description",
+            )
         )
 
     def test_build_report_for_failure_signal_health_document(self):
@@ -794,120 +873,143 @@ class TestGoalStateHealthReporter(CiTestCase):
         reporter = azure_helper.GoalStateHealthReporter(
             azure_helper.GoalState(mock.MagicMock(), mock.MagicMock()),
             azure_helper.AzureEndpointHttpClient(mock.MagicMock()),
-            self.test_azure_endpoint)
+            self.test_azure_endpoint,
+        )
         generated_health_document = reporter.build_report(
-            incarnation=self.default_parameters['incarnation'],
-            container_id=self.default_parameters['container_id'],
-            instance_id=self.default_parameters['instance_id'],
+            incarnation=self.default_parameters["incarnation"],
+            container_id=self.default_parameters["container_id"],
+            instance_id=self.default_parameters["instance_id"],
             status=self.provisioning_not_ready_status,
             substatus=self.provisioning_failure_substatus,
-            description=self.provisioning_failure_err_description)
+            description=self.provisioning_failure_err_description,
+        )
 
         self.assertEqual(health_document, generated_health_document)
 
         generated_xroot = ElementTree.fromstring(generated_health_document)
         self.assertEqual(
             self._text_from_xpath_in_xroot(
-                generated_xroot, './GoalStateIncarnation'),
-            str(self.default_parameters['incarnation']))
+                generated_xroot, "./GoalStateIncarnation"
+            ),
+            str(self.default_parameters["incarnation"]),
+        )
         self.assertEqual(
             self._text_from_xpath_in_xroot(
-                generated_xroot, './Container/ContainerId'),
-            self.default_parameters['container_id'])
+                generated_xroot, "./Container/ContainerId"
+            ),
+            self.default_parameters["container_id"],
+        )
         self.assertEqual(
             self._text_from_xpath_in_xroot(
-                generated_xroot,
-                './Container/RoleInstanceList/Role/InstanceId'),
-            self.default_parameters['instance_id'])
+                generated_xroot, "./Container/RoleInstanceList/Role/InstanceId"
+            ),
+            self.default_parameters["instance_id"],
+        )
         self.assertEqual(
             self._text_from_xpath_in_xroot(
                 generated_xroot,
-                './Container/RoleInstanceList/Role/Health/State'),
-            escape(self.provisioning_not_ready_status))
+                "./Container/RoleInstanceList/Role/Health/State",
+            ),
+            escape(self.provisioning_not_ready_status),
+        )
         self.assertEqual(
             self._text_from_xpath_in_xroot(
                 generated_xroot,
-                './Container/RoleInstanceList/Role/Health/Details/'
-                'SubStatus'),
-            escape(self.provisioning_failure_substatus))
+                "./Container/RoleInstanceList/Role/Health/Details/SubStatus",
+            ),
+            escape(self.provisioning_failure_substatus),
+        )
         self.assertEqual(
             self._text_from_xpath_in_xroot(
                 generated_xroot,
-                './Container/RoleInstanceList/Role/Health/Details/'
-                'Description'),
-            escape(self.provisioning_failure_err_description))
+                "./Container/RoleInstanceList/Role/Health/Details/Description",
+            ),
+            escape(self.provisioning_failure_err_description),
+        )
 
     def test_send_ready_signal_calls_build_report(self):
         with mock.patch.object(
-            azure_helper.GoalStateHealthReporter, 'build_report'
+            azure_helper.GoalStateHealthReporter, "build_report"
         ) as m_build_report:
             reporter = azure_helper.GoalStateHealthReporter(
                 azure_helper.GoalState(mock.MagicMock(), mock.MagicMock()),
                 azure_helper.AzureEndpointHttpClient(mock.MagicMock()),
-                self.test_azure_endpoint)
+                self.test_azure_endpoint,
+            )
             reporter.send_ready_signal()
 
             self.assertEqual(1, m_build_report.call_count)
             self.assertEqual(
                 mock.call(
-                    incarnation=self.default_parameters['incarnation'],
-                    container_id=self.default_parameters['container_id'],
-                    instance_id=self.default_parameters['instance_id'],
-                    status=self.provisioning_success_status),
-                m_build_report.call_args)
+                    incarnation=self.default_parameters["incarnation"],
+                    container_id=self.default_parameters["container_id"],
+                    instance_id=self.default_parameters["instance_id"],
+                    status=self.provisioning_success_status,
+                ),
+                m_build_report.call_args,
+            )
 
     def test_send_failure_signal_calls_build_report(self):
         with mock.patch.object(
-            azure_helper.GoalStateHealthReporter, 'build_report'
+            azure_helper.GoalStateHealthReporter, "build_report"
         ) as m_build_report:
             reporter = azure_helper.GoalStateHealthReporter(
                 azure_helper.GoalState(mock.MagicMock(), mock.MagicMock()),
                 azure_helper.AzureEndpointHttpClient(mock.MagicMock()),
-                self.test_azure_endpoint)
+                self.test_azure_endpoint,
+            )
             reporter.send_failure_signal(
-                description=self.provisioning_failure_err_description)
+                description=self.provisioning_failure_err_description
+            )
 
             self.assertEqual(1, m_build_report.call_count)
             self.assertEqual(
                 mock.call(
-                    incarnation=self.default_parameters['incarnation'],
-                    container_id=self.default_parameters['container_id'],
-                    instance_id=self.default_parameters['instance_id'],
+                    incarnation=self.default_parameters["incarnation"],
+                    container_id=self.default_parameters["container_id"],
+                    instance_id=self.default_parameters["instance_id"],
                     status=self.provisioning_not_ready_status,
                     substatus=self.provisioning_failure_substatus,
-                    description=self.provisioning_failure_err_description),
-                m_build_report.call_args)
+                    description=self.provisioning_failure_err_description,
+                ),
+                m_build_report.call_args,
+            )
 
     def test_build_report_escapes_chars(self):
-        incarnation = 'jd8\'9*&^<\'A><A[p&o+\"SD()*&&&LKAJSD23'
-        container_id = '&&<\"><><ds8\'9+7&d9a86!@($09asdl;<>'
-        instance_id = 'Opo>>>jas\'&d;[p&fp\"a<<!!@&&'
-        health_status = '&<897\"6&>&aa\'sd!@&!)((*<&>'
-        health_substatus = '&as\"d<<a&s>d<\'^@!5&6<7'
-        health_description = '&&&>!#$\"&&<as\'1!@$d&>><>&\"sd<67<]>>'
-
-        health_detail_subsection = \
+        incarnation = "jd8'9*&^<'A><A[p&o+\"SD()*&&&LKAJSD23"
+        container_id = "&&<\"><><ds8'9+7&d9a86!@($09asdl;<>"
+        instance_id = "Opo>>>jas'&d;[p&fp\"a<<!!@&&"
+        health_status = "&<897\"6&>&aa'sd!@&!)((*<&>"
+        health_substatus = "&as\"d<<a&s>d<'^@!5&6<7"
+        health_description = '&&&>!#$"&&<as\'1!@$d&>><>&"sd<67<]>>'
+
+        health_detail_subsection = (
             self._get_formatted_health_detail_subsection_xml_string(
                 health_substatus=escape(health_substatus),
-                health_description=escape(health_description))
+                health_description=escape(health_description),
+            )
+        )
         health_document = self._get_formatted_health_report_xml_string(
             incarnation=escape(incarnation),
             container_id=escape(container_id),
             instance_id=escape(instance_id),
             health_status=escape(health_status),
-            health_detail_subsection=health_detail_subsection)
+            health_detail_subsection=health_detail_subsection,
+        )
 
         reporter = azure_helper.GoalStateHealthReporter(
             azure_helper.GoalState(mock.MagicMock(), mock.MagicMock()),
             azure_helper.AzureEndpointHttpClient(mock.MagicMock()),
-            self.test_azure_endpoint)
+            self.test_azure_endpoint,
+        )
         generated_health_document = reporter.build_report(
             incarnation=incarnation,
             container_id=container_id,
             instance_id=instance_id,
             status=health_status,
             substatus=health_substatus,
-            description=health_description)
+            description=health_description,
+        )
 
         self.assertEqual(health_document, generated_health_document)
 
@@ -915,26 +1017,31 @@ class TestGoalStateHealthReporter(CiTestCase):
         reporter = azure_helper.GoalStateHealthReporter(
             azure_helper.GoalState(mock.MagicMock(), mock.MagicMock()),
             azure_helper.AzureEndpointHttpClient(mock.MagicMock()),
-            self.test_azure_endpoint)
-        long_err_msg = 'a9&ea8>>>e as1< d\"q2*&(^%\'a=5<' * 100
+            self.test_azure_endpoint,
+        )
+        long_err_msg = "a9&ea8>>>e as1< d\"q2*&(^%'a=5<" * 100
         generated_health_document = reporter.build_report(
-            incarnation=self.default_parameters['incarnation'],
-            container_id=self.default_parameters['container_id'],
-            instance_id=self.default_parameters['instance_id'],
+            incarnation=self.default_parameters["incarnation"],
+            container_id=self.default_parameters["container_id"],
+            instance_id=self.default_parameters["instance_id"],
             status=self.provisioning_not_ready_status,
             substatus=self.provisioning_failure_substatus,
-            description=long_err_msg)
+            description=long_err_msg,
+        )
 
         generated_xroot = ElementTree.fromstring(generated_health_document)
         generated_health_report_description = self._text_from_xpath_in_xroot(
             generated_xroot,
-            './Container/RoleInstanceList/Role/Health/Details/Description')
+            "./Container/RoleInstanceList/Role/Health/Details/Description",
+        )
         self.assertEqual(
             len(unescape(generated_health_report_description)),
-            HEALTH_REPORT_DESCRIPTION_TRIM_LEN)
+            HEALTH_REPORT_DESCRIPTION_TRIM_LEN,
+        )
 
     def test_trim_description_then_escape_conforms_to_len_limits_worst_case(
-            self):
+        self,
+    ):
         """When unescaped characters are XML-escaped, the length increases.
         Char      Escape String
         <         &lt;
@@ -963,46 +1070,53 @@ class TestGoalStateHealthReporter(CiTestCase):
         reporter = azure_helper.GoalStateHealthReporter(
             azure_helper.GoalState(mock.MagicMock(), mock.MagicMock()),
             azure_helper.AzureEndpointHttpClient(mock.MagicMock()),
-            self.test_azure_endpoint)
-        long_err_msg = '\'\"' * 10000
+            self.test_azure_endpoint,
+        )
+        long_err_msg = "'\"" * 10000
         generated_health_document = reporter.build_report(
-            incarnation=self.default_parameters['incarnation'],
-            container_id=self.default_parameters['container_id'],
-            instance_id=self.default_parameters['instance_id'],
+            incarnation=self.default_parameters["incarnation"],
+            container_id=self.default_parameters["container_id"],
+            instance_id=self.default_parameters["instance_id"],
             status=self.provisioning_not_ready_status,
             substatus=self.provisioning_failure_substatus,
-            description=long_err_msg)
+            description=long_err_msg,
+        )
 
         generated_xroot = ElementTree.fromstring(generated_health_document)
         generated_health_report_description = self._text_from_xpath_in_xroot(
             generated_xroot,
-            './Container/RoleInstanceList/Role/Health/Details/Description')
+            "./Container/RoleInstanceList/Role/Health/Details/Description",
+        )
         # The escaped description string should be less than
         # the Azure platform limit for the escaped description string.
         self.assertLessEqual(len(generated_health_report_description), 4096)
 
 
 class TestWALinuxAgentShim(CiTestCase):
-
     def setUp(self):
         super(TestWALinuxAgentShim, self).setUp()
         patches = ExitStack()
         self.addCleanup(patches.close)
 
         self.AzureEndpointHttpClient = patches.enter_context(
-            mock.patch.object(azure_helper, 'AzureEndpointHttpClient'))
+            mock.patch.object(azure_helper, "AzureEndpointHttpClient")
+        )
         self.find_endpoint = patches.enter_context(
-            mock.patch.object(wa_shim, 'find_endpoint'))
+            mock.patch.object(wa_shim, "find_endpoint")
+        )
         self.GoalState = patches.enter_context(
-            mock.patch.object(azure_helper, 'GoalState'))
+            mock.patch.object(azure_helper, "GoalState")
+        )
         self.OpenSSLManager = patches.enter_context(
-            mock.patch.object(azure_helper, 'OpenSSLManager', autospec=True))
+            mock.patch.object(azure_helper, "OpenSSLManager", autospec=True)
+        )
         patches.enter_context(
-            mock.patch.object(azure_helper.time, 'sleep', mock.MagicMock()))
+            mock.patch.object(azure_helper.time, "sleep", mock.MagicMock())
+        )
 
-        self.test_incarnation = 'TestIncarnation'
-        self.test_container_id = 'TestContainerId'
-        self.test_instance_id = 'TestInstanceId'
+        self.test_incarnation = "TestIncarnation"
+        self.test_container_id = "TestContainerId"
+        self.test_instance_id = "TestInstanceId"
         self.GoalState.return_value.incarnation = self.test_incarnation
         self.GoalState.return_value.container_id = self.test_container_id
         self.GoalState.return_value.instance_id = self.test_instance_id
@@ -1010,7 +1124,7 @@ class TestWALinuxAgentShim(CiTestCase):
     def test_eject_iso_is_called(self):
         shim = wa_shim()
         with mock.patch.object(
-            shim, 'eject_iso', autospec=True
+            shim, "eject_iso", autospec=True
         ) as m_eject_iso:
             shim.register_with_azure_and_fetch_data(iso_dev="/dev/sr0")
             m_eject_iso.assert_called_once_with("/dev/sr0")
@@ -1019,102 +1133,113 @@ class TestWALinuxAgentShim(CiTestCase):
         shim = wa_shim()
         shim.register_with_azure_and_fetch_data()
         self.assertEqual(
-            [mock.call(None)],
-            self.AzureEndpointHttpClient.call_args_list)
+            [mock.call(None)], self.AzureEndpointHttpClient.call_args_list
+        )
 
     def test_http_client_does_not_use_certificate_for_report_failure(self):
         shim = wa_shim()
-        shim.register_with_azure_and_report_failure(description='TestDesc')
+        shim.register_with_azure_and_report_failure(description="TestDesc")
         self.assertEqual(
-            [mock.call(None)],
-            self.AzureEndpointHttpClient.call_args_list)
+            [mock.call(None)], self.AzureEndpointHttpClient.call_args_list
+        )
 
     def test_correct_url_used_for_goalstate_during_report_ready(self):
-        self.find_endpoint.return_value = 'test_endpoint'
+        self.find_endpoint.return_value = "test_endpoint"
         shim = wa_shim()
         shim.register_with_azure_and_fetch_data()
         m_get = self.AzureEndpointHttpClient.return_value.get
         self.assertEqual(
-            [mock.call('http://test_endpoint/machine/?comp=goalstate')],
-            m_get.call_args_list)
+            [mock.call("http://test_endpoint/machine/?comp=goalstate")],
+            m_get.call_args_list,
+        )
         self.assertEqual(
-            [mock.call(
-                m_get.return_value.contents,
-                self.AzureEndpointHttpClient.return_value,
-                False
-            )],
-            self.GoalState.call_args_list)
+            [
+                mock.call(
+                    m_get.return_value.contents,
+                    self.AzureEndpointHttpClient.return_value,
+                    False,
+                )
+            ],
+            self.GoalState.call_args_list,
+        )
 
     def test_correct_url_used_for_goalstate_during_report_failure(self):
-        self.find_endpoint.return_value = 'test_endpoint'
+        self.find_endpoint.return_value = "test_endpoint"
         shim = wa_shim()
-        shim.register_with_azure_and_report_failure(description='TestDesc')
+        shim.register_with_azure_and_report_failure(description="TestDesc")
         m_get = self.AzureEndpointHttpClient.return_value.get
         self.assertEqual(
-            [mock.call('http://test_endpoint/machine/?comp=goalstate')],
-            m_get.call_args_list)
+            [mock.call("http://test_endpoint/machine/?comp=goalstate")],
+            m_get.call_args_list,
+        )
         self.assertEqual(
-            [mock.call(
-                m_get.return_value.contents,
-                self.AzureEndpointHttpClient.return_value,
-                False
-            )],
-            self.GoalState.call_args_list)
+            [
+                mock.call(
+                    m_get.return_value.contents,
+                    self.AzureEndpointHttpClient.return_value,
+                    False,
+                )
+            ],
+            self.GoalState.call_args_list,
+        )
 
     def test_certificates_used_to_determine_public_keys(self):
         # if register_with_azure_and_fetch_data() isn't passed some info about
         # the user's public keys, there's no point in even trying to parse the
         # certificates
         shim = wa_shim()
-        mypk = [{'fingerprint': 'fp1', 'path': 'path1'},
-                {'fingerprint': 'fp3', 'path': 'path3', 'value': ''}]
-        certs = {'fp1': 'expected-key',
-                 'fp2': 'should-not-be-found',
-                 'fp3': 'expected-no-value-key',
-                 }
+        mypk = [
+            {"fingerprint": "fp1", "path": "path1"},
+            {"fingerprint": "fp3", "path": "path3", "value": ""},
+        ]
+        certs = {
+            "fp1": "expected-key",
+            "fp2": "should-not-be-found",
+            "fp3": "expected-no-value-key",
+        }
         sslmgr = self.OpenSSLManager.return_value
         sslmgr.parse_certificates.return_value = certs
         data = shim.register_with_azure_and_fetch_data(pubkey_info=mypk)
         self.assertEqual(
             [mock.call(self.GoalState.return_value.certificates_xml)],
-            sslmgr.parse_certificates.call_args_list)
-        self.assertIn('expected-key', data['public-keys'])
-        self.assertIn('expected-no-value-key', data['public-keys'])
-        self.assertNotIn('should-not-be-found', data['public-keys'])
+            sslmgr.parse_certificates.call_args_list,
+        )
+        self.assertIn("expected-key", data["public-keys"])
+        self.assertIn("expected-no-value-key", data["public-keys"])
+        self.assertNotIn("should-not-be-found", data["public-keys"])
 
     def test_absent_certificates_produces_empty_public_keys(self):
-        mypk = [{'fingerprint': 'fp1', 'path': 'path1'}]
+        mypk = [{"fingerprint": "fp1", "path": "path1"}]
         self.GoalState.return_value.certificates_xml = None
         shim = wa_shim()
         data = shim.register_with_azure_and_fetch_data(pubkey_info=mypk)
-        self.assertEqual([], data['public-keys'])
+        self.assertEqual([], data["public-keys"])
 
     def test_correct_url_used_for_report_ready(self):
-        self.find_endpoint.return_value = 'test_endpoint'
+        self.find_endpoint.return_value = "test_endpoint"
         shim = wa_shim()
         shim.register_with_azure_and_fetch_data()
-        expected_url = 'http://test_endpoint/machine?comp=health'
+        expected_url = "http://test_endpoint/machine?comp=health"
         self.assertEqual(
             [mock.call(expected_url, data=mock.ANY, extra_headers=mock.ANY)],
-            self.AzureEndpointHttpClient.return_value.post
-                .call_args_list)
+            self.AzureEndpointHttpClient.return_value.post.call_args_list,
+        )
 
     def test_correct_url_used_for_report_failure(self):
-        self.find_endpoint.return_value = 'test_endpoint'
+        self.find_endpoint.return_value = "test_endpoint"
         shim = wa_shim()
-        shim.register_with_azure_and_report_failure(description='TestDesc')
-        expected_url = 'http://test_endpoint/machine?comp=health'
+        shim.register_with_azure_and_report_failure(description="TestDesc")
+        expected_url = "http://test_endpoint/machine?comp=health"
         self.assertEqual(
             [mock.call(expected_url, data=mock.ANY, extra_headers=mock.ANY)],
-            self.AzureEndpointHttpClient.return_value.post
-                .call_args_list)
+            self.AzureEndpointHttpClient.return_value.post.call_args_list,
+        )
 
     def test_goal_state_values_used_for_report_ready(self):
         shim = wa_shim()
         shim.register_with_azure_and_fetch_data()
         posted_document = (
-            self.AzureEndpointHttpClient.return_value.post
-                .call_args[1]['data']
+            self.AzureEndpointHttpClient.return_value.post.call_args[1]["data"]
         )
         self.assertIn(self.test_incarnation, posted_document)
         self.assertIn(self.test_container_id, posted_document)
@@ -1122,10 +1247,9 @@ class TestWALinuxAgentShim(CiTestCase):
 
     def test_goal_state_values_used_for_report_failure(self):
         shim = wa_shim()
-        shim.register_with_azure_and_report_failure(description='TestDesc')
+        shim.register_with_azure_and_report_failure(description="TestDesc")
         posted_document = (
-            self.AzureEndpointHttpClient.return_value.post
-                .call_args[1]['data']
+            self.AzureEndpointHttpClient.return_value.post.call_args[1]["data"]
         )
         self.assertIn(self.test_incarnation, posted_document)
         self.assertIn(self.test_container_id, posted_document)
@@ -1138,57 +1262,66 @@ class TestWALinuxAgentShim(CiTestCase):
             incarnation=escape(self.test_incarnation),
             container_id=escape(self.test_container_id),
             instance_id=escape(self.test_instance_id),
-            health_status=escape('Ready'),
-            health_detail_subsection='')
+            health_status=escape("Ready"),
+            health_detail_subsection="",
+        )
         posted_document = (
-            self.AzureEndpointHttpClient.return_value.post
-                .call_args[1]['data'])
+            self.AzureEndpointHttpClient.return_value.post.call_args[1]["data"]
+        )
         self.assertEqual(health_document, posted_document)
 
     def test_xml_elems_in_report_failure_post(self):
         shim = wa_shim()
-        shim.register_with_azure_and_report_failure(description='TestDesc')
+        shim.register_with_azure_and_report_failure(description="TestDesc")
         health_document = HEALTH_REPORT_XML_TEMPLATE.format(
             incarnation=escape(self.test_incarnation),
             container_id=escape(self.test_container_id),
             instance_id=escape(self.test_instance_id),
-            health_status=escape('NotReady'),
-            health_detail_subsection=HEALTH_DETAIL_SUBSECTION_XML_TEMPLATE
-            .format(
-                health_substatus=escape('ProvisioningFailed'),
-                health_description=escape('TestDesc')))
+            health_status=escape("NotReady"),
+            health_detail_subsection=(
+                HEALTH_DETAIL_SUBSECTION_XML_TEMPLATE.format(
+                    health_substatus=escape("ProvisioningFailed"),
+                    health_description=escape("TestDesc"),
+                )
+            ),
+        )
         posted_document = (
-            self.AzureEndpointHttpClient.return_value.post
-                .call_args[1]['data'])
+            self.AzureEndpointHttpClient.return_value.post.call_args[1]["data"]
+        )
         self.assertEqual(health_document, posted_document)
 
-    @mock.patch.object(azure_helper, 'GoalStateHealthReporter', autospec=True)
+    @mock.patch.object(azure_helper, "GoalStateHealthReporter", autospec=True)
     def test_register_with_azure_and_fetch_data_calls_send_ready_signal(
-            self, m_goal_state_health_reporter):
+        self, m_goal_state_health_reporter
+    ):
         shim = wa_shim()
         shim.register_with_azure_and_fetch_data()
         self.assertEqual(
             1,
-            m_goal_state_health_reporter.return_value.send_ready_signal
-            .call_count)
+            m_goal_state_health_reporter.return_value.send_ready_signal.call_count,  # noqa: E501
+        )
 
-    @mock.patch.object(azure_helper, 'GoalStateHealthReporter', autospec=True)
+    @mock.patch.object(azure_helper, "GoalStateHealthReporter", autospec=True)
     def test_register_with_azure_and_report_failure_calls_send_failure_signal(
-            self, m_goal_state_health_reporter):
+        self, m_goal_state_health_reporter
+    ):
         shim = wa_shim()
-        shim.register_with_azure_and_report_failure(description='TestDesc')
-        m_goal_state_health_reporter.return_value.send_failure_signal \
-            .assert_called_once_with(description='TestDesc')
+        shim.register_with_azure_and_report_failure(description="TestDesc")
+        m_goal_state_health_reporter.return_value.send_failure_signal.assert_called_once_with(  # noqa: E501
+            description="TestDesc"
+        )
 
     def test_register_with_azure_and_report_failure_does_not_need_certificates(
-            self):
+        self,
+    ):
         shim = wa_shim()
         with mock.patch.object(
-            shim, '_fetch_goal_state_from_azure', autospec=True
+            shim, "_fetch_goal_state_from_azure", autospec=True
         ) as m_fetch_goal_state_from_azure:
-            shim.register_with_azure_and_report_failure(description='TestDesc')
+            shim.register_with_azure_and_report_failure(description="TestDesc")
             m_fetch_goal_state_from_azure.assert_called_once_with(
-                need_certificate=False)
+                need_certificate=False
+            )
 
     def test_clean_up_can_be_called_at_any_time(self):
         shim = wa_shim()
@@ -1197,7 +1330,7 @@ class TestWALinuxAgentShim(CiTestCase):
     def test_openssl_manager_not_instantiated_by_shim_report_status(self):
         shim = wa_shim()
         shim.register_with_azure_and_fetch_data()
-        shim.register_with_azure_and_report_failure(description='TestDesc')
+        shim.register_with_azure_and_report_failure(description="TestDesc")
         shim.clean_up()
         self.OpenSSLManager.assert_not_called()
 
@@ -1209,178 +1342,204 @@ class TestWALinuxAgentShim(CiTestCase):
 
     def test_clean_up_after_report_failure(self):
         shim = wa_shim()
-        shim.register_with_azure_and_report_failure(description='TestDesc')
+        shim.register_with_azure_and_report_failure(description="TestDesc")
         shim.clean_up()
         self.OpenSSLManager.return_value.clean_up.assert_not_called()
 
     def test_fetch_goalstate_during_report_ready_raises_exc_on_get_exc(self):
-        self.AzureEndpointHttpClient.return_value.get \
-            .side_effect = SentinelException
+        self.AzureEndpointHttpClient.return_value.get.side_effect = (
+            SentinelException
+        )
         shim = wa_shim()
-        self.assertRaises(SentinelException,
-                          shim.register_with_azure_and_fetch_data)
+        self.assertRaises(
+            SentinelException, shim.register_with_azure_and_fetch_data
+        )
 
     def test_fetch_goalstate_during_report_failure_raises_exc_on_get_exc(self):
-        self.AzureEndpointHttpClient.return_value.get \
-            .side_effect = SentinelException
+        self.AzureEndpointHttpClient.return_value.get.side_effect = (
+            SentinelException
+        )
         shim = wa_shim()
-        self.assertRaises(SentinelException,
-                          shim.register_with_azure_and_report_failure,
-                          description='TestDesc')
+        self.assertRaises(
+            SentinelException,
+            shim.register_with_azure_and_report_failure,
+            description="TestDesc",
+        )
 
     def test_fetch_goalstate_during_report_ready_raises_exc_on_parse_exc(self):
         self.GoalState.side_effect = SentinelException
         shim = wa_shim()
-        self.assertRaises(SentinelException,
-                          shim.register_with_azure_and_fetch_data)
+        self.assertRaises(
+            SentinelException, shim.register_with_azure_and_fetch_data
+        )
 
     def test_fetch_goalstate_during_report_failure_raises_exc_on_parse_exc(
-            self):
+        self,
+    ):
         self.GoalState.side_effect = SentinelException
         shim = wa_shim()
-        self.assertRaises(SentinelException,
-                          shim.register_with_azure_and_report_failure,
-                          description='TestDesc')
+        self.assertRaises(
+            SentinelException,
+            shim.register_with_azure_and_report_failure,
+            description="TestDesc",
+        )
 
     def test_failure_to_send_report_ready_health_doc_bubbles_up(self):
-        self.AzureEndpointHttpClient.return_value.post \
-            .side_effect = SentinelException
+        self.AzureEndpointHttpClient.return_value.post.side_effect = (
+            SentinelException
+        )
         shim = wa_shim()
-        self.assertRaises(SentinelException,
-                          shim.register_with_azure_and_fetch_data)
+        self.assertRaises(
+            SentinelException, shim.register_with_azure_and_fetch_data
+        )
 
     def test_failure_to_send_report_failure_health_doc_bubbles_up(self):
-        self.AzureEndpointHttpClient.return_value.post \
-            .side_effect = SentinelException
+        self.AzureEndpointHttpClient.return_value.post.side_effect = (
+            SentinelException
+        )
         shim = wa_shim()
-        self.assertRaises(SentinelException,
-                          shim.register_with_azure_and_report_failure,
-                          description='TestDesc')
+        self.assertRaises(
+            SentinelException,
+            shim.register_with_azure_and_report_failure,
+            description="TestDesc",
+        )
 
 
 class TestGetMetadataGoalStateXMLAndReportReadyToFabric(CiTestCase):
-
     def setUp(self):
         super(TestGetMetadataGoalStateXMLAndReportReadyToFabric, self).setUp()
         patches = ExitStack()
         self.addCleanup(patches.close)
 
         self.m_shim = patches.enter_context(
-            mock.patch.object(azure_helper, 'WALinuxAgentShim'))
+            mock.patch.object(azure_helper, "WALinuxAgentShim")
+        )
 
     def test_data_from_shim_returned(self):
         ret = azure_helper.get_metadata_from_fabric()
         self.assertEqual(
-            self.m_shim.return_value.register_with_azure_and_fetch_data
-                .return_value,
-            ret)
+            self.m_shim.return_value.register_with_azure_and_fetch_data.return_value,  # noqa: E501
+            ret,
+        )
 
     def test_success_calls_clean_up(self):
         azure_helper.get_metadata_from_fabric()
         self.assertEqual(1, self.m_shim.return_value.clean_up.call_count)
 
-    def test_failure_in_registration_propagates_exc_and_calls_clean_up(
-            self):
-        self.m_shim.return_value.register_with_azure_and_fetch_data \
-            .side_effect = SentinelException
-        self.assertRaises(SentinelException,
-                          azure_helper.get_metadata_from_fabric)
+    def test_failure_in_registration_propagates_exc_and_calls_clean_up(self):
+        self.m_shim.return_value.register_with_azure_and_fetch_data.side_effect = (  # noqa: E501
+            SentinelException
+        )
+        self.assertRaises(
+            SentinelException, azure_helper.get_metadata_from_fabric
+        )
         self.assertEqual(1, self.m_shim.return_value.clean_up.call_count)
 
     def test_calls_shim_register_with_azure_and_fetch_data(self):
         m_pubkey_info = mock.MagicMock()
         azure_helper.get_metadata_from_fabric(
-            pubkey_info=m_pubkey_info, iso_dev="/dev/sr0")
+            pubkey_info=m_pubkey_info, iso_dev="/dev/sr0"
+        )
         self.assertEqual(
             1,
-            self.m_shim.return_value
-                .register_with_azure_and_fetch_data.call_count)
+            self.m_shim.return_value.register_with_azure_and_fetch_data.call_count,  # noqa: E501
+        )
         self.assertEqual(
             mock.call(iso_dev="/dev/sr0", pubkey_info=m_pubkey_info),
-            self.m_shim.return_value
-                .register_with_azure_and_fetch_data.call_args)
+            self.m_shim.return_value.register_with_azure_and_fetch_data.call_args,  # noqa: E501
+        )
 
     def test_instantiates_shim_with_kwargs(self):
         m_fallback_lease_file = mock.MagicMock()
         m_dhcp_options = mock.MagicMock()
         azure_helper.get_metadata_from_fabric(
-            fallback_lease_file=m_fallback_lease_file,
-            dhcp_opts=m_dhcp_options)
+            fallback_lease_file=m_fallback_lease_file, dhcp_opts=m_dhcp_options
+        )
         self.assertEqual(1, self.m_shim.call_count)
         self.assertEqual(
             mock.call(
                 fallback_lease_file=m_fallback_lease_file,
-                dhcp_options=m_dhcp_options),
-            self.m_shim.call_args)
+                dhcp_options=m_dhcp_options,
+            ),
+            self.m_shim.call_args,
+        )
 
 
 class TestGetMetadataGoalStateXMLAndReportFailureToFabric(CiTestCase):
-
     def setUp(self):
         super(
-            TestGetMetadataGoalStateXMLAndReportFailureToFabric, self).setUp()
+            TestGetMetadataGoalStateXMLAndReportFailureToFabric, self
+        ).setUp()
         patches = ExitStack()
         self.addCleanup(patches.close)
 
         self.m_shim = patches.enter_context(
-            mock.patch.object(azure_helper, 'WALinuxAgentShim'))
+            mock.patch.object(azure_helper, "WALinuxAgentShim")
+        )
 
     def test_success_calls_clean_up(self):
         azure_helper.report_failure_to_fabric()
-        self.assertEqual(
-            1,
-            self.m_shim.return_value.clean_up.call_count)
+        self.assertEqual(1, self.m_shim.return_value.clean_up.call_count)
 
     def test_failure_in_shim_report_failure_propagates_exc_and_calls_clean_up(
-            self):
-        self.m_shim.return_value.register_with_azure_and_report_failure \
-            .side_effect = SentinelException
-        self.assertRaises(SentinelException,
-                          azure_helper.report_failure_to_fabric)
-        self.assertEqual(
-            1,
-            self.m_shim.return_value.clean_up.call_count)
+        self,
+    ):
+        self.m_shim.return_value.register_with_azure_and_report_failure.side_effect = (  # noqa: E501
+            SentinelException
+        )
+        self.assertRaises(
+            SentinelException, azure_helper.report_failure_to_fabric
+        )
+        self.assertEqual(1, self.m_shim.return_value.clean_up.call_count)
 
     def test_report_failure_to_fabric_with_desc_calls_shim_report_failure(
-            self):
-        azure_helper.report_failure_to_fabric(description='TestDesc')
-        self.m_shim.return_value.register_with_azure_and_report_failure \
-            .assert_called_once_with(description='TestDesc')
+        self,
+    ):
+        azure_helper.report_failure_to_fabric(description="TestDesc")
+        self.m_shim.return_value.register_with_azure_and_report_failure.assert_called_once_with(  # noqa: E501
+            description="TestDesc"
+        )
 
     def test_report_failure_to_fabric_with_no_desc_calls_shim_report_failure(
-            self):
+        self,
+    ):
         azure_helper.report_failure_to_fabric()
         # default err message description should be shown to the user
         # if no description is passed in
-        self.m_shim.return_value.register_with_azure_and_report_failure \
-            .assert_called_once_with(
-                description=azure_helper
-                .DEFAULT_REPORT_FAILURE_USER_VISIBLE_MESSAGE)
+        self.m_shim.return_value.register_with_azure_and_report_failure.assert_called_once_with(  # noqa: E501
+            description=(
+                azure_helper.DEFAULT_REPORT_FAILURE_USER_VISIBLE_MESSAGE
+            )
+        )
 
     def test_report_failure_to_fabric_empty_desc_calls_shim_report_failure(
-            self):
-        azure_helper.report_failure_to_fabric(description='')
+        self,
+    ):
+        azure_helper.report_failure_to_fabric(description="")
         # default err message description should be shown to the user
         # if an empty description is passed in
-        self.m_shim.return_value.register_with_azure_and_report_failure \
-            .assert_called_once_with(
-                description=azure_helper
-                .DEFAULT_REPORT_FAILURE_USER_VISIBLE_MESSAGE)
+        self.m_shim.return_value.register_with_azure_and_report_failure.assert_called_once_with(  # noqa: E501
+            description=(
+                azure_helper.DEFAULT_REPORT_FAILURE_USER_VISIBLE_MESSAGE
+            )
+        )
 
     def test_instantiates_shim_with_kwargs(self):
         m_fallback_lease_file = mock.MagicMock()
         m_dhcp_options = mock.MagicMock()
         azure_helper.report_failure_to_fabric(
-            fallback_lease_file=m_fallback_lease_file,
-            dhcp_opts=m_dhcp_options)
+            fallback_lease_file=m_fallback_lease_file, dhcp_opts=m_dhcp_options
+        )
         self.m_shim.assert_called_once_with(
             fallback_lease_file=m_fallback_lease_file,
-            dhcp_options=m_dhcp_options)
+            dhcp_options=m_dhcp_options,
+        )
 
 
 class TestExtractIpAddressFromNetworkd(CiTestCase):
 
-    azure_lease = dedent("""\
+    azure_lease = dedent(
+        """\
     # This is private data. Do not parse.
     ADDRESS=10.132.0.5
     NETMASK=255.255.255.255
@@ -1399,7 +1558,8 @@ class TestExtractIpAddressFromNetworkd(CiTestCase):
     ROUTES=10.132.0.1/32,0.0.0.0 0.0.0.0/0,10.132.0.1
     CLIENTID=ff405663a200020000ab11332859494d7a8b4c
     OPTION_245=624c3620
-    """)
+    """
+    )
 
     def setUp(self):
         super(TestExtractIpAddressFromNetworkd, self).setUp()
@@ -1408,21 +1568,25 @@ class TestExtractIpAddressFromNetworkd(CiTestCase):
     def test_no_valid_leases_is_none(self):
         """No valid leases should return None."""
         self.assertIsNone(
-            wa_shim._networkd_get_value_from_leases(self.lease_d))
+            wa_shim._networkd_get_value_from_leases(self.lease_d)
+        )
 
     def test_option_245_is_found_in_single(self):
         """A single valid lease with 245 option should return it."""
-        populate_dir(self.lease_d, {'9': self.azure_lease})
+        populate_dir(self.lease_d, {"9": self.azure_lease})
         self.assertEqual(
-            '624c3620', wa_shim._networkd_get_value_from_leases(self.lease_d))
+            "624c3620", wa_shim._networkd_get_value_from_leases(self.lease_d)
+        )
 
     def test_option_245_not_found_returns_None(self):
         """A valid lease, but no option 245 should return None."""
         populate_dir(
             self.lease_d,
-            {'9': self.azure_lease.replace("OPTION_245", "OPTION_999")})
+            {"9": self.azure_lease.replace("OPTION_245", "OPTION_999")},
+        )
         self.assertIsNone(
-            wa_shim._networkd_get_value_from_leases(self.lease_d))
+            wa_shim._networkd_get_value_from_leases(self.lease_d)
+        )
 
     def test_multiple_returns_first(self):
         """Somewhat arbitrarily return the first address when multiple.
@@ -1432,10 +1596,14 @@ class TestExtractIpAddressFromNetworkd(CiTestCase):
         myval = "624c3601"
         populate_dir(
             self.lease_d,
-            {'9': self.azure_lease,
-             '2': self.azure_lease.replace("624c3620", myval)})
+            {
+                "9": self.azure_lease,
+                "2": self.azure_lease.replace("624c3620", myval),
+            },
+        )
         self.assertEqual(
-            myval, wa_shim._networkd_get_value_from_leases(self.lease_d))
+            myval, wa_shim._networkd_get_value_from_leases(self.lease_d)
+        )
 
 
 # vi: ts=4 expandtab