Adding sqlalchemy native tags in sqlalchemy commenter (#1206)

This commit is contained in:
Thiyagu55
2022-08-04 18:43:12 +05:30
committed by GitHub
parent 04f4fe5076
commit 14077a95c5
4 changed files with 69 additions and 4 deletions

View File

@ -46,10 +46,11 @@ class TestSqlalchemyInstrumentationWithSQLCommenter(TestBase):
engine=engine,
tracer_provider=self.tracer_provider,
enable_commenter=True,
commenter_options={"db_framework": False},
)
cnx = engine.connect()
cnx.execute("SELECT 1;").fetchall()
self.assertRegex(
self.caplog.records[-2].getMessage(),
r"SELECT 1 /\*traceparent='\d{1,2}-[a-zA-Z0-9_]{32}-[a-zA-Z0-9_]{16}-\d{1,2}'\*/;",
r"SELECT 1 /\*db_driver='(.*)',traceparent='\d{1,2}-[a-zA-Z0-9_]{32}-[a-zA-Z0-9_]{16}-\d{1,2}'\*/;",
)