def count_all_entities(self, entity_type):
"""
Count all entities of a given type currently stored in DB.
"""
result = 0
session = None
try:
session = SessionMaker()
session.open_session()
result = session.query(entity_type).count()
except Exception, excep:
LOGGER.warning(excep)
def get_all_entities(self, entity_type):
"""
Retrieve all entities of a given type.
"""
result = []
session = None
try:
session = SessionMaker()
session.open_session()
result = session.query(entity_type).all()
except Exception, excep:
LOGGER.warning(excep)
def get_all_entities(self, entity_type):
"""
Retrieve all entities of a given type.
"""
result = []
session = None
try:
session = SessionMaker()
session.open_session()
result = session.query(entity_type).all()
except Exception, excep:
LOGGER.warning(excep)
def count_all_entities(self, entity_type):
"""
Count all entities of a given type currently stored in DB.
"""
result = 0
session = None
try:
session = SessionMaker()
session.open_session()
result = session.query(entity_type).count()
except Exception, excep:
LOGGER.warning(excep)
def _evaluate_db_filter(self, filter_chain, expected_number):
"""
Evaluate filter on DB and assert number of results.
"""
session = SessionMaker()
try:
session.open_session()
query = session.query(Datatype1)
filter_str = filter_chain.get_sql_filter_equivalent("Datatype1")
query = query.filter(eval(filter_str))
result = query.all()
session.close_session()
except Exception, excep:
session.close_session()
raise excep
def _evaluate_db_filter(self, filter_chain, expected_number):
"""
Evaluate filter on DB and assert number of results.
"""
session = SessionMaker()
try:
session.open_session()
query = session.query(Datatype1)
filter_str = filter_chain.get_sql_filter_equivalent("Datatype1")
query = query.filter(eval(filter_str))
result = query.all()
session.close_session()
except Exception, excep:
session.close_session()
raise excep
def _evaluate_db_filter(self, filter_chain, expected_number):
"""
Evaluate filter on DB and assert number of results.
"""
session = SessionMaker()
try:
session.open_session()
query = session.query(Datatype1)
filter_str = filter_chain.get_sql_filter_equivalent("Datatype1")
query = query.filter(eval(filter_str))
result = query.all()
session.close_session()
except Exception as excep:
session.close_session()
raise excep
self.assertEquals(expected_number, len(result), "Expected %s DTs after filtering with %s, "
"but got %s instead." % (expected_number, filter_chain, len(result,)))
def _evaluate_db_filter(self, filter_chain, expected_number):
"""
Evaluate filter on DB and assert number of results.
"""
session = SessionMaker()
try:
session.open_session()
query = session.query(Datatype1)
filter_str = filter_chain.get_sql_filter_equivalent("Datatype1")
query = query.filter(eval(filter_str))
result = query.all()
session.close_session()
except Exception as excep:
session.close_session()
raise excep
assert expected_number == len(result), "Expected %s DTs after filtering with %s, "\
"but got %s instead." % (expected_number, filter_chain, len(result,))
