Fix exception handling ixload

[yardstick.git] / yardstick / tests / unit / network_services / traffic_profile / test_prox_profile.py

diff --git a/yardstick/tests/unit/network_services/traffic_profile/test_prox_profile.py b/yardstick/tests/unit/network_services/traffic_profile/test_prox_profile.py
index e466305..11bee03 100644 (file)
--- a/yardstick/tests/unit/network_services/traffic_profile/test_prox_profile.py
+++ b/yardstick/tests/unit/network_services/traffic_profile/test_prox_profile.py
@@ -100,13 +100,13 @@ class TestProxProfile(unittest.TestCase):
 
         profile = ProxProfile(tp_config)
 
-        self.assertFalse(profile.done)
+        self.assertFalse(profile.done.is_set())
         for _ in packet_sizes:
             with self.assertRaises(NotImplementedError):
                 profile.execute_traffic(traffic_generator)
 
         self.assertIsNone(profile.execute_traffic(traffic_generator))
-        self.assertTrue(profile.done)
+        self.assertTrue(profile.done.is_set())
 
     def test_bounds_iterator(self):
         tp_config = {
@@ -124,5 +124,5 @@ class TestProxProfile(unittest.TestCase):
         for _ in profile.bounds_iterator(mock_logger):
             pass
 
-        self.assertEqual(mock_logger.debug.call_count, 1)
+        mock_logger.debug.assert_called_once()
         self.assertEqual(mock_logger.info.call_count, 10)