directory_okta_test.py 文件源码

def setUp(self):
        class MockResponse:
            def __init__(self, status_code, data):
                self.status_code = status_code
                self.text = json.dumps(data)
                self.links = {}

        self.mock_response = MockResponse
        self.orig_directory_init = OktaDirectoryConnector.__init__
        OktaDirectoryConnector.__init__ = mock.Mock(return_value=None)
        directory = OktaDirectoryConnector({})
        directory.logger = mock.create_autospec(logging.Logger)
        directory.groups_client = okta.UserGroupsClient('example.com', 'xyz')
        directory.user_identity_type = 'enterpriseID'
        self.directory = directory