def test_random_region_multiple_regions(self):
# Create new region
netherlands = Region.objects.create(name="Netherlands", tva_percentage=21)
netherlands.save()
#Return either one
self.assertTrue(random_region() in Region.objects.all())
def test_personal_shopper_assignment(self):
#Create some personal shoppers
personal_shopper1 = PersonalShopper.objects.create(
first_name=names.get_first_name(gender="female"),
last_name=names.get_last_name(),
)
personal_shopper2 = PersonalShopper.objects.create(
first_name=names.get_first_name(gender="female"),
last_name=names.get_last_name(),
)
personal_shopper1.save()
personal_shopper2.save()
personal_shopper1.regions.add(self.region)
personal_shopper2.regions.add(self.region)
#Create some users without recent orders and assign them to personal shopper 1
for _ in xrange(5):
member = Member.objects.create(
first_name=names.get_first_name(gender='male'),
last_name=names.get_last_name(),
phone_number=random.randint(1, 9999999999),
region=random_region(),
personal_shopper=personal_shopper1,
)
member.save()
#Check if a new member would be assigned to personal shopper 2
self.assertEqual(assign_personal_shopper(self.region), personal_shopper2)
def test_random_region_single_region(self):
"""
Return a random region
"""
# Only one region, so it should return this one
self.assertTrue(random_region(), self.region)