def test_fetch_list(self):
contact1 = factoryboy.ContactFactory()
contact2 = factoryboy.ContactFactory()
contact3 = factoryboy.ContactFactory()
phone1 = factoryboy.PhoneNumberFactory(contact=contact1)
phone2 = factoryboy.PhoneNumberFactory(contact=contact1)
phone3 = factoryboy.PhoneNumberFactory(contact=contact2)
contact = private.Contact()
data = contact.fetch_list(2, 0)
self.assertDictEqual(data[0], {
'id': contact1.id,
'name': u'xyz',
'email': contact1.email
})
self.assertDictEqual(data[1], {
'id': contact2.id,
'name': u'xyz',
'email': contact2.email
})
data = contact.fetch_list(2, 1)
self.assertEqual(data, [{
'id': contact2.id,
'name': u'xyz',
'email': contact2.email
}, {
'id': contact3.id,
'name': u'xyz',
'email': contact3.email
}])
def search_contact(request, **kwargs):
"""List contact given limit and offset."""
limit = int(request.GET.get('limit', constants.DEFAULT_LIMIT))
offset = int(request.GET.get('offset', constants.DEFAULT_OFFSET))
search_term = request.GET.get('search_term')
contact = private.Contact()
data = contact.fetch_list(limit, offset, search_term)
return JsonResponse({'objects': data})
def create_contact(request, **kwargs):
"""Create contact."""
data = json.loads(request.body)
contact = private.Contact()
try:
contact.check_and_create(data.get('email'), data.get('name'),
data.get('phone_number'), data.get('code'),
data.get('number_type'),
data.get('image_url'))
except exception.ContactException as ex:
return JsonResponse({'error_message': ex.message})
return JsonResponse({'success_message': 'Contact added successfully.'})
def get_or_update_contact(request, **kwargs):
"""Get or update contact based on method request."""
contact_id = kwargs['id']
contact = private.Contact()
if request.method.lower() == 'get':
data = contact.get_contact(contact_id)
return JsonResponse(data)
elif request.method.lower() == 'post':
data = json.loads(request.body)
try:
contact.check_and_update(pk=contact_id, name=data.get('name'))
except exception.ContactException as ex:
return JsonResponse({'error_message': ex.message})
elif request.method.lower() == 'delete':
contact.delete(pk=contact_id)
return JsonResponse({'success_message': 'Contact updated successfully.'})
def test_can_create_update(self):
contact1 = factoryboy.ContactFactory()
contact = private.Contact()
self.assertTrue(contact._can_update_contact(contact1.id))
self.assertFalse(contact._can_update_contact(123123))
def test_can_create_contact(self):
contact1 = factoryboy.ContactFactory()
contact = private.Contact()
self.assertFalse(contact._can_create_contact(contact1.email))
self.assertTrue(contact._can_create_contact('*****@*****.**'))
def setUp(self):
super(ContactTest, self).setUp()
self.contact = private.Contact()