Move test that imports SDK from the API (open-telemetry#1185)

ocelotl · web-flow · commit 10e2583fb5c0 · 2020-09-30T11:00:59.000-07:00
diff --git a/opentelemetry-api/tests/trace/test_globals.py b/opentelemetry-api/tests/trace/test_globals.py
@@ -1,9 +1,7 @@
 import unittest
-from logging import WARNING
 from unittest.mock import patch
 
 from opentelemetry import context, trace
-from opentelemetry.sdk.trace import TracerProvider  # type:ignore
 
 
 class TestGlobals(unittest.TestCase):
@@ -22,23 +20,6 @@ def test_get_tracer(self):
         trace.get_tracer("foo", "var", mock_provider)
         mock_provider.get_tracer.assert_called_with("foo", "var")
 
-    def test_tracer_provider_override_warning(self):
-        """trace.set_tracer_provider should throw a warning when overridden"""
-        trace.set_tracer_provider(TracerProvider())
-        tracer_provider = trace.get_tracer_provider()
-        with self.assertLogs(level=WARNING) as test:
-            trace.set_tracer_provider(TracerProvider())
-            self.assertEqual(
-                test.output,
-                [
-                    (
-                        "WARNING:opentelemetry.trace:Overriding of current "
-                        "TracerProvider is not allowed"
-                    )
-                ],
-            )
-        self.assertIs(tracer_provider, trace.get_tracer_provider())
-
 
 class TestTracer(unittest.TestCase):
     def setUp(self):
diff --git a/opentelemetry-sdk/tests/metrics/test_globals.py b/opentelemetry-sdk/tests/metrics/test_globals.py
diff --git a/opentelemetry-sdk/tests/trace/test_globals.py b/opentelemetry-sdk/tests/trace/test_globals.py
@@ -0,0 +1,25 @@
+# type:ignore
+import unittest
+from logging import WARNING
+
+from opentelemetry import trace
+from opentelemetry.sdk.trace import TracerProvider  # type:ignore
+
+
+class TestGlobals(unittest.TestCase):
+    def test_tracer_provider_override_warning(self):
+        """trace.set_tracer_provider should throw a warning when overridden"""
+        trace.set_tracer_provider(TracerProvider())
+        tracer_provider = trace.get_tracer_provider()
+        with self.assertLogs(level=WARNING) as test:
+            trace.set_tracer_provider(TracerProvider())
+            self.assertEqual(
+                test.output,
+                [
+                    (
+                        "WARNING:opentelemetry.trace:Overriding of current "
+                        "TracerProvider is not allowed"
+                    )
+                ],
+            )
+        self.assertIs(tracer_provider, trace.get_tracer_provider())
