def get(self, request, pk=None): if pk: return Response(status=HTTP_200_OK, data=ContactService.get(pk)) created_by_user_id = request.GET.get('createdbyuserid') if created_by_user_id: return Response(status=HTTP_200_OK, data=ContactService.get_by_user_id(created_by_user_id)) search_field = request.GET.get('searchfield') if search_field: return Response(status=HTTP_200_OK, data=ContactService.search(search_field)) return Response(status=HTTP_200_OK, data=ContactService.get_all())
def test_should_get_contact_by_user_id(self): requests.get = MagicMock(return_value=MagicMock(status_code=200, json=MagicMock(return_value=CONTACT))) response = ContactService.get_by_user_id('17') self.assertEqual(response, CONTACT)
def test_should_get_contact_by_user_id(self): requests.get = MagicMock(return_value=MagicMock( status_code=200, json=MagicMock(return_value=CONTACT))) response = ContactService.get_by_user_id('17') self.assertEqual(response, CONTACT)