def test_recreate_existing_managed_index(self, es):
es.indices.exists.return_value = True
index = track.Index(name="test-index",
auto_managed=True,
types=[
track.Type(
name="test-type",
mapping={"test-type": "empty-for-test"})
])
index_settings = {"index.number_of_replicas": 0}
expected_body = {
"settings": {
"index.number_of_replicas": 0
},
"mappings": {
"test-type": "empty-for-test"
}
}
driver.setup_index(es, index, index_settings)
es.indices.exists.assert_called_with(index="test-index")
es.indices.delete.assert_called_with(index="test-index")
es.indices.create.assert_called_with(index="test-index",
body=expected_body)
def test_recreate_existing_managed_index(self, es):
es.indices.exists.return_value = True
index = track.Index(name="test-index",
auto_managed=True,
types=[track.Type(name="test-type", mapping_file=['{"mapping": "empty-for-test"}'])])
index_settings = {
"index.number_of_replicas": 0
}
driver.setup_index(es, index, index_settings, source=io.StringAsFileSource)
es.indices.exists.assert_called_with(index="test-index")
es.indices.delete.assert_called_with(index="test-index")
es.indices.create.assert_called_with(index="test-index", body=index_settings)
es.indices.put_mapping.assert_called_with(index="test-index", doc_type="test-type", body={"mapping": "empty-for-test"})
def test_recreate_existing_managed_index(self, es):
es.indices.exists.return_value = True
index = track.Index(name="test-index",
auto_managed=True,
types=[track.Type(name="test-type", mapping={"test-type": "empty-for-test"})])
index_settings = {
"index.number_of_replicas": 0
}
expected_body = {
"settings": {
"index.number_of_replicas": 0
},
"mappings": {
"test-type": "empty-for-test"
}
}
driver.setup_index(es, index, index_settings)
es.indices.exists.assert_called_with(index="test-index")
es.indices.delete.assert_called_with(index="test-index")
es.indices.create.assert_called_with(index="test-index", body=expected_body)
def test_setup_auto_managed_index(self, es):
es.indices.exists.return_value = False
index = track.Index(name="test-index",
auto_managed=True,
types=[track.Type(name="test-type", mapping_file=['{"test-type": "empty-for-test"}'])])
index_settings = {
"index.number_of_replicas": 0
}
expected_body = {
"settings": {
"index.number_of_replicas": 0
},
"mappings": {
"test-type": "empty-for-test"
}
}
driver.setup_index(es, index, index_settings, source=io.StringAsFileSource)
es.indices.exists.assert_called_with(index="test-index")
es.indices.delete.assert_not_called()
es.indices.create.assert_called_with(index="test-index", body=expected_body)
def test_do_not_change_manually_managed_index(self, es):
index = track.Index(name="test-index", auto_managed=False, types=[])
driver.setup_index(es, index, index_settings=None)
es.assert_not_called()
def test_do_not_change_manually_managed_index(self, es):
index = track.Index(name="test-index", auto_managed=False, types=[])
driver.setup_index(es, index, index_settings=None)
es.assert_not_called()
