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,))