def test_patch_unpatch(self):
# Test patch idempotence
patch()
patch()
db = sqlite3.connect(':memory:')
pin = Pin.get_from(db)
assert pin
pin.clone(tracer=self.tracer).onto(db)
db.cursor().execute('select \'blah\'').fetchall()
self.assert_structure(dict(name='sqlite.query'), )
self.reset()
# Test unpatch
unpatch()
db = sqlite3.connect(':memory:')
db.cursor().execute('select \'blah\'').fetchall()
self.assert_has_no_spans()
# Test patch again
patch()
db = sqlite3.connect(':memory:')
pin = Pin.get_from(db)
assert pin
pin.clone(tracer=self.tracer).onto(db)
db.cursor().execute('select \'blah\'').fetchall()
self.assert_structure(dict(name='sqlite.query'), )
def test_patch_unpatch(self):
tracer = get_dummy_tracer()
writer = tracer.writer
# Test patch idempotence
patch()
patch()
db = sqlite3.connect(":memory:")
pin = Pin.get_from(db)
assert pin
pin.clone(tracer=tracer).onto(db)
db.cursor().execute("select 'blah'").fetchall()
spans = writer.pop()
assert spans, spans
eq_(len(spans), 1)
# Test unpatch
unpatch()
db = sqlite3.connect(":memory:")
db.cursor().execute("select 'blah'").fetchall()
spans = writer.pop()
assert not spans, spans
# Test patch again
patch()
db = sqlite3.connect(":memory:")
pin = Pin.get_from(db)
assert pin
pin.clone(tracer=tracer).onto(db)
db.cursor().execute("select 'blah'").fetchall()
spans = writer.pop()
assert spans, spans
eq_(len(spans), 1)
def test_patch_unpatch(self):
tracer = get_dummy_tracer()
writer = tracer.writer
# Test patch idempotence
patch()
patch()
db = sqlite3.connect(":memory:")
pin = Pin.get_from(db)
assert pin
pin.clone(tracer=tracer).onto(db)
db.cursor().execute("select 'blah'").fetchall()
spans = writer.pop()
assert spans, spans
eq_(len(spans), 1)
# Test unpatch
unpatch()
db = sqlite3.connect(":memory:")
db.cursor().execute("select 'blah'").fetchall()
spans = writer.pop()
assert not spans, spans
# Test patch again
patch()
db = sqlite3.connect(":memory:")
pin = Pin.get_from(db)
assert pin
pin.clone(tracer=tracer).onto(db)
db.cursor().execute("select 'blah'").fetchall()
spans = writer.pop()
assert spans, spans
eq_(len(spans), 1)
def tearDown(self):
unpatch()
def tearDown(self):
unpatch()
super(TestSQLite, self).tearDown()
def tearDown(self):
unpatch()
