def test_correct_span_names(self):
cursor = self.cursor
tracer = self.tracer
cursor.rowcount = 0
pin = Pin('pin_name', tracer=tracer)
traced_cursor = TracedCursor(cursor, pin, {})
traced_cursor.execute('arg_1', kwarg1='kwarg1')
self.assert_structure(dict(name='sql.query'))
assert_is_measured(self.get_root_span())
self.reset()
traced_cursor.executemany('arg_1', kwarg1='kwarg1')
self.assert_structure(dict(name='sql.query'))
assert_is_measured(self.get_root_span())
self.reset()
traced_cursor.callproc('arg_1', 'arg2')
self.assert_structure(dict(name='sql.query'))
assert_is_measured(self.get_root_span())
self.reset()
traced_cursor.fetchone('arg_1', kwarg1='kwarg1')
self.assert_has_no_spans()
traced_cursor.fetchmany('arg_1', kwarg1='kwarg1')
self.assert_has_no_spans()
traced_cursor.fetchall('arg_1', kwarg1='kwarg1')
self.assert_has_no_spans()
def test_correct_span_names_can_be_overridden_by_pin(self):
cursor = self.cursor
tracer = self.tracer
cursor.rowcount = 0
pin = Pin('pin_name', app='changed', tracer=tracer)
traced_cursor = TracedCursor(cursor, pin)
traced_cursor.execute('arg_1', kwarg1='kwarg1')
self.assert_structure(dict(name='changed.query'))
self.reset()
traced_cursor.executemany('arg_1', kwarg1='kwarg1')
self.assert_structure(dict(name='changed.query'))
self.reset()
traced_cursor.callproc('arg_1', 'arg2')
self.assert_structure(dict(name='changed.query'))
self.reset()
traced_cursor.fetchone('arg_1', kwarg1='kwarg1')
self.assert_has_no_spans()
traced_cursor.fetchmany('arg_1', kwarg1='kwarg1')
self.assert_has_no_spans()
traced_cursor.fetchall('arg_1', kwarg1='kwarg1')
self.assert_has_no_spans()
def test_correct_span_names(self):
cursor = self.cursor
tracer = self.tracer
cursor.rowcount = 0
pin = Pin("pin_name", tracer=tracer)
traced_cursor = TracedCursor(cursor, pin, {})
traced_cursor.execute("arg_1", kwarg1="kwarg1")
self.assert_structure(dict(name="sql.query"))
assert_is_measured(self.get_root_span())
self.reset()
traced_cursor.executemany("arg_1", kwarg1="kwarg1")
self.assert_structure(dict(name="sql.query"))
assert_is_measured(self.get_root_span())
self.reset()
traced_cursor.callproc("arg_1", "arg2")
self.assert_structure(dict(name="sql.query"))
assert_is_measured(self.get_root_span())
self.reset()
traced_cursor.fetchone("arg_1", kwarg1="kwarg1")
self.assert_has_no_spans()
traced_cursor.fetchmany("arg_1", kwarg1="kwarg1")
self.assert_has_no_spans()
traced_cursor.fetchall("arg_1", kwarg1="kwarg1")
self.assert_has_no_spans()
def test_when_pin_disabled_then_no_tracing(self):
cursor = self.cursor
tracer = self.tracer
cursor.rowcount = 0
cursor.execute.return_value = '__result__'
cursor.executemany.return_value = '__result__'
tracer.enabled = False
pin = Pin('pin_name', tracer=tracer)
traced_cursor = TracedCursor(cursor, pin, {})
assert '__result__' == traced_cursor.execute('arg_1', kwarg1='kwarg1')
assert len(tracer.writer.pop()) == 0
assert '__result__' == traced_cursor.executemany('arg_1',
kwarg1='kwarg1')
assert len(tracer.writer.pop()) == 0
cursor.callproc.return_value = 'callproc'
assert 'callproc' == traced_cursor.callproc('arg_1', 'arg_2')
assert len(tracer.writer.pop()) == 0
cursor.fetchone.return_value = 'fetchone'
assert 'fetchone' == traced_cursor.fetchone('arg_1', 'arg_2')
assert len(tracer.writer.pop()) == 0
cursor.fetchmany.return_value = 'fetchmany'
assert 'fetchmany' == traced_cursor.fetchmany('arg_1', 'arg_2')
assert len(tracer.writer.pop()) == 0
cursor.fetchall.return_value = 'fetchall'
assert 'fetchall' == traced_cursor.fetchall('arg_1', 'arg_2')
assert len(tracer.writer.pop()) == 0
def test_when_pin_disabled_then_no_tracing(self):
cursor = self.cursor
tracer = self.tracer
cursor.rowcount = 0
cursor.execute.return_value = "__result__"
cursor.executemany.return_value = "__result__"
tracer.enabled = False
pin = Pin("pin_name", tracer=tracer)
traced_cursor = TracedCursor(cursor, pin, {})
assert "__result__" == traced_cursor.execute("arg_1", kwarg1="kwarg1")
assert len(tracer.pop()) == 0
assert "__result__" == traced_cursor.executemany("arg_1", kwarg1="kwarg1")
assert len(tracer.pop()) == 0
cursor.callproc.return_value = "callproc"
assert "callproc" == traced_cursor.callproc("arg_1", "arg_2")
assert len(tracer.pop()) == 0
cursor.fetchone.return_value = "fetchone"
assert "fetchone" == traced_cursor.fetchone("arg_1", "arg_2")
assert len(tracer.pop()) == 0
cursor.fetchmany.return_value = "fetchmany"
assert "fetchmany" == traced_cursor.fetchmany("arg_1", "arg_2")
assert len(tracer.pop()) == 0
cursor.fetchall.return_value = "fetchall"
assert "fetchall" == traced_cursor.fetchall("arg_1", "arg_2")
assert len(tracer.pop()) == 0
def test_executemany_wrapped_is_called_and_returned(self):
cursor = self.cursor
cursor.rowcount = 0
cursor.executemany.return_value = '__result__'
pin = Pin('pin_name', tracer=self.tracer)
traced_cursor = TracedCursor(cursor, pin, {})
# DEV: We always pass through the result
assert '__result__' == traced_cursor.executemany('__query__', 'arg_1', kwarg1='kwarg1')
cursor.executemany.assert_called_once_with('__query__', 'arg_1', kwarg1='kwarg1')
def test_executemany_wrapped_is_called_and_returned(self):
cursor = self.cursor
cursor.rowcount = 0
cursor.executemany.return_value = "__result__"
pin = Pin("pin_name", tracer=self.tracer)
traced_cursor = TracedCursor(cursor, pin, {})
# DEV: We always pass through the result
assert "__result__" == traced_cursor.executemany("__query__", "arg_1", kwarg1="kwarg1")
cursor.executemany.assert_called_once_with("__query__", "arg_1", kwarg1="kwarg1")
def test_executemany_wrapped_is_called_and_returned(self):
cursor = self.cursor
cursor.rowcount = 0
pin = Pin('pin_name', tracer=self.tracer)
traced_cursor = TracedCursor(cursor, pin)
assert traced_cursor is traced_cursor.executemany(
'__query__', 'arg_1', kwarg1='kwarg1')
cursor.executemany.assert_called_once_with('__query__',
'arg_1',
kwarg1='kwarg1')
