Added get_cookies() and create_cookie()

Author: troublegum

test_server.py

@@ -0,0 +1,30 @@
+from micropyserver import MicroPyServer
+import utils
+
+server = MicroPyServer()
+
+
+def index(request):
+    server.send('OK')
+
+
+def stop(request):
+    server.stop()
+
+
+def set_cookies(request):
+    cookies_header = utils.create_cookie("name", "value", expires="Sat, 01-Jan-2030 00:00:00 GMT")
+    utils.send_response(server, "OK", extend_headers=[cookies_header])
+
+
+def get_cookies(request):
+    cookies = utils.get_cookies(request)
+    utils.send_response(server, str(cookies))
+
+
+server.add_route("/", index)
+server.add_route("/stop", stop)
+server.add_route("/set_cookies", set_cookies)
+server.add_route("/get_cookies", get_cookies)
+
+server.start()

test_utils.py

@@ -24,12 +24,12 @@ def test_parse_query_string_method_should_return_two_params(self):
     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 '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 '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")
@@ -40,8 +40,27 @@ def test_unquote_method_should_return_string(self):
     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 'параметр и значение %'")
+            "параметр и значение %",
+            "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__":

utils.py

@@ -178,3 +178,34 @@ def unquote(string):
             extend(item)
 
     return bytes(res).decode("utf-8")
+
+
+def get_cookies(request):
+    """ return cookies """
+    lines = request.split("\r\n")
+    cookie_string = None
+    for line in lines:
+        if line.find(":") is not -1:
+            header, value = line.split(':', 1)
+            if header.lower() == "cookie":
+                cookie_string = value
+    cookies = {}
+    if cookie_string:
+        for cookie in cookie_string.split('; '):
+            name, value = cookie.strip().split('=', 1)
+            cookies[name] = value
+
+    return cookies
+
+
+def create_cookie(name, value, path="/", domain=None, expires=None):
+    """ create cookie header """
+    cookie = "Set-Cookie: " + str(name) + "=" + str(value)
+    if path:
+        cookie = cookie + "; path=" + path
+    if domain:
+        cookie = cookie + "; domain=" + domain
+    if expires:
+        cookie = cookie + "; expires=" + expires
+
+    return cookie