def get(self, request):
value = request.query_params["address1"]
matches = match_location(address1_name=value)
if matches:
serializer = Address1Serializer(matches, many=True)
return Response(serializer.data)
else:
return Response({"id": "-1", "name": "None"})
def test_filter_by_address1(self):
Address2.objects.all().delete()
Address2Factory.create_batch(20)
original = Address2.objects.all()[0]
self.Address2List = Address2Factory.create_batch(20)
matches = match_location(original.address1.name, original.name)
# Address2Factory associates a new Address1 with each Address2
self.assertEqual(len(matches), 1)
def test_filter_by_address1_multiple_address2s(self):
Address2.objects.all().delete()
Address2Factory.create_batch(20)
# Associate a new Address2 with the first Address2s Address1
original = Address2.objects.all()[0]
second = Address2.objects.all()[1]
second.address1 = original.address1
second.save()
# Now there should only be 2 valid Address2s for this address1
matches = match_location(original.address1.name, original.name)
self.assertEqual(len(matches), 2)
def test_address2_match_found_if_one_character_off(self):
original = Address2.objects.all()[0]
close_name = original.name + "s"
matches = match_location(address2_name=close_name)
self.assertEqual(original, matches[0])
def test_address2_matching_works(self):
original = Address2.objects.all()[0]
matches = match_location(address2_name=original.name)
self.assertEqual(original, matches[0])