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()
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()