Use dd_run_check in tests (DataDog#10658)

kkourtchikov · Nov 23, 2021 · 99bffb1 · 99bffb1
1 parent a620547
commit 99bffb1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cloud_foundry_api/tests/test_cloud_foundry_api.py b/cloud_foundry_api/tests/test_cloud_foundry_api.py
@@ -75,10 +75,10 @@ def test_init_bad_instance(_):
 @mock.patch.object(CloudFoundryApiCheck, "discover_api", return_value=("v3", "uaa_url"))
 @mock.patch.object(CloudFoundryApiCheck, "get_orgs", return_value={"org_id": "org_name"})
 @mock.patch.object(CloudFoundryApiCheck, "get_spaces", return_value={"space_id": "space_name"})
-def test_check(_, __, ___, aggregator, instance, dd_events):
+def test_check(_, __, ___, aggregator, instance, dd_events, dd_run_check):
     with mock.patch.object(CloudFoundryApiCheck, "get_events", return_value=dd_events):
         check = CloudFoundryApiCheck('cloud_foundry_api', {}, [instance])
-        check.check({})
+        dd_run_check(check)
 
     aggregator.assert_metric(
         "cloud_foundry_api.events.count",