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)