def test__FacilityQuery__is_eligible_to_retire__sector_exists__returns_correct_facilities( seeded_session, ggo_sector): ggo = Mock(sector=ggo_sector) query = FacilityQuery(seeded_session) \ .is_eligible_to_retire(ggo) assert query.count() > 0 assert all(f.sector == ggo_sector for f in query.all()) assert all(f.facility_type == FacilityType.CONSUMPTION for f in query.all())
def test__FacilityQuery__is_consumer__returns_correct_facilities(seeded_session): query = FacilityQuery(seeded_session) \ .is_consumer() assert query.count() > 0 assert all(f.facility_type == FacilityType.CONSUMPTION for f in query.all())
def test__FacilityQuery__is_retire_receiver__returns_correct_facilities(seeded_session): query = FacilityQuery(seeded_session) \ .is_retire_receiver() assert query.count() > 0 assert all(f.retiring_priority is not None for f in query.all())
def test__FacilityQuery__is_type__returns_correct_facilities( seeded_session, facility_type): query = FacilityQuery(seeded_session) \ .is_type(facility_type) assert query.count() > 0 assert all(f.facility_type == facility_type for f in query.all())
def test__FacilityQuery__belongs_to__Facility_exists__returns_correct_facility( seeded_session, user): query = FacilityQuery(seeded_session) \ .belongs_to(user) assert query.count() > 0 assert all(f.user_id == user.id for f in query.all())
def test__FacilityQuery__has_any_gsrn__some_ids_exists__returns_correct_facility( seeded_session, gsrn): query = FacilityQuery(seeded_session) \ .has_any_gsrn(gsrn) assert query.count() > 0 assert all(f.gsrn in gsrn for f in query.all())
def test__FacilityQuery__has_any_public_id__some_ids_exists__returns_correct_facility( seeded_session, public_ids): query = FacilityQuery(seeded_session) \ .has_any_public_id(public_ids) assert query.count() > 0 assert all(f.public_id in public_ids for f in query.all())
def get_facilities(self, user, filters, session): """ :param User user: :param FacilityFilters filters: :param Session session: :rtype: list[Facility] """ query = FacilityQuery(session) \ .belongs_to(user) if filters is not None: query = query.apply_filters(filters) return query.all()