def test_calls_force_string(self, mock_force_strings): m = get_valid_questionnaire() mock_force_strings.return_value = {} put_questionnaire_data([m]) mock_force_strings.assert_called_once_with({ 'name': {}, 'definition': {'en': ''}})
def test_calls_force_string(self, mock_force_strings): m = get_valid_questionnaire() mock_force_strings.return_value = {} put_questionnaire_data([m]) mock_force_strings.assert_called_once_with({ 'name': {}, 'definition': { 'en': '' } })
def test_calls_bulk(self, mock_bulk, mock_es): mock_bulk.return_value = 0, [] questionnaire = get_valid_questionnaire() put_questionnaire_data([questionnaire]) source = dict(QuestionnaireSerializer(questionnaire).data) source['filter_data'] = {} source['list_data']['country'] = None data = [{ '_index': '{}sample_2015'.format(settings.ES_INDEX_PREFIX), '_type': 'questionnaire', '_id': questionnaire.id, '_source': source, }] mock_bulk.assert_called_once_with(mock_es, data)
def test_calls_bulk(self, mock_bulk, mock_es): mock_bulk.return_value = 0, [] questionnaire = get_valid_questionnaire() put_questionnaire_data([questionnaire]) source = dict(QuestionnaireSerializer( questionnaire ).data) source['filter_data'] = {} source['list_data']['country'] = None data = [{ '_index': '{}sample_2015'.format(settings.ES_INDEX_PREFIX), '_type': 'questionnaire', '_id': questionnaire.id, '_source': source, }] mock_bulk.assert_called_once_with(mock_es, data)
def create_temp_indices(configuration_list: list): """ For each index, create the index and update it with the questionnaires of the corresponding configuration as found in the database. Make sure to override the settings and use a custom prefix for the indices. Args: ``configuration_list`` (list): A list of tuples containing the configurations for which a ES index shall be created and consisting of - [0]: code of the configuration - [1]: edition of the configuration """ for code, edition in configuration_list: mappings = get_mappings() create_or_update_index(get_configuration(code=code, edition=edition), mappings) put_questionnaire_data(Questionnaire.with_status.public().filter( configuration__code=code))
def create_temp_indices(configuration_list: list): """ For each index, create the index and update it with the questionnaires of the corresponding configuration as found in the database. Make sure to override the settings and use a custom prefix for the indices. Args: ``configuration_list`` (list): A list of tuples containing the configurations for which a ES index shall be created and consisting of - [0]: code of the configuration - [1]: edition of the configuration """ for code, edition in configuration_list: mappings = get_mappings() create_or_update_index( get_configuration(code=code, edition=edition), mappings) put_questionnaire_data( Questionnaire.with_status.public().filter(configuration__code=code) )
def test_returns_values(self, mock_bulk, mock_es): mock_bulk.return_value = 'foo', 'bar' actions, errors = put_questionnaire_data([]) self.assertEqual(actions, 'foo') self.assertEqual(errors, 'bar')
def test_calls_indices_refresh(self, mock_es): put_questionnaire_data([]) mock_es.indices.refresh.assert_called_once_with(index='')
def test_calls_get_alias(self, mock_get_alias, mock_es, mock_bulk): mock_get_alias.return_value = '' mock_bulk.return_value = None, None put_questionnaire_data([get_valid_questionnaire()]) mock_get_alias.assert_called_once()