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)
