def test_fail_authenticated_session(self, tmpdir, ssl_session): url = 'https://example.com' args = {} session = flexmock() if ssl_session: args['ssl_certs_dir'] = str(tmpdir) session.should_receive('ssl_login').once().and_return(False) else: session.should_receive('krb_login').once().and_return(False) opts = {'krb_rdns': False, 'use_fast_upload': True} opts.update(KOJI_RETRY_OPTS) (flexmock(koji_util.koji).should_receive('ClientSession').with_args( url, opts=opts).and_return(session)) with pytest.raises(RuntimeError): create_koji_session(url, args)
def test_create_simple_session(self): url = 'https://example.com' session = flexmock() opts = {'krb_rdns': False, 'use_fast_upload': True} opts.update(KOJI_RETRY_OPTS) (flexmock(koji_util.koji).should_receive('ClientSession').with_args( url, opts=opts).and_return(session)) assert create_koji_session(url) == session
def get_koji_session(config): from atomic_reactor.utils.koji import create_koji_session auth_info = { "proxyuser": config.koji['auth'].get('proxyuser'), "ssl_certs_dir": config.koji['auth'].get('ssl_certs_dir'), "krb_principal": config.koji['auth'].get('krb_principal'), "krb_keytab": config.koji['auth'].get('krb_keytab_path') } use_fast_upload = config.koji.get('use_fast_upload', True) return create_koji_session(config.koji['hub_url'], auth_info, use_fast_upload)