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 setUp(self): patch()
def setUp(self): super(TestSQLite, self).setUp() patch()