test_utils.py

import unittest
import utils


class TestUtils(unittest.TestCase):
    def test_get_request_method_should_return_get(self):
        request = "GET / HTTP/1.1\r\nHost: localhost"
        self.assertEqual(utils.get_request_method(request), "GET", "Should be GET")

    def test_get_request_method_should_return_post(self):
        request = "POST /post HTTP/1.1\r\nHost: localhost\r\nContent-Type: application/x-www-form-urlencoded\r\nContent-Length: 0\r\n\r\n"
        self.assertEqual(utils.get_request_method(request), "POST", "Should be POST")

    def test_get_request_query_string_method_should_return_string(self):
        request = "GET /?param_one=one&param_two=two HTTP/1.1\r\nHost: localhost\r\n\r\n"
        self.assertEqual(utils.get_request_query_string(request), "param_one=one&param_two=two",
                         "Should be 'param_one=one&param_two=two'")

    def test_parse_query_string_method_should_return_two_params(self):
        query_string = "param_one=one&param_two=two"
        self.assertEqual(utils.parse_query_string(query_string), {"param_one": "one", "param_two": "two"},
                         "Should be result with two params")

    def test_get_request_query_params_method_should_return_two_params(self):
        request = "GET /?param_one=one&param_two=two HTTP/1.1\r\nHost: localhost\r\n\r\n"
        self.assertEqual(utils.get_request_query_params(request), {"param_one": "one", "param_two": "two"},
                         "Should be result with two params")

    def test_get_request_post_params_method_should_return_two_params(self):
        request = "POST /post HTTP/1.1\r\nHost: localhost\r\nContent-Type: application/x-www-form-urlencoded\r\nContent-Length: 27\r\n\r\nparam_one=one&param_two=two"
        self.assertEqual(utils.get_request_post_params(request), {"param_one": "one", "param_two": "two"},
                         "Should be result with two params")

    def test_unquote_method_should_return_empty_string(self):
        self.assertEqual(utils.unquote(""), "", "Should be empty string")

    def test_unquote_method_should_return_string(self):
        self.assertEqual(utils.unquote("param"), "param", "Should be 'param'")

    def test_unquote_method_should_return_unquoted_string(self):
        self.assertEqual(utils.unquote(
            "%D0%BF%D0%B0%D1%80%D0%B0%D0%BC%D0%B5%D1%82%D1%80%20%D0%B8%20%D0%B7%D0%BD%D0%B0%D1%87%D0%B5%D0%BD%D0%B8%D0%B5%20%25"),
            "параметр и значение %",
            "Should be 'параметр и значение %'")

    def test_get_cookies_method_should_return_param(self):
        request = "GET / HTTP/1.1\r\nHost: localhost\r\nCookie:foo=bar"
        self.assertEqual(utils.get_cookies(request), {"foo": "bar"}, "Should be result with one param")

    def test_get_cookies_method_should_return_two_params(self):
        request = "GET / HTTP/1.1\r\nHost: localhost\r\nCookie:foo=bar; abc=def\r\nContent-Length: 4\r\n\r\ntest"
        self.assertEqual(utils.get_cookies(request), {"foo": "bar", "abc": "def"}, "Should be result with two params")

    def test_create_cookie_method_should_return_correct_cookie(self):
        name = "name"
        value = "value"
        path = "/"
        domain = "localhost"
        expires = "Sat, 01-Jan-2030 00:00:00 GMT"
        utils.create_cookie(name, value, path, domain, expires)
        self.assertEqual(utils.create_cookie(name, value, path, domain, expires),
                         "Set-Cookie: name=value; path=/; domain=localhost; expires=Sat, 01-Jan-2030 00:00:00 GMT",
                         "Should be correct cookie header")


if __name__ == "__main__":
    unittest.main()