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 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)