def setUp(self): engine = create_engine("sqlite://") # in-memory db Base.metadata.create_all(bind=engine) factory = SQLAlchemySessionContextFactory(engine) self.baseplate_observer = TestBaseplateObserver() baseplate = Baseplate() baseplate.register(self.baseplate_observer) baseplate.add_to_context("db", factory) self.context = mock.Mock() self.server_span = baseplate.make_server_span(self.context, "test")
def setUp(self): engine = engine_from_config({"database.url": "sqlite://"}) # in-memory db Base.metadata.create_all(bind=engine) factory = SQLAlchemySessionContextFactory(engine) self.baseplate_observer = TestBaseplateObserver() baseplate = Baseplate() baseplate.register(self.baseplate_observer) baseplate.add_to_context("db", factory) self.context = baseplate.make_context_object() self.server_span = baseplate.make_server_span(self.context, "test")
class SQLAlchemyTests(unittest.TestCase): def setUp(self): self.engine = create_engine("sqlite://") # in-memory db Base.metadata.create_all(bind=self.engine) self.factory = SQLAlchemySessionContextFactory(self.engine) def test_session(self): root_span = mock.Mock(autospec=RootSpan) span = mock.Mock(autospec=Span) span.id = 1234 span.trace_id = 2345 root_span.make_child.return_value = span session = self.factory.make_object_for_context("db", root_span) new_object = TestObject(name="cool") session.add(new_object) session.commit() self.assertEqual(root_span.make_child.call_args, mock.call("db.execute"))
def setUp(self): self.engine = create_engine("sqlite://") # in-memory db Base.metadata.create_all(bind=self.engine) self.factory = SQLAlchemySessionContextFactory(self.engine)