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 == ""