def test_should_return_AllData_for_pro_sms_account(self):
user = get_project_manager()
request = MagicMock()
organization = MagicMock(spec=Organization)
organization.is_pro_sms = True
with patch('datawinners.alldata.helper.get_organization'
) as mock_get_organization:
mock_get_organization.return_value = organization
request.user = user
assert helper.get_page_heading(request) == 'Questionnaires & Polls'
def test_should_return_project_id_name_map_for_pm_in_alphabetical_order_of_names(self,
mock_get_questionnaires_for_user):
dbm = MagicMock(spec=DatabaseManager)
user_mock = get_project_manager()
user_mock.id = 3
all_projects_for_user = [{'_id': 'd3456cc', 'name': 'test questionnaire', 'is_project_manager': True},
{'_id': '256cc', 'name': '2nd questionnaire', 'is_project_manager': True}]
mock_get_questionnaires_for_user.return_value = all_projects_for_user
project_id_name_map = get_project_id_name_map_for_user(dbm, user_mock)
expected_project_id_map = OrderedDict([('256cc', '2nd questionnaire'), ('d3456cc', 'test questionnaire')])
self.assertDictEqual(expected_project_id_map, project_id_name_map)
def test_should_return_all_projects_for_user_as_project_manager(
self, mock_get_all_projects, mock_get_questionnaires_for_user):
user = get_project_manager()
all_projects_for_user = [{
'_id': 'd3456cc',
'name': 'test questionnaire',
'is_project_manager': True
}, {
'_id': '256cc',
'name': '2nd questionnaire',
'is_project_manager': True
}]
questionnaires_as_ds = [{
'value': {
'_id': 'd3456cc',
'name': 'test questionnaire'
}
}, {
'value': {
'_id': 'ds_question',
'name': 'Data sender questionnaire'
}
}]
expected_questionnaires = [{
'_id': 'd3456cc',
'name': 'test questionnaire',
'is_project_manager': True
}, {
'_id': '256cc',
'name': '2nd questionnaire',
'is_project_manager': True
}, {
'_id': 'ds_question',
'name': 'Data sender questionnaire'
}]
mock_get_all_projects.return_value = questionnaires_as_ds
mock_get_questionnaires_for_user.return_value = all_projects_for_user
questionnaires_to_display = helper.get_all_project_for_user(user)
self.assertEqual(len(questionnaires_to_display), 3)
self.assertEqual(questionnaires_to_display, expected_questionnaires)
def test_should_return_enabled_and_display_for_other(self):
user = get_project_manager()
disabled, hide = helper.get_visibility_settings_for(user)
assert hide == ""
assert disabled == ""