Fix flaky ExtensionCleanupTest class (#1898)

tests/ga/test_extension.py

@@ -86,7 +86,7 @@ class TestExtensionCleanup(AgentTestCase):
 
     def setUp(self):
         AgentTestCase.setUp(self)
-        self.mock_sleep = patch("time.sleep", lambda *_: mock_sleep(0.0001))
+        self.mock_sleep = patch("time.sleep", lambda *_: mock_sleep(0.01))
         self.mock_sleep.start()
 
     def tearDown(self):
@@ -110,10 +110,12 @@ def _is_extension_dir(path):
     def _assert_ext_handler_status(self, aggregate_status, expected_status, version, expected_ext_handler_count=0):
         self.assertIsNotNone(aggregate_status, "Aggregate status should not be None")
         handler_statuses = aggregate_status['aggregateStatus']['handlerAggregateStatus']
-        self.assertEqual(expected_ext_handler_count, len(handler_statuses))
+        self.assertEqual(expected_ext_handler_count, len(handler_statuses),
+                         "All ExtensionHandlers: {0}".format(handler_statuses))
         for ext_handler_status in handler_statuses:
-            self.assertEquals(expected_status, ext_handler_status['status'])
-            self.assertEquals(version, ext_handler_status['handlerVersion'])
+            debug_info = "ExtensionHandler: {0}".format(ext_handler_status)
+            self.assertEquals(expected_status, ext_handler_status['status'], debug_info)
+            self.assertEquals(version, ext_handler_status['handlerVersion'], debug_info)
         return
 
     @contextlib.contextmanager