class IndexManagementIntTest(IonIntegrationTestCase):
def setUp(self):
self._start_container()
self.container.start_rel_from_url('res/deploy/r2dm.yml')
self.ims_cli = IndexManagementServiceClient()
self.rr_cli = ResourceRegistryServiceClient()
self.index_name = 'test_index'
def test_create_elasticsearch_index(self):
index_name = self.index_name
ims_cli = self.ims_cli
rr_cli = self.rr_cli
options = SearchOptions()
options.attribute_match = ['test_field']
index_id = ims_cli.create_index(
name=index_name,
content_type=IndexManagementService.ELASTICSEARCH_INDEX,
options=options
)
index_result = self.rr_cli.read(index_id)
self.assertIsInstance(index_result,ElasticSearchIndex)
self.assertTrue(index_result.name == index_name)
#======================================
# Clean up
#======================================
rr_cli.delete(index_id)
def test_create_couchdb_index(self):
index_name = self.index_name
ims_cli = self.ims_cli
rr_cli = self.rr_cli
options = SearchOptions()
options.attribute_match = ['name']
index_id = ims_cli.create_index(
index_name,
content_type=IndexManagementService.COUCHDB_INDEX,
options=options,
datastore_name='fake',
view_name='fake/by_fake'
)
index_result = self.rr_cli.read(index_id)
self.assertIsInstance(index_result,CouchDBIndex)
self.assertTrue(index_result.name==index_name)
#======================================
# Clean up
#======================================
rr_cli.delete(index_id)
def test_read_index(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
index_name = self.index_name
index_res = Index(name=index_name)
index_id, _ = rr_cli.create(index_res)
index = ims_cli.read_index(index_id)
self.assertIsInstance(index,Index)
self.assertTrue(index.name==index_name)
rr_cli.delete(index_id)
def test_delete_index(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
index_name = self.index_name
index_res = Index(name=index_name)
index_id, _ = rr_cli.create(index_res)
ims_cli.delete_index(index_id)
with self.assertRaises(NotFound):
rr_cli.delete(index_id)
def test_update_index(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
index_name = self.index_name
index_res = Index(name=index_name)
index_id, _ = rr_cli.create(index_res)
index = ims_cli.read_index(index_id)
index.name = 'another'
ims_cli.update_index(index)
index = rr_cli.read(index_id)
self.assertTrue(index.name == 'another')
def test_find_indexes(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
index_name = self.index_name
#======================================
# Index Pool
#======================================
indexes = [
Index(name='first'),
Index(name='second'),
Index(name='third')
]
id_pool = list()
for index in indexes:
id_pool.append(rr_cli.create(index)[0])
index_id = ims_cli.find_indexes(index_name='second')
index = ims_cli.read_index(index_id)
self.assertTrue(index.name=='second')
#======================================
# Clean up
#======================================
for index_id in id_pool:
rr_cli.delete(index_id)
def test_create_collection(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
with self.assertRaises(BadRequest):
ims_cli.create_collection('failing_collection')
resources = [ Resource(), Resource(), Resource() ]
resources = [ rr_cli.create(i)[0] for i in resources ]
collection_id = ims_cli.create_collection('working_collection',resources)
collection = rr_cli.read(collection_id)
collection_resources = ims_cli.list_collection_resources(collection_id, id_only=True)
self.assertTrue(set(collection_resources) == set(resources), '%s != %s' % (set(collection_resources) , set(resources)))
def test_read_collection(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
collection = Collection(name='working_collection')
collection_id, _ = rr_cli.create(collection)
collection = ims_cli.read_collection(collection_id)
self.assertTrue(collection.name == 'working_collection')
def test_update_collection(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
collection = Collection(name='useful_collection')
collection_id, _ = rr_cli.create(collection)
collection = rr_cli.read(collection_id)
collection.name = 'nub'
ims_cli.update_collection(collection)
collection = rr_cli.read(collection_id)
self.assertTrue(collection.name=='nub')
def test_delete_collection(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
res = Resource()
res_id, rev = rr_cli.create(res)
collection_id = ims_cli.create_collection(name='test_collection', resources=[res_id])
ims_cli.delete_collection(collection_id)
with self.assertRaises(NotFound):
rr_cli.read(collection_id)
def test_list_collection_resources(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
#========================================
# Resource Pool
#========================================
resources = [ InformationResource(name='bean_counter'), InformationResource(name='lunar_rock'), InformationResource('aperature'), InformationResource('lemons') ]
resources = [ rr_cli.create(i)[0] for i in resources ]
collection = Collection(name='park_bench')
collection_id = ims_cli.create_collection(name='park_bench', resources=resources)
retval = ims_cli.list_collection_resources(collection_id, id_only=True)
retval.sort()
resources.sort()
self.assertTrue(retval == resources, '%s != %s' %(retval , resources))
def test_find_collection(self):
res_id, _ = self.rr_cli.create(Resource(name='test_res'))
collection_id = self.ims_cli.create_collection('test', [res_id])
retval = self.ims_cli.find_collection(collection_name='test')
self.assertTrue(retval[0] == collection_id)
retval = self.ims_cli.find_collection(resource_ids=[res_id])
self.assertTrue(retval[0] == collection_id)
class IndexManagementIntTest(IonIntegrationTestCase):
def setUp(self):
self._start_container()
self.container.start_rel_from_url("res/deploy/r2dm.yml")
self.ims_cli = IndexManagementServiceClient()
self.rr_cli = ResourceRegistryServiceClient()
self.index_name = "test_index"
def test_create_elasticsearch_index(self):
index_name = self.index_name
ims_cli = self.ims_cli
rr_cli = self.rr_cli
options = SearchOptions()
options.attribute_match = ["test_field"]
index_id = ims_cli.create_index(
name=index_name, content_type=IndexManagementService.ELASTICSEARCH_INDEX, options=options
)
index_result = self.rr_cli.read(index_id)
self.assertIsInstance(index_result, ElasticSearchIndex)
self.assertTrue(index_result.name == index_name)
# ======================================
# Clean up
# ======================================
rr_cli.delete(index_id)
def test_create_couchdb_index(self):
index_name = self.index_name
ims_cli = self.ims_cli
rr_cli = self.rr_cli
options = SearchOptions()
options.attribute_match = ["name"]
index_id = ims_cli.create_index(
index_name,
content_type=IndexManagementService.COUCHDB_INDEX,
options=options,
datastore_name="fake",
view_name="fake/by_fake",
)
index_result = self.rr_cli.read(index_id)
self.assertIsInstance(index_result, CouchDBIndex)
self.assertTrue(index_result.name == index_name)
# ======================================
# Clean up
# ======================================
rr_cli.delete(index_id)
def test_read_index(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
index_name = self.index_name
index_res = Index(name=index_name)
index_id, _ = rr_cli.create(index_res)
index = ims_cli.read_index(index_id)
self.assertIsInstance(index, Index)
self.assertTrue(index.name == index_name)
rr_cli.delete(index_id)
def test_delete_index(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
index_name = self.index_name
index_res = Index(name=index_name)
index_id, _ = rr_cli.create(index_res)
ims_cli.delete_index(index_id)
with self.assertRaises(NotFound):
rr_cli.delete(index_id)
def test_update_index(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
index_name = self.index_name
index_res = Index(name=index_name)
index_id, _ = rr_cli.create(index_res)
index = ims_cli.read_index(index_id)
index.name = "another"
ims_cli.update_index(index)
index = rr_cli.read(index_id)
self.assertTrue(index.name == "another")
def test_find_indexes(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
index_name = self.index_name
# ======================================
# Index Pool
# ======================================
indexes = [Index(name="first"), Index(name="second"), Index(name="third")]
id_pool = list()
for index in indexes:
id_pool.append(rr_cli.create(index)[0])
index_id = ims_cli.find_indexes(index_name="second")
index = ims_cli.read_index(index_id)
self.assertTrue(index.name == "second")
# ======================================
# Clean up
# ======================================
for index_id in id_pool:
rr_cli.delete(index_id)
def test_list_indexes(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
index_name = self.index_name
# ======================================
# Index Pool
# ======================================
indexes = [Index(name="first"), Index(name="second"), Index(name="third")]
id_pool = list()
for index in indexes:
id_pool.append(rr_cli.create(index)[0])
names = set(ims_cli.list_indexes().keys())
self.assertTrue(names == set(["first", "second", "third"]))
# ======================================
# Clean up
# ======================================
for index_id in id_pool:
rr_cli.delete(index_id)
def test_create_collection(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
with self.assertRaises(BadRequest):
ims_cli.create_collection("failing_collection")
resources = [Resource(), Resource(), Resource()]
resources = [rr_cli.create(i)[0] for i in resources]
collection_id = ims_cli.create_collection("working_collection", resources)
collection = rr_cli.read(collection_id)
collection_resources = ims_cli.list_collection_resources(collection_id, id_only=True)
self.assertTrue(
set(collection_resources) == set(resources), "%s != %s" % (set(collection_resources), set(resources))
)
def test_read_collection(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
collection = Collection(name="working_collection")
collection_id, _ = rr_cli.create(collection)
collection = ims_cli.read_collection(collection_id)
self.assertTrue(collection.name == "working_collection")
def test_update_collection(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
collection = Collection(name="useful_collection")
collection_id, _ = rr_cli.create(collection)
collection = rr_cli.read(collection_id)
collection.name = "nub"
ims_cli.update_collection(collection)
collection = rr_cli.read(collection_id)
self.assertTrue(collection.name == "nub")
def test_delete_collection(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
res = Resource()
res_id, rev = rr_cli.create(res)
collection_id = ims_cli.create_collection(name="test_collection", resources=[res_id])
ims_cli.delete_collection(collection_id)
with self.assertRaises(NotFound):
rr_cli.read(collection_id)
def test_list_collection_resources(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
# ========================================
# Resource Pool
# ========================================
resources = [
InformationResource(name="bean_counter"),
InformationResource(name="lunar_rock"),
InformationResource("aperature"),
InformationResource("lemons"),
]
resources = [rr_cli.create(i)[0] for i in resources]
collection = Collection(name="park_bench")
collection_id = ims_cli.create_collection(name="park_bench", resources=resources)
retval = ims_cli.list_collection_resources(collection_id, id_only=True)
retval.sort()
resources.sort()
self.assertTrue(retval == resources, "%s != %s" % (retval, resources))
def test_find_collection(self):
res_id, _ = self.rr_cli.create(Resource(name="test_res"))
collection_id = self.ims_cli.create_collection("test", [res_id])
retval = self.ims_cli.find_collection(collection_name="test")
self.assertTrue(retval[0] == collection_id)
retval = self.ims_cli.find_collection(resource_ids=[res_id])
self.assertTrue(retval[0] == collection_id)
class IndexManagementIntTest(IonIntegrationTestCase):
def setUp(self):
self._start_container()
self.container.start_rel_from_url('res/deploy/r2dm.yml')
self.ims_cli = IndexManagementServiceClient()
self.rr_cli = ResourceRegistryServiceClient()
self.index_name = 'test_index'
def test_create_datastore_index(self):
index_name = self.index_name
ims_cli = self.ims_cli
rr_cli = self.rr_cli
options = SearchOptions()
options.attribute_match = ['test_field']
index_id = ims_cli.create_index(
name=index_name,
content_type=IndexManagementService.DATASTORE_INDEX,
options=options)
index_result = self.rr_cli.read(index_id)
self.assertIsInstance(index_result, ElasticSearchIndex)
self.assertTrue(index_result.name == index_name)
#======================================
# Clean up
#======================================
rr_cli.delete(index_id)
def test_read_index(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
index_name = self.index_name
index_res = Index(name=index_name)
index_id, _ = rr_cli.create(index_res)
index = ims_cli.read_index(index_id)
self.assertIsInstance(index, Index)
self.assertTrue(index.name == index_name)
rr_cli.delete(index_id)
def test_delete_index(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
index_name = self.index_name
index_res = Index(name=index_name)
index_id, _ = rr_cli.create(index_res)
ims_cli.delete_index(index_id)
with self.assertRaises(NotFound):
rr_cli.delete(index_id)
def test_update_index(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
index_name = self.index_name
index_res = Index(name=index_name)
index_id, _ = rr_cli.create(index_res)
index = ims_cli.read_index(index_id)
index.name = 'another'
ims_cli.update_index(index)
index = rr_cli.read(index_id)
self.assertTrue(index.name == 'another')
def test_find_indexes(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
index_name = self.index_name
#======================================
# Index Pool
#======================================
indexes = [
Index(name='first'),
Index(name='second'),
Index(name='third')
]
id_pool = list()
for index in indexes:
id_pool.append(rr_cli.create(index)[0])
index_id = ims_cli.find_indexes(index_name='second')
index = ims_cli.read_index(index_id)
self.assertTrue(index.name == 'second')
#======================================
# Clean up
#======================================
for index_id in id_pool:
rr_cli.delete(index_id)
def test_create_collection(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
with self.assertRaises(BadRequest):
ims_cli.create_collection('failing_collection')
resources = [Resource(), Resource(), Resource()]
resources = [rr_cli.create(i)[0] for i in resources]
collection_id = ims_cli.create_collection('working_collection',
resources)
collection = rr_cli.read(collection_id)
collection_resources = ims_cli.list_collection_resources(collection_id,
id_only=True)
self.assertTrue(
set(collection_resources) == set(resources),
'%s != %s' % (set(collection_resources), set(resources)))
def test_read_collection(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
collection = Collection(name='working_collection')
collection_id, _ = rr_cli.create(collection)
collection = ims_cli.read_collection(collection_id)
self.assertTrue(collection.name == 'working_collection')
def test_update_collection(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
collection = Collection(name='useful_collection')
collection_id, _ = rr_cli.create(collection)
collection = rr_cli.read(collection_id)
collection.name = 'nub'
ims_cli.update_collection(collection)
collection = rr_cli.read(collection_id)
self.assertTrue(collection.name == 'nub')
def test_delete_collection(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
res = Resource()
res_id, rev = rr_cli.create(res)
collection_id = ims_cli.create_collection(name='test_collection',
resources=[res_id])
ims_cli.delete_collection(collection_id)
with self.assertRaises(NotFound):
rr_cli.read(collection_id)
def test_list_collection_resources(self):
ims_cli = self.ims_cli
rr_cli = self.rr_cli
#========================================
# Resource Pool
#========================================
resources = [
InformationResource(name='bean_counter'),
InformationResource(name='lunar_rock'),
InformationResource('aperature'),
InformationResource('lemons')
]
resources = [rr_cli.create(i)[0] for i in resources]
collection = Collection(name='park_bench')
collection_id = ims_cli.create_collection(name='park_bench',
resources=resources)
retval = ims_cli.list_collection_resources(collection_id, id_only=True)
retval.sort()
resources.sort()
self.assertTrue(retval == resources, '%s != %s' % (retval, resources))
def test_find_collection(self):
res_id, _ = self.rr_cli.create(Resource(name='test_res'))
collection_id = self.ims_cli.create_collection('test', [res_id])
retval = self.ims_cli.find_collection(collection_name='test')
self.assertTrue(retval[0] == collection_id)
retval = self.ims_cli.find_collection(resource_ids=[res_id])
self.assertTrue(retval[0] == collection_id)
