replace assert_called_once()

no available in python < 3.6
open-telemetry · Nov 13, 2019 · 9c44761 · 9c44761
1 parent 981f2ba
commit 9c44761
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/opentelemetry-sdk/tests/trace/test_trace.py b/opentelemetry-sdk/tests/trace/test_trace.py
@@ -38,8 +38,8 @@ def test_shutdown(self):
 
         tracer.shutdown()
 
-        mock_processor1.shutdown.assert_called_once()
-        mock_processor2.shutdown.assert_called_once()
+        self.assertEqual(mock_processor1.shutdown.call_count, 1)
+        self.assertEqual(mock_processor2.shutdown.call_count, 1)
 
         shutdown_python_code = """
 import atexit