def test_model_query(self):
job_doc1 = copy.deepcopy(self.fake_job_0)
job_doc2 = copy.deepcopy(self.fake_job_0)
self.dbapi.add_job(user_id=self.fake_user_id,
doc=job_doc1)
job_id2 = self.dbapi.add_job(user_id=self.fake_user_id,
doc=job_doc2)
result = self.dbapi.delete_job(user_id=self.fake_user_id,
job_id=job_id2)
session = api.get_db_session()
with session.begin():
try:
query = api.model_query(session, models.Job,
read_deleted='no')
query = query.filter_by(user_id=self.fake_user_id)
result = query.all()
self.assertEqual(len(result), 1)
query = api.model_query(session, models.Job,
read_deleted='only')
query = query.filter_by(user_id=self.fake_user_id)
result = query.all()
self.assertEqual(len(result), 1)
query = api.model_query(session, models.Job,
read_deleted='yes')
query = query.filter_by(user_id=self.fake_user_id)
result = query.all()
self.assertEqual(len(result), 2)
except Exception as e:
raise freezer_api_exc.StorageEngineError(
message='sqlalchemy operation failed {0}'.format(e))
session.close()
def test_raises_model_query(self):
session = api.get_db_session()
session.begin()
self.assertRaises(ValueError,
api.model_query, session, models.Job,
read_deleted='both')
session.close()