Python IndexManagementService.find_indexes方法代码示例

# 需要导入模块: from ion.services.dm.inventory.index_management_service import IndexManagementService [as 别名]
# 或者: from ion.services.dm.inventory.index_management_service.IndexManagementService import find_indexes [as 别名]
class IndexManagementUnitTest(PyonTestCase):
    def setUp(self):
        mock_clients = self._create_service_mock("index_management")
        self.index_management = IndexManagementService()
        self.index_management.clients = mock_clients

        self.rr_create = mock_clients.resource_registry.create
        self.rr_read = mock_clients.resource_registry.read
        self.rr_update = mock_clients.resource_registry.update
        self.rr_delete = mock_clients.resource_registry.delete
        self.rr_find_resources = mock_clients.resource_registry.find_resources
        self.rr_find_assocs = mock_clients.resource_registry.find_associations
        self.rr_find_subj = mock_clients.resource_registry.find_subjects
        self.rr_find_obj = mock_clients.resource_registry.find_objects
        self.rr_delete_assoc = mock_clients.resource_registry.delete_association

        self.get_datastore = Mock()
        self.db_create = Mock()

        self.get_datastore.return_value = DotDict({"datastore_name": "test_datastore"})
        self.index_management.container = DotDict({"datastore_manager": DotDict({"get_datastore": self.get_datastore})})
        self.index_management.elasticsearch_host = "notarealhost"
        self.index_management.elasticsearch_port = 9000
        self.index_name = "test_index"

    def test_create_index(self):
        """
        test_create_index
        Unit test for basic creation of an index
        """

        # Mocks
        self.rr_create.return_value = ("index_id", "rev")
        self.rr_find_resources.return_value = ([], [])

        retval = self.index_management.create_index(
            name="mock", content_type=IndexManagementService.ELASTICSEARCH_INDEX, options="ugh"
        )

        self.assertTrue(retval == "index_id", "invalid return value: %s" % retval)
        self.assertTrue(self.rr_create.called)

        retval = self.index_management.create_index(name="argh", content_type=IndexManagementService.COUCHDB_INDEX)

        self.assertTrue(retval == "index_id", "invalid return value: %s" % retval)

        with self.assertRaises(BadRequest):
            self.index_management.create_index(name="another", content_type="not_listed")

    def test_dup_index(self):
        # Mocks
        self.rr_find_resources.return_value = ([1], [1])

        # Execution
        with self.assertRaises(BadRequest):
            self.index_management.create_index("mock_index_id")

    def test_read_index(self):
        # mocks
        return_obj = dict(mock="mock")
        self.rr_read.return_value = return_obj

        # execution
        retval = self.index_management.read_index("mock_index_id")

        # assertions
        self.assertEquals(return_obj, retval, "The resource should be returned.")

    def test_update_index(self):
        with self.assertRaises(BadRequest):
            self.index_management.update_index()
        with self.assertRaises(BadRequest):
            self.index_management.update_index("hi")
        self.index_management.update_index(Index())

    def test_delete_index(self):

        self.index_management.delete_index("index_id")
        self.rr_delete.assert_called_with("index_id")

    def test_list_indexes(self):
        # Mocks
        self.rr_find_resources.return_value = (
            [
                DotDict({"_id": "1", "name": "1"}),
                DotDict({"_id": "2", "name": "2"}),
                DotDict({"_id": "3", "name": "3"}),
                DotDict({"_id": "4", "name": "4"}),
            ],
            [1, 2, 3, 4],
        )

        # Execution
        retval = self.index_management.list_indexes()

        # Assertions
        self.assertTrue(retval == {"1": "1", "2": "2", "3": "3", "4": "4"}, "Index mismatch")

    def test_find_indexes(self):
        self.index_management.list_indexes = Mock()
#.........这里部分代码省略.........

# 需要导入模块: from ion.services.dm.inventory.index_management_service import IndexManagementService [as 别名]
# 或者: from ion.services.dm.inventory.index_management_service.IndexManagementService import find_indexes [as 别名]

#.........这里部分代码省略.........
        self.assertEquals(return_obj, retval, 'The resource should be returned.')


    def test_update_index(self):
        with self.assertRaises(BadRequest):
            self.index_management.update_index()
        with self.assertRaises(BadRequest):
            self.index_management.update_index('hi')
        self.index_management.update_index(Index())



    def test_delete_index(self):

        self.index_management.delete_index('index_id')
        self.rr_delete.assert_called_with('index_id')



    def test_list_indexes(self):
        # Mocks
        self.rr_find_resources.return_value = ([
                DotDict({'_id':'1','name':'1'}),
                DotDict({'_id':'2','name':'2'}),
                DotDict({'_id':'3','name':'3'}),
                DotDict({'_id':'4','name':'4'})
            ],[1,2,3,4])

        # Execution
        retval = self.index_management.list_indexes()

        # Assertions
        self.assertTrue(retval == {'1':'1','2':'2','3':'3','4':'4'}, 'Index mismatch')
    
    def test_find_indexes(self):
        self.index_management.list_indexes=Mock()
        self.index_management.list_indexes.return_value = {'index_name':'1'}
        retval = self.index_management.find_indexes('index_name')
        self.assertTrue(retval=='1')
        self.index_management.list_indexes.return_value = {}
        retval = self.index_management.find_indexes('index_name')
        self.assertTrue(retval==None)

    def test_create_collection(self):
        self.rr_create.return_value = 'collection_id', 'rev'

        self.rr_find_resources.return_value = ([0],[0])
        with self.assertRaises(BadRequest):
            self.index_management.create_collection('test',[0])

        self.rr_find_resources.return_value = ([],[])

        with self.assertRaises(BadRequest):
            self.index_management.create_collection('test',[])

        retval = self.index_management.create_collection('test',[0])
        self.assertTrue(retval=='collection_id')
            
    def test_read_collection(self):
        self.rr_read.return_value = 'retval'
        retval = self.index_management.read_collection('test')
        self.assertTrue(retval=='retval')

    def test_update_collection(self):
        with self.assertRaises(BadRequest):
            ind = Index()
            self.index_management.update_collection(ind)
        self.index_management.update_collection(Collection())
        self.assertTrue(self.rr_update.called)

    def test_delete_collection(self):
        self.rr_find_assocs.return_value = ['assoc']

        retval = self.index_management.delete_collection('collection_id')
        self.assertTrue(retval)
        self.rr_delete.assert_called_once_with('collection_id')
        self.rr_delete_assoc.assert_called_once_with('assoc')

    def test_list_collection_resources(self):
        self.rr_find_obj.return_value = (['test_id'],[''])


        result1 = self.index_management.list_collection_resources('collection_id', id_only=True)
        self.assertTrue(result1 == ['test_id'])
    
    def test_find_collection(self):
        self.rr_find_resources.return_value = (['test'],[])

        retval = self.index_management.find_collection(collection_name='hi')
        self.assertTrue(retval == ['test'] , '%s' % retval)

        fake_collection = Collection(resources=['test_res_id'])
        fake_assoc = Association(s='test_id')
        self.rr_find_assocs.return_value = [fake_assoc]

        retval = self.index_management.find_collection(resource_ids=['test_res_id'])
        self.assertTrue(retval == ['test_id'], '%s' % retval)

        with self.assertRaises(BadRequest):
            self.index_management.find_collection()