def _wait_for_code(self):
s = HTTPServer(('127.0.0.1', 80), _TokenHandler)
s.timeout = 30
s.error = None
s.code = None
s.handle_request()
if s.error:
raise RuntimeError("Code lookup failed: {}".format(s.error))
return s.code
