def test_survey_response_is_edited_and_new_submission_and_datarecord_is_created(
self):
test_data = TestData(self.manager)
survey_response_service = SurveyResponseService(self.manager)
values = {
'ID': test_data.entity1.short_code,
'Q1': 'name',
'Q2': '80',
'Q3': 'a'
}
transport_info = TransportInfo('web', 'src', 'dest')
request = Request(values, transport_info)
saved_response = survey_response_service.save_survey(
'CL1', values, [], transport_info, "rep2")
self.assertDictEqual(
OrderedDict([('Q1', 'name'), ('Q3', ['RED']), ('Q2', 80),
('ID', u'1')]), saved_response.processed_data)
new_values = {
'ID': test_data.entity1.short_code,
'Q1': 'new_name',
'Q2': '430',
'Q3': 'b'
}
survey_response_to_edit = SurveyResponse.get(
self.manager, saved_response.survey_response_id)
edited_response = survey_response_service.edit_survey(
'CL1', new_values, [], survey_response_to_edit)
self.assertTrue(edited_response.success)
self.assertEqual(0, edited_response.errors.__len__())
self.assertIsNotNone(edited_response.datarecord_id)
self.assertIsNotNone(edited_response.survey_response_id)
self.assertEqual(test_data.entity_type, edited_response.entity_type)
self.assertEqual('CL1', edited_response.form_code)
self.assertDictEqual(
OrderedDict([('Q1', 'new_name'), ('Q3', ['YELLOW']), ('Q2', 430),
('ID', u'1')]), edited_response.processed_data)
self.assertNotEquals(saved_response.datarecord_id,
edited_response.datarecord_id)
old_data_record = DataRecord.get(self.manager,
saved_response.datarecord_id)
self.assertTrue(old_data_record.voided)
new_data_record = DataRecord.get(self.manager,
edited_response.datarecord_id)
self.assertFalse(new_data_record.voided)
edited_survey_response = SurveyResponse.get(
self.manager, edited_response.survey_response_id)
self.assertEquals(1,
len(edited_survey_response._doc.data_record_history))
self.assertEquals(old_data_record.id,
edited_survey_response._doc.data_record_history[0])
def edit_survey_response(self,
request,
survey_response,
owner_id,
additional_feed_dictionary=None,
logger=None):
assert request is not None
form_code, values = self._parse(request.message)
service = SurveyResponseService(self.dbm,
logger,
feeds_dbm=self.feeds_dbm,
admin_id=self.admin_id)
return service.edit_survey(form_code, values, [], survey_response,
additional_feed_dictionary, owner_id)
def update_survey_response(self,
request,
logger=None,
survey_response=None,
additional_feed_dictionary=None):
assert request is not None
form_code, values = self._parse(request.message)
media_submission_service = MediaSubmissionService(self.dbm,
request.media,
form_code,
is_update=True)
media_files = media_submission_service.create_media_documents(values)
service = SurveyResponseService(self.dbm, logger, self.feeds_dbm)
response = service.edit_survey(form_code, values, [], survey_response,
additional_feed_dictionary)
self._delete_removed_attachments(request, survey_response.id,
media_submission_service)
thumbnails = self._add_new_attachments(media_files, survey_response.id)
media_submission_service.create_preview_documents(thumbnails)
return response
class TestSurveyResponseService(TestCase):
def setUp(self):
self.dbm = Mock(spec=DatabaseManager)
self.survey_response_service = SurveyResponseService(self.dbm)
def form_model(self):
question1 = UniqueIdField(unique_id_type='clinic', name="entity_question", code="q1",
label="What is associated entity",
)
question2 = IntegerField(name="question1_Name", code="q2", label="What is your name",
constraints=[NumericRangeConstraint(min=10, max=100)])
return FormModel(self.dbm, name="aids", label="Aids form_model",
form_code="aids", fields=[question1, question2])
# TODO : Need to add validations for incompatible data types -> eg. string for number. This validation is hadled outside the service for now.
def test_edit_survey_response_when_fields_constraints_are_not_satisfied(self):
survey_response = Mock(spec=SurveyResponse)
with patch('mangrove.transport.services.survey_response_service.by_short_code') as get_reporter:
with patch('mangrove.datastore.entity.by_short_code') as get_entity:
with patch(
'mangrove.transport.services.survey_response_service.get_form_model_by_code') as get_form_model:
get_reporter.return_value = Mock(spec=Entity)
get_entity.return_value = Mock(spec=Entity)
get_form_model.return_value = self.form_model()
transport_info = TransportInfo('web', 'src', 'dest')
values = {'form_code': 'aids', 'q1': 'a1', 'q2': '200'}
request = Request(values, transport_info)
response = self.survey_response_service.edit_survey('aids', values, [], survey_response)
self.assertFalse(response.success)
self.assertEquals('aids', response.form_code)
self.assertEquals(OrderedDict([('q2', u'Answer 200 for question q2 is greater than allowed.')]),
response.errors)
self.assertEquals(['clinic'], response.entity_type)
self.assertEquals(OrderedDict([('q1', 'a1')]), response.processed_data)
self.assertIsNotNone(response.survey_response_id)
assert not survey_response.update.called
def test_survey_response_event_created_when_survey_response_created(self):
manager = Mock(spec=DatabaseManager)
feed_manager = Mock(spec=DatabaseManager)
project = Mock(spec=Project)
survey_response_service = SurveyResponseService(manager, feeds_dbm=feed_manager)
values = {'ID': 'short_code', 'Q1': 'name', 'Q2': '80', 'Q3': 'a'}
transport_info = TransportInfo('web', 'src', 'dest')
request = Request(values, transport_info)
additional_dictionary = {'project': {'name': 'someproject', 'status': 'active', 'id': 'someid'}}
with patch('mangrove.transport.services.survey_response_service.by_short_code') as get_reporter:
with patch(
'mangrove.transport.services.survey_response_service.get_form_model_by_code') as get_form_model_by_code:
with patch('mangrove.datastore.entity.by_short_code') as by_short_code:
with patch(
'mangrove.transport.services.survey_response_service.EnrichedSurveyResponseBuilder')as builder:
with patch("mangrove.form_model.form_submission.DataRecordDocument") as DataRecordDocumentMock:
with patch('mangrove.transport.services.survey_response_service.Project.from_form_model') as from_form_model:
get_reporter.return_value = Mock(spec=Entity)
builder.return_value = Mock(spec=EnrichedSurveyResponseBuilder)
entity_mock = MagicMock(spec=Entity)
entity_mock._doc = EntityDocument()
by_short_code.return_value = entity_mock
mock_form_model = MagicMock(spec=FormModel)
mock_form_model._dbm = manager
mock_form_model._doc = MagicMock()
mock_form_model._data = {}
mock_form_model.validate_submission.return_value = OrderedDict(values), OrderedDict('')
mock_form_model.is_entity_registration_form.return_value = False
mock_form_model.entity_questions = []
mock_form_model.entity_type = 'sometype'
get_form_model_by_code.return_value = mock_form_model
from_form_model.return_value = project
project.data_senders = []
survey_response_service.save_survey('CL1', values, [], transport_info, '',
additional_dictionary)
self.assertEquals(1, feed_manager._save_document.call_count)
def test_feeds_created_if_subject_not_found_for_a_submission(self):
manager = Mock(spec=DatabaseManager)
feed_manager = Mock(spec=DatabaseManager)
project = Mock(spec=Project)
values = {'ID': 'short_code', 'Q1': 'name', 'Q2': '80', 'Q3': 'a'}
transport_info = TransportInfo('web', 'src', 'dest')
request = Request(values, transport_info)
additional_dictionary = {'project': {'name': 'someproject', 'status': 'active', 'id': 'someid'}}
survey_response_service = SurveyResponseService(manager, feeds_dbm=feed_manager)
with patch('mangrove.transport.services.survey_response_service.by_short_code') as get_reporter:
with patch(
'mangrove.transport.services.survey_response_service.get_form_model_by_code') as get_form_model_by_code:
with patch('mangrove.datastore.entity.by_short_code') as by_short_code:
with patch(
'mangrove.transport.services.survey_response_service.DataFormSubmission') as data_form_submission:
with patch(
'mangrove.transport.services.survey_response_service.EnrichedSurveyResponseBuilder')as builder:
with patch('mangrove.transport.services.survey_response_service.Project.from_form_model') as from_form_model:
get_reporter.return_value = Mock(spec=Entity)
builder.return_value = Mock(spec=EnrichedSurveyResponseBuilder)
instance_mock = data_form_submission.return_value
type(instance_mock).is_valid = PropertyMock(return_value=True)
type(instance_mock).data_record_id = PropertyMock(return_value='sdsddsd')
instance_mock.save.side_effect = MangroveException("subject not found")
from_form_model.return_value = project
project.data_senders = []
by_short_code.return_value = Mock(spec=Entity)
mock_form_model = MagicMock(spec=FormModel)
mock_form_model._dbm = manager
mock_form_model._doc = MagicMock()
mock_form_model._data = {}
mock_form_model.validate_submission.return_value = values, ""
get_form_model_by_code.return_value = mock_form_model
try:
survey_response_service.save_survey('CL1', values, [], transport_info, '',
additional_dictionary)
self.fail('the subject not found exception should be propagated')
except MangroveException:
feed_manager._save_document.assert_called_once()
def test_response_has_no_error_when_feed_creation_fails(self):
manager = Mock(spec=DatabaseManager)
feed_manager = Mock(spec=DatabaseManager)
project = Mock(spec=Project)
values = {'ID': 'short_code', 'Q1': 'name', 'Q2': '80', 'Q3': 'a'}
transport_info = TransportInfo('web', 'src', 'dest')
request = Request(values, transport_info)
additional_dictionary = {'project': {'name': 'someproject', 'status': 'active', 'id': 'someid'}}
survey_response_service = SurveyResponseService(manager, feeds_dbm=feed_manager)
with patch('mangrove.transport.services.survey_response_service.by_short_code') as get_reporter:
with patch(
'mangrove.transport.services.survey_response_service.get_form_model_by_code') as get_form_model_by_code:
with patch('mangrove.datastore.entity.by_short_code') as by_short_code:
with patch(
'mangrove.transport.services.survey_response_service.DataFormSubmission') as data_form_submission:
with patch(
'mangrove.transport.services.survey_response_service.EnrichedSurveyResponseBuilder')as builder:
with patch('mangrove.transport.services.survey_response_service.Project.from_form_model') as from_form_model:
get_reporter.return_value = Mock(spec=Entity)
builder.feed_document.side_effect = Exception('Some error')
builder.return_value = builder
instance_mock = data_form_submission.return_value
type(instance_mock).is_valid = PropertyMock(return_value=True)
type(instance_mock).data_record_id = PropertyMock(return_value='sdsddsd')
type(instance_mock).errors = PropertyMock(return_value='')
by_short_code.return_value = Mock(spec=Entity)
mock_form_model = MagicMock(spec=FormModel)
mock_form_model._dbm = manager
mock_form_model._doc = MagicMock()
mock_form_model._data = {}
mock_form_model.validate_submission.return_value = values, ""
get_form_model_by_code.return_value = mock_form_model
from_form_model.return_value = project
project.data_senders = []
response = survey_response_service.save_survey('CL1', values, [], transport_info, '',
additional_dictionary)
self.assertFalse(response.errors)
self.assertTrue(response.feed_error_message)