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