def setUp(self): TestUser.setUp(self) self.user = models.User(self.username, self.password, self.email) self.mock_token ='foobarbaz123456' with models.sessionmaker(self.engine) as session: session.add(self.user)
def test_sessionmaker_with_engine(self): engine = mock.MagicMock() session = models.sessionmaker(engine=engine) self.assertEqual(session.bind, engine)
def tearDown(self): with models.sessionmaker(self.engine) as session: user = session.query(models.User).\ filter_by(id=1).first() session.delete(user)