def test_get_form_ids_having_multimedia_with_user_types(self): start = datetime(2013, 7, 1) end = datetime(2013, 7, 30) xmlns = 'http://a.b.org' app_id = '1234' user_id = 'abc' with patch('corehq.pillows.xform.get_user_type', lambda _: MOBILE_USER_TYPE): self._send_form_to_es( app_id=app_id, xmlns=xmlns, received_on=datetime(2013, 7, 2), user_id=user_id, attachment_dict={ 'my_image': {'content_type': 'image/jpg'} } ) with patch('corehq.pillows.xform.get_user_type', lambda _: WEB_USER_TYPE): self._send_form_to_es( app_id=app_id, xmlns=xmlns, received_on=datetime(2013, 7, 2), user_id=user_id, attachment_dict={ 'my_image': {'content_type': 'image/jpg'} } ) form_ids = get_form_ids_having_multimedia( self.domain, app_id, xmlns, start, end, user_types=[MOBILE_USER_TYPE] ) self.assertEqual(len(form_ids), 1) form_ids = get_form_ids_having_multimedia( self.domain, app_id, xmlns, start, end, user_types=[MOBILE_USER_TYPE, WEB_USER_TYPE] ) self.assertEqual(len(form_ids), 2) form_ids = get_form_ids_having_multimedia( self.domain, app_id, xmlns, start, end, user_types=[] ) self.assertEqual(len(form_ids), 2)
def test_get_form_ids_having_multimedia_with_user_types(self): start = datetime(2013, 7, 1) end = datetime(2013, 7, 30) xmlns = 'http://a.b.org' app_id = '1234' user_id = 'abc' with patch('corehq.pillows.xform.get_user_type', lambda _: MOBILE_USER_TYPE): self._send_form_to_es( app_id=app_id, xmlns=xmlns, received_on=datetime(2013, 7, 2), user_id=user_id, attachment_dict={ 'my_image': {'content_type': 'image/jpg'} } ) with patch('corehq.pillows.xform.get_user_type', lambda _: WEB_USER_TYPE): self._send_form_to_es( app_id=app_id, xmlns=xmlns, received_on=datetime(2013, 7, 2), user_id=user_id, attachment_dict={ 'my_image': {'content_type': 'image/jpg'} } ) form_ids = get_form_ids_having_multimedia( self.domain, app_id, xmlns, DateSpan(start, end), [MOBILE_USER_TYPE] ) self.assertEqual(len(form_ids), 1) form_ids = get_form_ids_having_multimedia( self.domain, app_id, xmlns, DateSpan(start, end), [MOBILE_USER_TYPE, WEB_USER_TYPE] ) self.assertEqual(len(form_ids), 2) form_ids = get_form_ids_having_multimedia( self.domain, app_id, xmlns, DateSpan(start, end), [] ) self.assertEqual(len(form_ids), 2)
def test_get_form_ids_having_multimedia(self): start = datetime(2013, 7, 1) end = datetime(2013, 7, 30) xmlns = 'http://a.b.org' app_id = '1234' user_id = 'abc' self._send_form_to_es( app_id=app_id, xmlns=xmlns, received_on=datetime(2013, 7, 2), user_id=user_id, attachment_dict={'my_image': { 'content_type': 'image/jpg' }}) # Decoy form self._send_form_to_es( app_id=app_id, xmlns=xmlns, received_on=datetime(2013, 7, 2), user_id=user_id, ) form_ids = get_form_ids_having_multimedia( self.domain, app_id, xmlns, DateSpan(start, end), [], ) self.assertEqual(len(form_ids), 1)
def test_get_form_ids_having_multimedia(self): start = datetime(2013, 7, 1) end = datetime(2013, 7, 30) xmlns = 'http://a.b.org' app_id = '1234' user_id = 'abc' self._send_form_to_es( app_id=app_id, xmlns=xmlns, received_on=datetime(2013, 7, 2), user_id=user_id, attachment_dict={ 'my_image': {'content_type': 'image/jpg'} } ) # Decoy form self._send_form_to_es( app_id=app_id, xmlns=xmlns, received_on=datetime(2013, 7, 2), user_id=user_id, ) form_ids = get_form_ids_having_multimedia( self.domain, app_id, xmlns, start, end ) self.assertEqual(len(form_ids), 1)
def test_get_form_ids_having_multimedia_with_group(self): start = datetime(2013, 7, 1) end = datetime(2013, 7, 30) xmlns = 'http://a.b.org' app_id = '1234' user_id = 'escobar' self._send_group_to_es( _id='medellin', users=['escobar'], ) self._send_form_to_es( app_id=app_id, xmlns=xmlns, received_on=datetime(2013, 7, 2), user_id=user_id, attachment_dict={ 'my_image': {'content_type': 'image/jpg'} } ) self._send_form_to_es( app_id=app_id, xmlns=xmlns, received_on=datetime(2013, 7, 2), user_id='murphy', attachment_dict={ 'my_image': {'content_type': 'image/jpg'} } ) form_ids = get_form_ids_having_multimedia( self.domain, app_id, xmlns, start, end, group='medellin', ) self.assertEqual(len(form_ids), 1)
def test_get_form_ids_having_multimedia_with_group(self): start = datetime(2013, 7, 1) end = datetime(2013, 7, 30) xmlns = 'http://a.b.org' app_id = '1234' user_id = 'escobar' self._send_group_to_es( _id='medellin', users=['escobar'], ) self._send_form_to_es( app_id=app_id, xmlns=xmlns, received_on=datetime(2013, 7, 2), user_id=user_id, attachment_dict={'my_image': { 'content_type': 'image/jpg' }}) self._send_form_to_es( app_id=app_id, xmlns=xmlns, received_on=datetime(2013, 7, 2), user_id='murphy', attachment_dict={'my_image': { 'content_type': 'image/jpg' }}) form_ids = get_form_ids_having_multimedia( self.domain, app_id, xmlns, start, end, group='medellin', ) self.assertEqual(len(form_ids), 1)