def twill_allowed(self, twillInfo):
endpoint = self.v1endpoint(self.server_port)
authreq = AuthRequest(endpoint, assoc=None)
url = authreq.redirectURL(self.realm, self.return_to)
c = twill.commands
try:
c.go(url)
c.code(200)
c.get_browser()._browser.set_handle_redirect(False)
c.formvalue(1, 'remember', 'true')
c.find('name="login_as" value="bob"')
c.submit("yes")
c.code(302)
# Since we set remember=yes, the second time we shouldn't
# see that page.
c.go(url)
c.code(302)
headers = c.get_browser()._browser.response().info()
finalURL = headers['Location']
self.failUnless(finalURL.startswith(self.return_to))
except twill.commands.TwillAssertionError as e:
from traceback import format_exc
msg = '%s\nTwill output:%s\nTwill errors:%s\nFinal page:\n%s' % (
format_exc(),
self.twillOutput.getvalue(),
self.twillErr.getvalue(),
c.get_browser().get_html())
self.fail(msg)
def twill_allowed(self, twillInfo):
endpoint = self.v1endpoint(self.server_port)
authreq = AuthRequest(endpoint, assoc=None)
url = authreq.redirectURL(self.realm, self.return_to)
c = twill.commands
try:
c.go(url)
c.code(200)
c.get_browser()._browser.set_handle_redirect(False)
c.formvalue(1, 'remember', 'true')
c.find('name="login_as" value="bob"')
c.submit("yes")
c.code(302)
# Since we set remember=yes, the second time we shouldn't
# see that page.
c.go(url)
c.code(302)
headers = c.get_browser()._browser.response().info()
finalURL = headers['Location']
self.failUnless(finalURL.startswith(self.return_to))
except twill.commands.TwillAssertionError, e:
from traceback import format_exc
msg = '%s\nTwill output:%s\nTwill errors:%s\nFinal page:\n%s' % (
format_exc(), self.twillOutput.getvalue(),
self.twillErr.getvalue(), c.get_browser().get_html())
self.fail(msg)
def setUp(self):
self.endpoint = OpenIDServiceEndpoint()
self.endpoint.delegate = 'http://server.unittest/joe'
self.endpoint.server_url = 'http://server.unittest/'
self.assoc = self
self.assoc.handle = 'assoc@handle'
self.authreq = AuthRequest(self.endpoint, self.assoc)
def twill_checkidv1(self, twillInfo):
endpoint = self.v1endpoint(self.server_port)
authreq = AuthRequest(endpoint, assoc=None)
url = authreq.redirectURL(self.realm, self.return_to)
c = twill.commands
try:
c.go(url)
c.get_browser()._browser.set_handle_redirect(False)
c.submit("yes")
c.code(302)
headers = c.get_browser()._browser.response().info()
finalURL = headers['Location']
self.failUnless('openid.mode=id_res' in finalURL, finalURL)
self.failUnless('openid.identity=' in finalURL, finalURL)
except twill.commands.TwillAssertionError, e:
msg = '%s\nFinal page:\n%s' % (str(e), c.get_browser().get_html())
self.fail(msg)
def twill_checkidv1(self, twillInfo):
endpoint = self.v1endpoint(self.server_port)
authreq = AuthRequest(endpoint, assoc=None)
url = authreq.redirectURL(self.realm, self.return_to)
c = twill.commands
try:
c.go(url)
c.get_browser()._browser.set_handle_redirect(False)
c.submit("yes")
c.code(302)
headers = c.get_browser()._browser.response().info()
finalURL = headers["Location"]
self.failUnless("openid.mode=id_res" in finalURL, finalURL)
self.failUnless("openid.identity=" in finalURL, finalURL)
except twill.commands.TwillAssertionError, e:
msg = "%s\nFinal page:\n%s" % (str(e), c.get_browser().get_html())
self.fail(msg)
class TestAuthRequest(unittest.TestCase):
def setUp(self):
self.endpoint = OpenIDServiceEndpoint()
self.endpoint.delegate = 'http://server.unittest/joe'
self.endpoint.server_url = 'http://server.unittest/'
self.assoc = self
self.assoc.handle = 'assoc@handle'
self.authreq = AuthRequest(self.endpoint, self.assoc)
def test_addExtensionArg(self):
self.authreq.addExtensionArg('bag', 'color', 'brown')
self.authreq.addExtensionArg('bag', 'material', 'paper')
self.failUnlessEqual(self.authreq.extra_args,
{'openid.bag.color': 'brown',
'openid.bag.material': 'paper'})
url = self.authreq.redirectURL('http://7.utest/', 'http://7.utest/r')
self.failUnless(url.find('openid.bag.color=brown') != -1,
'extension arg not found in %s' % (url,))
self.failUnless(url.find('openid.bag.material=paper') != -1,
'extension arg not found in %s' % (url,))
def twill_allowed(self, twillInfo):
endpoint = self.v1endpoint(self.server_port)
authreq = AuthRequest(endpoint, assoc=None)
url = authreq.redirectURL(self.realm, self.return_to)
try:
browser = twill.commands.get_browser()
browser.get_all_forms()
browser.formvalue(1, 'remember', 'true')
browser.find('name="login_as" value="bob"')
browser.submit("yes")
browser.code(302)
browser.go(url)
browser.code(302)
finalURL = browser._browser._response.info().getheaders('location')
self.failUnless(finalURL.startswith(self.return_to))
except twill.commands.TwillAssertionError, e:
from traceback import format_exc
msg = '%s\nTwill output:%s\nTwill errors:%s\nFinal page:\n%s' % (
format_exc(),
self.twillOutput.getvalue(),
self.twillErr.getvalue(),
twill.commands.get_browser().get_html())
self.fail(msg)
