def assertGeneratesPairing(self, host, guest): old_pks = [p.pk for p in Pairing.objects.all()] generate_pairings() try: pairing = Pairing.objects.exclude(pk__in=old_pks).get() self.assertEqual(set(pairing.flats.all()), set([host, guest])) except Pairing.DoesNotExist: if guest is not None: self.fail("no pairing generated but expected %s" % [host, guest])
def post_create_flatshare(sender, instance, created, **kwargs): if created: from pairings.models import generate_pairings generate_pairings()