fix tests

open-telemetry · Apr 24, 2020 · 0740696 · 0740696
1 parent 269156e
commit 0740696
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/ext/opentelemetry-ext-docker-tests/tests/mysql/test_mysql_functional.py b/ext/opentelemetry-ext-docker-tests/tests/mysql/test_mysql_functional.py
@@ -65,7 +65,7 @@ def validate_spans(self):
         self.assertEqual(root_span.name, "rootSpan")
         self.assertEqual(db_span.name, "mysql.opentelemetry-tests")
         self.assertIsNotNone(db_span.parent)
-        self.assertEqual(db_span.parent.name, root_span.name)
+        self.assertIs(db_span.parent, root_span.get_context())
         self.assertIs(db_span.kind, trace_api.SpanKind.CLIENT)
         self.assertEqual(db_span.attributes["db.instance"], MYSQL_DB_NAME)
         self.assertEqual(db_span.attributes["net.peer.name"], MYSQL_HOST)

diff --git a/ext/opentelemetry-ext-docker-tests/tests/postgres/test_psycopg_functional.py b/ext/opentelemetry-ext-docker-tests/tests/postgres/test_psycopg_functional.py
@@ -68,7 +68,7 @@ def validate_spans(self):
         self.assertEqual(root_span.name, "rootSpan")
         self.assertEqual(child_span.name, "postgresql.opentelemetry-tests")
         self.assertIsNotNone(child_span.parent)
-        self.assertEqual(child_span.parent.name, root_span.name)
+        self.assertIs(child_span.parent, root_span.get_context())
         self.assertIs(child_span.kind, trace_api.SpanKind.CLIENT)
         self.assertEqual(
             child_span.attributes["db.instance"], POSTGRES_DB_NAME