X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=yardstick%2Ftests%2Funit%2Fbenchmark%2Fscenarios%2Flib%2Ftest_delete_router_gateway.py;h=0c9cf7c17a143db413d62100a6c76ab6f1ced86d;hb=4fe8e4e120afcd7585d1e81320d0e84d3a18f85d;hp=3cfc4ed21e6b35c6212f8e5baa7feada54836874;hpb=14a0f0b13e37cf0a65603915b7c75b86fd4a4383;p=yardstick.git

diff --git a/yardstick/tests/unit/benchmark/scenarios/lib/test_delete_router_gateway.py b/yardstick/tests/unit/benchmark/scenarios/lib/test_delete_router_gateway.py
index 3cfc4ed21..0c9cf7c17 100644
--- a/yardstick/tests/unit/benchmark/scenarios/lib/test_delete_router_gateway.py
+++ b/yardstick/tests/unit/benchmark/scenarios/lib/test_delete_router_gateway.py
@@ -23,13 +23,5 @@ class DeleteRouterGatewayTestCase(unittest.TestCase):
         args = {"options": options}
         obj = DeleteRouterGateway(args, {})
         obj.run({})
-        self.assertTrue(mock_get_neutron_client.called)
-        self.assertTrue(mock_remove_gateway_router.called)
-
-
-def main():
-    unittest.main()
-
-
-if __name__ == '__main__':
-    main()
+        mock_get_neutron_client.assert_called_once()
+        mock_remove_gateway_router.assert_called_once()