def test_bad_op(self, mock_es):
config = CFG
config.system.elasticsearch = True
config.server.elasticsearch.host = ''
config.server.elasticsearch.port = ''
config.op = 'not_real'
ibs = IndexBootStrap()
ibs.CFG = config
with self.assertRaises(BadRequest):
ibs.on_start()
def test_bad_op(self, mock_es):
config = CFG
config.system.elasticsearch=True
config.server.elasticsearch.host = ''
config.server.elasticsearch.port = ''
config.op = 'not_real'
ibs = IndexBootStrap()
ibs.CFG = config
with self.assertRaises(BadRequest):
ibs.on_start()
def test_clean_bootstrap(self, mock_es):
config = CFG
config.system.elasticsearch=True
config.server.elasticsearch.host = ''
config.server.elasticsearch.port = ''
config.op = 'clean_bootstrap'
ibs = IndexBootStrap()
ibs.CFG = config
ibs.index_bootstrap = Mock()
ibs.on_start()
self.assertTrue(ibs.index_bootstrap.called)
def test_index_bootstrap(self, mock_es, ims_cli):
#---------------------------------------------
# Mocks
#---------------------------------------------
mock_es().index_create.return_value = {'ok':True, 'status':200}
mock_es().raw.return_value = {'ok':True, 'status':200}
mock_es().river_couchdb_create.return_value = {'ok':True, 'status':200}
db = DotDict()
db.datastore_name = 'test'
db.server.test.create = Mock()
container = DotDict()
container.datastore_manager.get_datastore = Mock()
container.datastore_manager.get_datastore.return_value = db
config = CFG
config.system.force_clean=False
config.system.elasticsearch=True
config.server.elasticsearch.host = ''
config.server.elasticsearch.port = ''
config.op = 'index_bootstrap'
#---------------------------------------------
# Execution
#---------------------------------------------
ibs = IndexBootStrap()
ibs.CFG = config
ibs.container = container
ibs.on_start()
index_count = len(STD_INDEXES) + len(EDGE_INDEXES) + 1 # for _river
self.assertTrue(mock_es().index_create.call_count == index_count, '(%s != %s) Improper number of indices created' %(mock_es().index_create.call_count , index_count))
river_count = len(STD_INDEXES) + len(EDGE_INDEXES)
self.assertTrue(mock_es().river_couchdb_create.call_count == river_count, 'Improper number of rivers created')
total_count = len(STD_INDEXES) + len(COUCHDB_INDEXES) + len(EDGE_INDEXES)
self.assertTrue(ims_cli().create_index.call_count == total_count, 'Improper number of index resources created')
def test_clean_bootstrap(self, mock_es):
config = CFG
config.system.elasticsearch = True
config.server.elasticsearch.host = ''
config.server.elasticsearch.port = ''
config.op = 'clean_bootstrap'
ibs = IndexBootStrap()
ibs.CFG = config
ibs.index_bootstrap = Mock()
ibs.on_start()
self.assertTrue(ibs.index_bootstrap.called)
def test_index_bootstrap(self, mock_es, ims_cli):
#---------------------------------------------
# Mocks
#---------------------------------------------
mock_es().index_create.return_value = {'ok': True, 'status': 200}
mock_es().raw.return_value = {'ok': True, 'status': 200}
mock_es().river_couchdb_create.return_value = {
'ok': True,
'status': 200
}
db = DotDict()
db.datastore_name = 'test'
db.server.test.create = Mock()
container = DotDict()
container.datastore_manager.get_datastore = Mock()
container.datastore_manager.get_datastore.return_value = db
config = CFG
config.system.force_clean = False
config.system.elasticsearch = True
config.server.elasticsearch.host = ''
config.server.elasticsearch.port = ''
config.op = 'index_bootstrap'
#---------------------------------------------
# Execution
#---------------------------------------------
ibs = IndexBootStrap()
ibs.CFG = config
ibs.container = container
ibs.on_start()
index_count = len(STD_INDEXES) + len(EDGE_INDEXES) + 1 # for _river
self.assertTrue(
mock_es().index_create.call_count == index_count,
'(%s != %s) Improper number of indices created' %
(mock_es().index_create.call_count, index_count))
river_count = len(STD_INDEXES) + len(EDGE_INDEXES)
self.assertTrue(
mock_es().river_couchdb_create.call_count == river_count,
'Improper number of rivers created')
total_count = len(STD_INDEXES) + len(COUCHDB_INDEXES) + len(
EDGE_INDEXES)
self.assertTrue(ims_cli().create_index.call_count == total_count,
'Improper number of index resources created')
