예제 #1
0
    def test01_operations(self):
        dbm = MgmtDatabaseManager('localhost')
        mgmt_api = ManagementEngineAPI(dbm)

        is_init = mgmt_api.is_initialized()
        self.assertEqual(is_init, False)

        mgmt_api.initialize(KS_PASSWD)
        
        is_init = mgmt_api.is_initialized()
        self.assertEqual(is_init, True)
예제 #2
0
    def test00_init(self):
        with self.assertRaises(MEDatabaseException):
            dbm = MgmtDatabaseManager('some-host-name')

        cl = MongoClient('localhost')
        cl.drop_database('test_fabnet_mgmt_db')
        MgmtDatabaseManager.MGMT_DB_NAME = 'test_fabnet_mgmt_db'

        dbm = MgmtDatabaseManager('localhost')
        mgmt_api = ManagementEngineAPI(dbm)

        with self.assertRaises(MEAuthException):
            mgmt_api.get_cluster_config(None)

        is_init = mgmt_api.is_initialized()
        self.assertEqual(is_init, True)
        
        with self.assertRaises(MEAuthException):
            mgmt_api.authenticate('test', 'test')
        with self.assertRaises(MEAuthException):
            mgmt_api.authenticate('admin', 'test')

        session_id = mgmt_api.authenticate('admin', 'admin')
        
        methods = mgmt_api.get_allowed_methods(session_id)
        self.assertTrue('create_user' in methods)

        user_info = mgmt_api.get_user_info(session_id, 'admin')
        self.assertEqual(user_info[DBK_USERNAME], 'admin')
        self.assertEqual(user_info[DBK_ROLES], [ROLE_UM])

        mgmt_api.change_user_password(session_id, None, 'qwerty')
        mgmt_api.create_user(session_id, 'megaadmin', 'qwerty', \
                [ROLE_RO, ROLE_CF, ROLE_SS, ROLE_UPGR, ROLE_UM])
        mgmt_api.change_user_password(session_id, 'megaadmin', 'qwerty')

        mgmt_api.change_user_roles(session_id, 'megaadmin', [ROLE_RO, ROLE_CF, ROLE_UPGR])

        mgmt_api.logout(session_id)
        with self.assertRaises(MEAuthException):
            mgmt_api.get_cluster_config(session_id)

        with self.assertRaises(MEAuthException):
            mgmt_api.authenticate('admin', 'admin')

        session_id = mgmt_api.authenticate('admin', 'qwerty')

        roles = mgmt_api.get_available_roles(session_id)
        self.assertEqual(type(roles), dict)
        self.assertTrue(ROLE_SS in roles)

        ma_session_id = mgmt_api.authenticate('megaadmin', 'qwerty')

        with self.assertRaises(MEPermException):
            mgmt_api.change_user_password(ma_session_id, 'admin', 'qwerty')

        with self.assertRaises(MEPermException):
            mgmt_api.create_user(ma_session_id, 'rouser', 'qwerty', [ROLE_RO])

        with self.assertRaises(MEInvalidArgException):
            mgmt_api.create_user(session_id, 'rouser', 'qwerty', ['ooops'])
        mgmt_api.create_user(session_id, 'rouser', 'qwerty', [ROLE_RO])
        with self.assertRaises(MEAlreadyExistsException):
            mgmt_api.create_user(session_id, 'rouser', 'qwerty', [ROLE_RO])

        with self.assertRaises(MENotFoundException):
            mgmt_api.change_user_roles(session_id, 'someuser', [ROLE_RO, ROLE_SS])

        with self.assertRaises(MEInvalidArgException):
            mgmt_api.create_user(session_id, 'test', 'dd', ROLE_RO)

        with self.assertRaises(MEInvalidArgException):
            mgmt_api.create_user(session_id, '', 'dd', [ROLE_RO])

        with self.assertRaises(MEInvalidArgException):
            mgmt_api.change_user_roles(session_id, 'rouser', ROLE_RO)
        
        mgmt_api.remove_user(session_id, 'rouser')
        with self.assertRaises(MEAuthException):
            mgmt_api.authenticate('rouser', 'qwerty')
        mgmt_api.logout(session_id)

        config = mgmt_api.get_cluster_config(ma_session_id)
        self.assertEqual(config, {})
        config = {DBK_CONFIG_CLNAME: 'testcluster',
                  DBK_CONFIG_KS: base64.b64encode(open(KS_PATH, 'rb').read())}
        mgmt_api.configure_cluster(ma_session_id, config)

        mgmt_api.logout(ma_session_id)