/tests/web/test_core.py

https://bitbucket.org/prologic/circuits/ · Python · 112 lines · 81 code · 30 blank · 1 comment · 12 complexity · 4bd3a0529b66c71a53313c5778bd8e44 MD5 · raw file

  1. #!/usr/bin/env python
  2. import pytest
  3. from circuits.six import b, u
  4. from circuits.web import Controller
  5. from .helpers import urlencode, urlopen, HTTPError
  6. class Root(Controller):
  7. def index(self):
  8. return "Hello World!"
  9. def test_args(self, *args, **kwargs):
  10. return "{0}\n{1}".format(repr(args), repr(kwargs))
  11. def test_default_args(self, a=None, b=None):
  12. return "a={0}\nb={1}".format(a, b)
  13. def test_redirect(self):
  14. return self.redirect("/")
  15. def test_forbidden(self):
  16. return self.forbidden()
  17. def test_notfound(self):
  18. return self.notfound()
  19. def test_failure(self):
  20. raise Exception()
  21. def test_root(webapp):
  22. f = urlopen(webapp.server.http.base)
  23. s = f.read()
  24. assert s == b"Hello World!"
  25. def test_404(webapp):
  26. try:
  27. urlopen("%s/foo" % webapp.server.http.base)
  28. except HTTPError as e:
  29. assert e.code == 404
  30. assert e.msg == "Not Found"
  31. else:
  32. assert False
  33. def test_args(webapp):
  34. args = ("1", "2", "3")
  35. kwargs = {"1": "one", "2": "two", "3": "three"}
  36. url = "%s/test_args/%s" % (webapp.server.http.base, "/".join(args))
  37. data = urlencode(kwargs).encode('utf-8')
  38. f = urlopen(url, data)
  39. data = f.read().split(b"\n")
  40. assert eval(data[0]) == args
  41. assert eval(data[1]) == kwargs
  42. @pytest.mark.parametrize("data,expected", [
  43. ((["1"], {}), b("a=1\nb=None")),
  44. ((["1", "2"], {}), b("a=1\nb=2")),
  45. ((["1"], {"b": "2"}), b("a=1\nb=2")),
  46. ])
  47. def test_default_args(webapp, data, expected):
  48. args, kwargs = data
  49. url = u("{0:s}/test_default_args/{1:s}".format(
  50. webapp.server.http.base,
  51. u("/").join(args)
  52. ))
  53. data = urlencode(kwargs).encode("utf-8")
  54. f = urlopen(url, data)
  55. assert f.read() == expected
  56. def test_redirect(webapp):
  57. f = urlopen("%s/test_redirect" % webapp.server.http.base)
  58. s = f.read()
  59. assert s == b"Hello World!"
  60. def test_forbidden(webapp):
  61. try:
  62. urlopen("%s/test_forbidden" % webapp.server.http.base)
  63. except HTTPError as e:
  64. assert e.code == 403
  65. assert e.msg == "Forbidden"
  66. else:
  67. assert False
  68. def test_notfound(webapp):
  69. try:
  70. urlopen("%s/test_notfound" % webapp.server.http.base)
  71. except HTTPError as e:
  72. assert e.code == 404
  73. assert e.msg == "Not Found"
  74. else:
  75. assert False
  76. def test_failure(webapp):
  77. try:
  78. urlopen("%s/test_failure" % webapp.server.http.base)
  79. except HTTPError as e:
  80. assert e.code == 500
  81. else:
  82. assert False