def test_delete(self):
profile = ProfileFactory()
profile.partner_staff_member = profile.user.pk
profile.save()
models.delete_partner_relationship(None, profile.user)
profile_updated = models.UserProfile.objects.get(pk=profile.pk)
self.assertIsNone(profile_updated.partner_staff_member)
user = models.User.objects.get(pk=profile.user.pk)
self.assertFalse(user.is_active)
def test_status_queries(self):
profile = ProfileFactory()
profile.status = Profile.STATUS.active
profile.save()
assert Profile.objects.active().count() == 1
profile.status = Profile.STATUS.inactive
profile.save()
assert Profile.objects.inactive().count() == 1
profile.status = Profile.STATUS.banned
profile.save()
assert Profile.objects.banned().count() == 1
def test_save_vendor_number(self):
profile = ProfileFactory()
profile.vendor_number = ""
profile.save()
self.assertIsNone(profile.vendor_number)
def test_save_staff_id(self):
profile = ProfileFactory()
profile.staff_id = ""
profile.save()
self.assertIsNone(profile.staff_id)
