def __init__(self, opts): """constructor provides access to the configuration options""" super(FunctionComponent, self).__init__(opts) # check_config is handling all of the error checking config_opts = check_config(opts) self.atd_url = config_opts.get("atd_url") self.atd_username = config_opts.get("atd_username") self.atd_password = config_opts.get("atd_password") self.timeout_mins = config_opts.get("timeout_mins") self.polling_interval = config_opts.get("polling_interval") self.vm_profile_list = config_opts.get("vm_profile_list") self.filePriority = config_opts.get("filePriority") self.trust_cert = config_opts.get("trust_cert") # Verify can make connection to ATD with given config values _get_atd_session_headers(self)
def test_helper_get_atd_session_headers(self, mocked_requests_get): sim_content = {"results": {"session": "session_key", "userId": "1"}} mocked_requests_get.return_value = self._generateResponse( sim_content, 200) creds = MockCredsAfterCheck() atd_session_headers = _get_atd_session_headers(creds) # Assert headers returned are correct assert atd_session_headers == self.sim_session_headers