def test_multiple(self): pilot1 = helper.create_pilot('kim', 'Kim', 'Pilot1') pilot2 = helper.create_pilot('sam', 'Sam', 'Pilot2') airstrip1 = helper.create_airstrip('ID1', 'Airstrip1') airstrip2 = helper.create_airstrip('ID2', 'Airstrip2') expected = [ (pilot1.username, airstrip1.ident), (pilot1.username, airstrip2.ident), (pilot2.username, airstrip1.ident), (pilot2.username, airstrip2.ident), ] self.assertEqual(util.get_pilot_airstrip_pairs(), expected) # Filter on Pilot expected = [ (pilot1.username, airstrip1.ident), (pilot1.username, airstrip2.ident), ] self.assertEqual(util.get_pilot_airstrip_pairs(pilot=pilot1), expected) # Filter on Airstrip expected = [ (pilot1.username, airstrip2.ident), (pilot2.username, airstrip2.ident), ] self.assertEqual(util.get_pilot_airstrip_pairs(airstrip=airstrip2), expected) # Filter on Base base1 = helper.create_airstrip('BASE', 'Base1', is_base=True) airstrip1.bases.add(base1) expected = [ (pilot1.username, airstrip1.ident), (pilot2.username, airstrip1.ident), ] self.assertEqual(util.get_pilot_airstrip_pairs(base=base1), expected)
def test_single(self): pilot = helper.create_pilot() airstrip = helper.create_airstrip() expected = [(pilot.username, airstrip.ident)] self.assertEqual(util.get_pilot_airstrip_pairs(), expected)
def test_empty(self): self.assertEqual(util.get_pilot_airstrip_pairs(), [])