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