def test_should_auto_close_polls(self, mock_send_bulk_sms): self.poll_to_post['question'] = 'Will you comply, yes or no?' self.poll_to_post['ptype'] = 'yesno' second_poll = self.poll_to_post.copy() second_poll['question'] = 'Please say, yes or no?' second_poll['keyword'] = 'anotherkey' # self.poll_to_send['text'] = 'Will you comply, yes or no? Reply With: NECOCPoll YES/NO' response = self.client.post(self.POLL_ENDPOINT, data=json.dumps(self.poll_to_post), content_type="application/json") self.assertEqual(201, response.status_code) retrieved_poll = Poll.objects(**self.poll_to_post) self.assertEqual(1, retrieved_poll.count()) self.assertEqual(True, retrieved_poll[0].open) response = self.client.post(self.POLL_ENDPOINT, data=json.dumps(second_poll), content_type="application/json") self.assertEqual(201, response.status_code) retrieved_poll = Poll.objects(**self.poll_to_post) self.assertEqual(1, retrieved_poll.count()) self.assertEqual(False, retrieved_poll[0].open) retrieved_poll = Poll.objects(**second_poll) self.assertEqual(1, retrieved_poll.count()) self.assertEqual(True, retrieved_poll[0].open)
def test_should_auto_close_polls(self, mock_send_bulk_sms): self.poll_to_post['question'] = 'Will you comply, yes or no?' self.poll_to_post['ptype'] = 'yesno' second_poll = self.poll_to_post.copy() second_poll['question'] = 'Please say, yes or no?' second_poll['keyword'] = 'anotherkey' # self.poll_to_send['text'] = 'Will you comply, yes or no? Reply With: NECOCPoll YES/NO' response = self.client.post(self.POLL_ENDPOINT, data=json.dumps(self.poll_to_post), content_type="application/json") self.assertEqual(201, response.status_code) retrieved_poll = Poll.objects(**self.poll_to_post) self.assertEqual(1, retrieved_poll.count()) self.assertEqual(True, retrieved_poll[0].open) response = self.client.post(self.POLL_ENDPOINT, data=json.dumps(second_poll), content_type="application/json") self.assertEqual(201, response.status_code) retrieved_poll = Poll.objects(**self.poll_to_post) self.assertEqual(1, retrieved_poll.count()) self.assertEqual(False, retrieved_poll[0].open) retrieved_poll = Poll.objects(**second_poll) self.assertEqual(1, retrieved_poll.count()) self.assertEqual(True, retrieved_poll[0].open)
def get_queryset(self): query_params = {key: value or None for key, value in self.request.GET.items()} user_group = self.request.user.group.name if user_group in getattr(settings, 'DISTRICT_GROUPS', []): target_locations = get_user_district_locations(self.request.user) query_params.update({'target_locations__in': target_locations}) if 'ordering' in query_params: ordering_params = query_params['ordering'] del query_params['ordering'] query_set = Poll.objects(**query_params).order_by('%s' % ordering_params) else: query_set = Poll.objects(**query_params).order_by('-created_at') return query_set
def get_queryset(self): query_params = { key: value or None for key, value in self.request.GET.items() } user_group = self.request.user.group.name if user_group in getattr(settings, 'DISTRICT_GROUPS', []): target_locations = get_user_district_locations(self.request.user) query_params.update({'target_locations__in': target_locations}) if 'ordering' in query_params: ordering_params = query_params['ordering'] del query_params['ordering'] query_set = Poll.objects(**query_params).order_by('%s' % ordering_params) else: query_set = Poll.objects(**query_params).order_by('-created_at') return query_set
def test_should_post_a_yesno_poll_and_resend_appropriate_outgoing_message(self, mock_send_bulk_sms): self.poll_to_post['question'] = 'Will you comply, yes or no?' self.poll_to_post['ptype'] = 'yesno' self.poll_to_send['text'] = 'Will you comply, yes or no? Reply With: NECOCPoll YES/NO' response = self.client.post(self.POLL_ENDPOINT, data=json.dumps(self.poll_to_post), content_type="application/json") self.assertEqual(201, response.status_code) # mock_send_bulk_sms.delay.assert_called_once_with(ANY, ANY, self.poll_to_send['text']) self.assertTrue(mock_send_bulk_sms.delay.called_once_with(ANY, ANY, self.poll_to_send['text'])) retrieved_poll = Poll.objects(**self.poll_to_post) self.assertEqual(1, retrieved_poll.count()) self.assertTrue(retrieved_poll[0].is_yesno_poll())
class PollListCreateView(ListCreateAPIView): model = Poll serializer_class = PollSerializer queryset = Poll.objects() permission_classes = [ Or(build_permission_class('dms.can_manage_polls'), And(build_permission_class('dms.can_view_polls'), IsGetRequest)) ] def get_queryset(self): query_params = { key: value or None for key, value in self.request.GET.items() } user_group = self.request.user.group.name if user_group in getattr(settings, 'DISTRICT_GROUPS', []): target_locations = get_user_district_locations(self.request.user) query_params.update({'target_locations__in': target_locations}) if 'ordering' in query_params: ordering_params = query_params['ordering'] del query_params['ordering'] query_set = Poll.objects(**query_params).order_by('%s' % ordering_params) else: query_set = Poll.objects(**query_params).order_by('-created_at') return query_set def post_save(self, obj, created=True): locations = self.get_location(obj) phone_numbers = list( UserProfile.objects(location__in=locations).values_list('phone')) if obj.ptype == 'yesno': text = '%s Reply With: NECOCPoll YES/NO' % obj.question else: text = '%s Reply With: NECOCPoll %s ...' % (obj.question, obj.keyword) send_bulk_sms.delay(obj, phone_numbers, text) def get_location(self, obj): locations = Location.objects(id__in=obj.target_locations) if locations.filter(type='subcounty'): return locations districts_children = [district.children() for district in locations] return flatten(districts_children) def get_location_id(self, user): profile = UserProfile.objects(user=user).first() return profile.location.id if profile else ''
def test_should_post_a_poll_and_save_logs(self, mock_requests): some_id = 1234 mock_response = MagicMock() mock_response.status_code = 201 mock_response.json.return_value = {"messages": [some_id], "sms": [some_id]} mock_requests.post.return_value = mock_response response = self.client.post(self.POLL_ENDPOINT, data=json.dumps(self.poll_to_post), content_type="application/json") self.assertEqual(201, response.status_code) self.assertTrue(mock_requests.post.called_once_with(API_URL, json.dumps(self.poll_to_send), self.headers)) retrieved_poll = Poll.objects(**self.poll_to_post) self.assertEqual(1, retrieved_poll.count())
def test_should_post_a_yesno_poll_and_resend_appropriate_outgoing_message( self, mock_send_bulk_sms): self.poll_to_post['question'] = 'Will you comply, yes or no?' self.poll_to_post['ptype'] = 'yesno' self.poll_to_send[ 'text'] = 'Will you comply, yes or no? Reply With: NECOCPoll YES/NO' response = self.client.post(self.POLL_ENDPOINT, data=json.dumps(self.poll_to_post), content_type="application/json") self.assertEqual(201, response.status_code) # mock_send_bulk_sms.delay.assert_called_once_with(ANY, ANY, self.poll_to_send['text']) self.assertTrue( mock_send_bulk_sms.delay.called_once_with( ANY, ANY, self.poll_to_send['text'])) retrieved_poll = Poll.objects(**self.poll_to_post) self.assertEqual(1, retrieved_poll.count()) self.assertTrue(retrieved_poll[0].is_yesno_poll())
def test_should_post_a_poll_and_save_logs(self, mock_requests): some_id = 1234 mock_response = MagicMock() mock_response.status_code = 201 mock_response.json.return_value = { "messages": [some_id], "sms": [some_id] } mock_requests.post.return_value = mock_response response = self.client.post(self.POLL_ENDPOINT, data=json.dumps(self.poll_to_post), content_type="application/json") self.assertEqual(201, response.status_code) self.assertTrue( mock_requests.post.called_once_with(API_URL, json.dumps(self.poll_to_send), self.headers)) retrieved_poll = Poll.objects(**self.poll_to_post) self.assertEqual(1, retrieved_poll.count())
def validate_keyword(self, attrs, source): if len(Poll.objects(keyword=attrs[source])): raise rest_serializers.ValidationError('Keyword must be unique') return attrs
def validate_keyword(self, attrs, source): if len(Poll.objects(keyword=attrs[source])): raise rest_serializers.ValidationError('Keyword must be unique') return attrs