def test_callproc_and_commit_are_not_traced(self):
connection = PsycopgConnectionTracing('dbname=test', tracer=self.tracer, trace_callproc=False,
trace_commit=False, connection_factory=MockDBAPIConnection,
cursor_factory=MockDBAPICursor)
with connection as cursor:
cursor.callproc('my_procedure')
assert not self.tracer.finished_spans()
def test_executemany_and_commit_are_not_traced(self):
connection = PsycopgConnectionTracing('dbname=test', tracer=self.tracer, trace_executemany=False,
trace_commit=False, connection_factory=MockDBAPIConnection,
cursor_factory=MockDBAPICursor)
with connection as cursor:
cursor.executemany('INSERT INTO some_table VALUES (%s, %s, %s)')
assert not self.tracer.finished_spans()
def test_execute_and_commit_are_not_traced(self):
connection = PsycopgConnectionTracing('dbname=test', tracer=self.tracer, trace_execute=False,
trace_commit=False, connection_factory=MockDBAPIConnection,
cursor_factory=MockDBAPICursor)
with connection as cursor:
cursor.execute('SELECT * FROM SOME_TABLE')
assert not self.tracer.finished_spans()
def setup(self):
self.tracer = MockTracer()
self.connection = PsycopgConnectionTracing(
'dbname=test',
tracer=self.tracer,
connection_factory=MockDBAPIConnection,
cursor_factory=MockDBAPICursor)
assert isinstance(self.connection, MockDBAPIConnection)
def test_callproc_and_rollback_are_not_traced(self):
connection = PsycopgConnectionTracing('dbname=test', tracer=self.tracer, trace_callproc=False,
trace_rollback=False, connection_factory=MockDBAPIConnection,
cursor_factory=MockDBAPICursor)
with connection as cursor:
with patch.object(MockDBAPICursor, 'callproc', side_effect=SomeException()) as callproc:
callproc.__name__ = 'callproc'
cursor.callproc('my_procedure')
assert not self.tracer.finished_spans()
def test_executemany_and_rollback_are_not_traced(self):
connection = PsycopgConnectionTracing('dbname=test', tracer=self.tracer, trace_executemany=False,
trace_rollback=False, connection_factory=MockDBAPIConnection,
cursor_factory=MockDBAPICursor)
with connection as cursor:
with patch.object(MockDBAPICursor, 'executemany', side_effect=SomeException()) as executemany:
executemany.__name__ = 'executemany'
cursor.executemany('INSERT INTO some_table VALUES (%s, %s, %s)')
assert not self.tracer.finished_spans()
def test_execute_and_rollback_are_not_traced(self):
connection = PsycopgConnectionTracing('dbname=test', tracer=self.tracer, trace_execute=False,
trace_rollback=False, connection_factory=MockDBAPIConnection,
cursor_factory=MockDBAPICursor)
with connection as cursor:
with patch.object(MockDBAPICursor, 'execute', side_effect=SomeException()) as execute:
execute.__name__ = 'execute'
cursor.execute('SELECT * FROM some_table')
assert not self.tracer.finished_spans()
def test_custom_span_tags(self):
span_tags = dict(one=123, two=234)
tracer = MockTracer()
connection = PsycopgConnectionTracing('dbname=test', tracer=tracer, span_tags=span_tags,
connection_factory=MockDBAPIConnection, cursor_factory=MockDBAPICursor)
with connection as cursor:
cursor.execute('insert')
cursor.executemany('insert', [1, 2])
cursor.callproc('procedure')
spans = tracer.finished_spans()
assert len(spans) == 4
for span in spans:
assert span.tags[tags.DATABASE_TYPE] == 'sql'
assert span.tags[tags.SPAN_KIND] == tags.SPAN_KIND_RPC_CLIENT
assert span.tags['one'] == 123
assert span.tags['two'] == 234