コード例 #1
0
 def setUp(self):
     self.mgr = KBaseWSNotebookManager()
コード例 #2
0
ファイル: test_kbasewsmanager.py プロジェクト: cbun/narrative
 def setUp(self):
     self.mgr = KBaseWSNotebookManager()
コード例 #3
0
class NarrBaseTestCase(unittest.TestCase):
    # Before test:
    # - Log in (for tests that require login)
    # also sets the token in the environment variable so the manager can get to it.
    @classmethod
    def setUpClass(self):
        self.user_id = test_user_id
        self.pwd = getpass("Password for {}: ".format(test_user_id))
        self.token = biokbase.auth.Token(user_id=self.user_id,
                                         password=self.pwd)
        # by default, user's left logged out

    @classmethod
    def setUp(self):
        self.mgr = KBaseWSNotebookManager()

    @classmethod
    def tearDown(self):
        self.logout()
        pass

    @classmethod
    def tearDownClass(self):
        pass

    @classmethod
    def login(self):
        biokbase.auth.set_environ_token(self.token.token)

    @classmethod
    def logout(self):
        biokbase.auth.set_environ_token(None)

    def test_manager_instantiated(self):
        self.assertIsInstance(
            self.mgr, biokbase.narrative.kbasewsmanager.KBaseWSNotebookManager)

    # test get_userid()
    def test_user_id_loggedin(self):
        self.login()
        self.assertEquals(self.mgr.get_userid(), self.user_id)

    # test get_userid()
    def test_user_id_loggedout(self):
        self.assertEquals(self.mgr.get_userid(), None)

    # test wsclient()
    def test_wsclient(self):
        self.assertIsInstance(self.mgr.wsclient(), Workspace)

    # test info_string (just make sure it's a string)
    def test_info_string(self):
        self.assertIsInstance(self.mgr.info_string(), basestring)

    # test list notebooks while logged in returns a list of strings
    def test_list_notebooks_loggedin(self):
        self.login()
        self.test_list_notebooks()

    def test_list_notebooks_loggedout(self):
        self.test_list_notebooks()

    def test_list_notebooks(self):
        nb_list = self.mgr.list_notebooks()
        # assert we actually get something
        self.assertIsInstance(nb_list, list)
        # assert it's a list of formatted dicts
        format_failure = self.check_nb_list_format(nb_list)
        self.assertIsNone(format_failure)

    def check_nb_list_format(self, nb_list):
        for nb_info in nb_list:
            if not 'name' in nb_info:
                return 'Missing a "name" key!'
            if not 'notebook_id' in nb_info:
                return 'Missing a "notebook_id key!'
            if not name_regex.match(nb_info['name']):
                return 'Incorrect format for "name" key: {}'.format(
                    nb_info['name'])
            if not obj_regex.match(nb_info['notebook_id']):
                return 'Incorrect format for "notebook_id" key: {}'.format(
                    nb_info['notebook_id'])
        # If we make it this far, don't return anything! Hooray!
        return None

    def test_clean_id(self):
        spacey_str = 'test test  test test   test'
        unspacey_str = 'test_test__test_test___test'
        self.assertEquals(self.mgr._clean_id(spacey_str), unspacey_str)
コード例 #4
0
ファイル: test_kbasewsmanager.py プロジェクト: cbun/narrative
class NarrBaseTestCase(unittest.TestCase):
    # Before test:
    # - Log in (for tests that require login)
    # also sets the token in the environment variable so the manager can get to it.
    @classmethod
    def setUpClass(self):
        self.user_id = test_user_id
        self.pwd = getpass("Password for {}: ".format(test_user_id))
        self.token = biokbase.auth.Token(user_id=self.user_id, password=self.pwd)
        # by default, user's left logged out

    @classmethod
    def setUp(self):
        self.mgr = KBaseWSNotebookManager()

    @classmethod
    def tearDown(self):
        self.logout()
        pass

    @classmethod
    def tearDownClass(self):
        pass

    @classmethod
    def login(self):
        biokbase.auth.set_environ_token(self.token.token)

    @classmethod
    def logout(self):
        biokbase.auth.set_environ_token(None)

    def test_manager_instantiated(self):
        self.assertIsInstance(self.mgr, biokbase.narrative.kbasewsmanager.KBaseWSNotebookManager)

    # test get_userid()
    def test_user_id_loggedin(self):
        self.login()
        self.assertEquals(self.mgr.get_userid(), self.user_id)

    # test get_userid()
    def test_user_id_loggedout(self):
        self.assertEquals(self.mgr.get_userid(), None)

    # test wsclient()
    def test_wsclient(self):
        self.assertIsInstance(self.mgr.wsclient(), Workspace)

    # test info_string (just make sure it's a string)
    def test_info_string(self):
        self.assertIsInstance(self.mgr.info_string(), basestring)

    # test list notebooks while logged in returns a list of strings
    def test_list_notebooks_loggedin(self):
        self.login()
        self.test_list_notebooks()

    def test_list_notebooks_loggedout(self):
        self.test_list_notebooks()

    def test_list_notebooks(self):
        nb_list = self.mgr.list_notebooks()
        # assert we actually get something
        self.assertIsInstance(nb_list, list)
        # assert it's a list of formatted dicts
        format_failure = self.check_nb_list_format(nb_list)
        self.assertIsNone(format_failure)

    def check_nb_list_format(self, nb_list):
        for nb_info in nb_list:
            if not 'name' in nb_info:
                return 'Missing a "name" key!'
            if not 'notebook_id' in nb_info:
                return 'Missing a "notebook_id key!'
            if not name_regex.match(nb_info['name']):
                return 'Incorrect format for "name" key: {}'.format(nb_info['name'])
            if not obj_regex.match(nb_info['notebook_id']):
                return 'Incorrect format for "notebook_id" key: {}'.format(nb_info['notebook_id'])
        # If we make it this far, don't return anything! Hooray!
        return None

    def test_clean_id(self):
        spacey_str = 'test test  test test   test'
        unspacey_str = 'test_test__test_test___test'
        self.assertEquals(self.mgr._clean_id(spacey_str), unspacey_str)