Use call_count == 1

apache · Apr 18, 2024 · c181728 · c181728
1 parent 0d25646
commit c181728
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/providers/apache/druid/hooks/test_druid.py b/tests/providers/apache/druid/hooks/test_druid.py
@@ -122,13 +122,13 @@ def test_submit_with_false_ssl_arg(self, requests_mock):
         with pytest.raises(AirflowException):
             self.db_hook.submit_indexing_job("Long json file")
 
-        assert task_post.called_once
+        assert task_post.call_count == 1
         assert False is task_post.request_history[0].verify
 
         assert status_check.call_count > 1
         assert False is status_check.request_history[0].verify
 
-        assert shutdown_post.called_once
+        assert shutdown_post.call_count == 1
         assert False is shutdown_post.request_history[0].verify
 
     def test_submit_with_true_ssl_arg(self, requests_mock):