def create_pumping_station(self):
"""Create a pumping station and save it to the database."""
# Before we can create and save a PumpingStation, we have to set the
# required fields.
# If a required field refers to another model, we do not save that model
# to the database, otherwise we would also have to create the required
# fields of that model.
open_water = OpenWater()
open_water.pk = 1
# If we do not set the pk field of the model or set it after we have
# assigned the model to the PumpingStation, we get an IntegrityError.
label = Label()
label.pk = self.label_pk
self.label_pk += 1
pumping_station = PumpingStation()
pumping_station.open_water = open_water
pumping_station.label = label
pumping_station.into = True
pumping_station.computed_level_control = True
pumping_station.percentage = 100
pumping_station.save()
return pumping_station
def test_b(self):
"""Test no intake are available for level control.
There is an intake available for level control but that belongs to
another open water.
"""
open_water = OpenWater()
open_water.pk = 2
intake = find_pumping_station_level_control(open_water, True)
self.assertTrue(intake is None)
def test_no_intakes_or_pumps_exist():
"""Test no intakes or pumps exist."""
open_water = OpenWater()
open_water.pk = 1
intake = find_pumping_station_level_control(open_water, True)
assert intake is None
