Merge pull request #11 from kbandla/add_tests

converting a few tests to pytest 'style' and testing deprecation decorators
kbandla · Mar 24, 2015 · f525972 · f525972
2 parents 41aaa09 + 35ccd4e
commit f525972
Show file tree

Hide file tree

Showing 4 changed files with 277 additions and 248 deletions.
diff --git a/dpkt/decorators.py b/dpkt/decorators.py
@@ -5,7 +5,7 @@
 def deprecated_method_decorator(deprecated_method):
     def wrapper(*args, **kwargs):
         warnings.simplefilter('always', DeprecationWarning)  # turn off filter
-        warnings.warn("Call to deprecated method %s." % deprecated_method.__name__,
+        warnings.warn("Call to deprecated method %s" % deprecated_method.__name__,
                       category=DeprecationWarning, stacklevel=2)
         warnings.simplefilter('default', DeprecationWarning)  # reset filter
         return deprecated_method(*args, **kwargs)

diff --git a/dpkt/dns.py b/dpkt/dns.py
@@ -84,7 +84,7 @@ def pack_name(name, off, label_ptrs):
 def unpack_name(buf, off):
     name = ''
     saved_off = 0
-    for i in range(100):  # XXX
+    for _ in range(100):  # XXX
         n = ord(buf[off])
         if n == 0:
             off += 1
@@ -344,15 +344,15 @@ def unpack_rr(self, buf, off):
     def unpack(self, buf):
         dpkt.Packet.unpack(self, buf)
         off = self.__hdr_len__
-        cnt = self.qd
+        cnt = self.qd  # FIXME: This relies on this being properly set somewhere else
         self.qd = []
-        for i in range(cnt):
+        for _ in range(cnt):
             q, off = self.unpack_q(buf, off)
             self.qd.append(q)
         for x in ('an', 'ns', 'ar'):
             cnt = getattr(self, x, 0)
             setattr(self, x, [])
-            for i in range(cnt):
+            for _ in range(cnt):
                 rr, off = self.unpack_rr(buf, off)
                 getattr(self, x).append(rr)
         self.data = ''
@@ -374,35 +374,57 @@ def __str__(self):
         del self.label_ptrs
         return buf
 
-
-if __name__ == '__main__':
-    import unittest
+def test_basic():
     from ip import IP
-
-    class DNSTestCase(unittest.TestCase):
-        def test_basic(self):
-            s = 'E\x00\x02\x08\xc15\x00\x00\x80\x11\x92aBk0\x01Bk0w\x005\xc07\x01\xf4\xda\xc2d\xd2\x81\x80\x00\x01\x00\x03\x00\x0b\x00\x0b\x03www\x06google\x03com\x00\x00\x01\x00\x01\xc0\x0c\x00\x05\x00\x01\x00\x00\x03V\x00\x17\x03www\x06google\x06akadns\x03net\x00\xc0,\x00\x01\x00\x01\x00\x00\x01\xa3\x00\x04@\xe9\xabh\xc0,\x00\x01\x00\x01\x00\x00\x01\xa3\x00\x04@\xe9\xabc\xc07\x00\x02\x00\x01\x00\x00KG\x00\x0c\x04usw5\x04akam\xc0>\xc07\x00\x02\x00\x01\x00\x00KG\x00\x07\x04usw6\xc0t\xc07\x00\x02\x00\x01\x00\x00KG\x00\x07\x04usw7\xc0t\xc07\x00\x02\x00\x01\x00\x00KG\x00\x08\x05asia3\xc0t\xc07\x00\x02\x00\x01\x00\x00KG\x00\x05\x02za\xc07\xc07\x00\x02\x00\x01\x00\x00KG\x00\x0f\x02zc\x06akadns\x03org\x00\xc07\x00\x02\x00\x01\x00\x00KG\x00\x05\x02zf\xc07\xc07\x00\x02\x00\x01\x00\x00KG\x00\x05\x02zh\xc0\xd5\xc07\x00\x02\x00\x01\x00\x00KG\x00\x07\x04eur3\xc0t\xc07\x00\x02\x00\x01\x00\x00KG\x00\x07\x04use2\xc0t\xc07\x00\x02\x00\x01\x00\x00KG\x00\x07\x04use4\xc0t\xc0\xc1\x00\x01\x00\x01\x00\x00\xfb4\x00\x04\xd0\xb9\x84\xb0\xc0\xd2\x00\x01\x00\x01\x00\x001\x0c\x00\x04?\xf1\xc76\xc0\xed\x00\x01\x00\x01\x00\x00\xfb4\x00\x04?\xd7\xc6S\xc0\xfe\x00\x01\x00\x01\x00\x001\x0c\x00\x04?\xd00.\xc1\x0f\x00\x01\x00\x01\x00\x00\n\xdf\x00\x04\xc1-\x01g\xc1"\x00\x01\x00\x01\x00\x00\x101\x00\x04?\xd1\xaa\x88\xc15\x00\x01\x00\x01\x00\x00\r\x1a\x00\x04PCC\xb6\xc0o\x00\x01\x00\x01\x00\x00\x10\x7f\x00\x04?\xf1I\xd6\xc0\x87\x00\x01\x00\x01\x00\x00\n\xdf\x00\x04\xce\x84dl\xc0\x9a\x00\x01\x00\x01\x00\x00\n\xdf\x00\x04A\xcb\xea\x1b\xc0\xad\x00\x01\x00\x01\x00\x00\x0b)\x00\x04\xc1l\x9a\t'
-            ip = IP(s)
-            dns = DNS(ip.udp.data)
-            self.failUnless(dns.qd[0].name == 'www.google.com' and
-                            dns.an[1].name == 'www.google.akadns.net')
-            s = '\x05\xf5\x01\x00\x00\x01\x00\x00\x00\x00\x00\x00\x03www\x03cnn\x03com\x00\x00\x01\x00\x01'
-            dns = DNS(s)
-            self.failUnless(s == str(dns))
-
-        def test_PTR(self):
-            s = 'g\x02\x81\x80\x00\x01\x00\x01\x00\x03\x00\x00\x011\x011\x03211\x03141\x07in-addr\x04arpa\x00\x00\x0c\x00\x01\xc0\x0c\x00\x0c\x00\x01\x00\x00\r6\x00$\x07default\nv-umce-ifs\x05umnet\x05umich\x03edu\x00\xc0\x0e\x00\x02\x00\x01\x00\x00\r6\x00\r\x06shabby\x03ifs\xc0O\xc0\x0e\x00\x02\x00\x01\x00\x00\r6\x00\x0f\x0cfish-license\xc0m\xc0\x0e\x00\x02\x00\x01\x00\x00\r6\x00\x0b\x04dns2\x03itd\xc0O'
-            dns = DNS(s)
-            self.failUnless(dns.qd[0].name == '1.1.211.141.in-addr.arpa' and
-                            dns.an[0].ptrname == 'default.v-umce-ifs.umnet.umich.edu' and
-                            dns.ns[0].nsname == 'shabby.ifs.umich.edu' and
-                            dns.ns[1].ttl == 3382L and
-                            dns.ns[2].nsname == 'dns2.itd.umich.edu')
-            self.failUnless(s == str(dns))
-
-        def test_pack_name(self):
-            # Empty name is \0
-            x = pack_name('', 0, {})
-            self.assertEqual(x, '\0')
-
-    unittest.main()
+    s = 'E\x00\x02\x08\xc15\x00\x00\x80\x11\x92aBk0\x01Bk0w\x005\xc07\x01\xf4\xda\xc2d\xd2\x81\x80\x00\x01\x00\x03\x00\x0b\x00\x0b\x03www\x06google\x03com\x00\x00\x01\x00\x01\xc0\x0c\x00\x05\x00\x01\x00\x00\x03V\x00\x17\x03www\x06google\x06akadns\x03net\x00\xc0,\x00\x01\x00\x01\x00\x00\x01\xa3\x00\x04@\xe9\xabh\xc0,\x00\x01\x00\x01\x00\x00\x01\xa3\x00\x04@\xe9\xabc\xc07\x00\x02\x00\x01\x00\x00KG\x00\x0c\x04usw5\x04akam\xc0>\xc07\x00\x02\x00\x01\x00\x00KG\x00\x07\x04usw6\xc0t\xc07\x00\x02\x00\x01\x00\x00KG\x00\x07\x04usw7\xc0t\xc07\x00\x02\x00\x01\x00\x00KG\x00\x08\x05asia3\xc0t\xc07\x00\x02\x00\x01\x00\x00KG\x00\x05\x02za\xc07\xc07\x00\x02\x00\x01\x00\x00KG\x00\x0f\x02zc\x06akadns\x03org\x00\xc07\x00\x02\x00\x01\x00\x00KG\x00\x05\x02zf\xc07\xc07\x00\x02\x00\x01\x00\x00KG\x00\x05\x02zh\xc0\xd5\xc07\x00\x02\x00\x01\x00\x00KG\x00\x07\x04eur3\xc0t\xc07\x00\x02\x00\x01\x00\x00KG\x00\x07\x04use2\xc0t\xc07\x00\x02\x00\x01\x00\x00KG\x00\x07\x04use4\xc0t\xc0\xc1\x00\x01\x00\x01\x00\x00\xfb4\x00\x04\xd0\xb9\x84\xb0\xc0\xd2\x00\x01\x00\x01\x00\x001\x0c\x00\x04?\xf1\xc76\xc0\xed\x00\x01\x00\x01\x00\x00\xfb4\x00\x04?\xd7\xc6S\xc0\xfe\x00\x01\x00\x01\x00\x001\x0c\x00\x04?\xd00.\xc1\x0f\x00\x01\x00\x01\x00\x00\n\xdf\x00\x04\xc1-\x01g\xc1"\x00\x01\x00\x01\x00\x00\x101\x00\x04?\xd1\xaa\x88\xc15\x00\x01\x00\x01\x00\x00\r\x1a\x00\x04PCC\xb6\xc0o\x00\x01\x00\x01\x00\x00\x10\x7f\x00\x04?\xf1I\xd6\xc0\x87\x00\x01\x00\x01\x00\x00\n\xdf\x00\x04\xce\x84dl\xc0\x9a\x00\x01\x00\x01\x00\x00\n\xdf\x00\x04A\xcb\xea\x1b\xc0\xad\x00\x01\x00\x01\x00\x00\x0b)\x00\x04\xc1l\x9a\t'
+    ip = IP(s)
+    my_dns = DNS(ip.udp.data)
+    assert my_dns.qd[0].name == 'www.google.com' and my_dns.an[1].name == 'www.google.akadns.net'
+    s = '\x05\xf5\x01\x00\x00\x01\x00\x00\x00\x00\x00\x00\x03www\x03cnn\x03com\x00\x00\x01\x00\x01'
+    my_dns = DNS(s)
+    assert s == str(my_dns)
+
+def test_PTR():
+    s = 'g\x02\x81\x80\x00\x01\x00\x01\x00\x03\x00\x00\x011\x011\x03211\x03141\x07in-addr\x04arpa\x00\x00\x0c\x00\x01\xc0\x0c\x00\x0c\x00\x01\x00\x00\r6\x00$\x07default\nv-umce-ifs\x05umnet\x05umich\x03edu\x00\xc0\x0e\x00\x02\x00\x01\x00\x00\r6\x00\r\x06shabby\x03ifs\xc0O\xc0\x0e\x00\x02\x00\x01\x00\x00\r6\x00\x0f\x0cfish-license\xc0m\xc0\x0e\x00\x02\x00\x01\x00\x00\r6\x00\x0b\x04dns2\x03itd\xc0O'
+    my_dns = DNS(s)
+    assert my_dns.qd[0].name == '1.1.211.141.in-addr.arpa' and \
+           my_dns.an[0].ptrname == 'default.v-umce-ifs.umnet.umich.edu' and \
+           my_dns.ns[0].nsname == 'shabby.ifs.umich.edu' and \
+           my_dns.ns[1].ttl == 3382L and \
+           my_dns.ns[2].nsname == 'dns2.itd.umich.edu'
+    assert s == str(my_dns)
+
+def test_pack_name():
+    # Empty name is \0
+    x = pack_name('', 0, {})
+    assert x == '\0'
+
+def test_deprecated_methods():
+    """Test deprecated methods. Note: when they are removed so should this test"""
+    s = 'g\x02\x81\x80\x00\x01\x00\x01\x00\x03\x00\x00\x011\x011\x03211\x03141\x07in-addr\x04arpa\x00\x00\x0c\x00\x01\xc0\x0c\x00\x0c\x00\x01\x00\x00\r6\x00$\x07default\nv-umce-ifs\x05umnet\x05umich\x03edu\x00\xc0\x0e\x00\x02\x00\x01\x00\x00\r6\x00\r\x06shabby\x03ifs\xc0O\xc0\x0e\x00\x02\x00\x01\x00\x00\r6\x00\x0f\x0cfish-license\xc0m\xc0\x0e\x00\x02\x00\x01\x00\x00\r6\x00\x0b\x04dns2\x03itd\xc0O'
+    my_dns = DNS(s)
+    my_dns.get_aa()
+    my_dns.get_opcode()
+    qr = my_dns.get_qr()
+    my_dns.set_qr(qr)
+    qr2 = my_dns.get_qr()
+    assert qr == qr2
+
+def test_deprecated_method_performance():
+    """Test the performance hit for the deprecation decorator"""
+    from timeit import Timer
+
+    s = 'g\x02\x81\x80\x00\x01\x00\x01\x00\x03\x00\x00\x011\x011\x03211\x03141\x07in-addr\x04arpa\x00\x00\x0c\x00\x01\xc0\x0c\x00\x0c\x00\x01\x00\x00\r6\x00$\x07default\nv-umce-ifs\x05umnet\x05umich\x03edu\x00\xc0\x0e\x00\x02\x00\x01\x00\x00\r6\x00\r\x06shabby\x03ifs\xc0O\xc0\x0e\x00\x02\x00\x01\x00\x00\r6\x00\x0f\x0cfish-license\xc0m\xc0\x0e\x00\x02\x00\x01\x00\x00\r6\x00\x0b\x04dns2\x03itd\xc0O'
+    my_dns = DNS(s)
+    t1 = Timer(lambda: my_dns.aa).timeit(10000)  
+    t2 = Timer(my_dns.get_aa).timeit(10000)
+    print 'Performance of dns.aa vs. dns.get_aa(): %f %f' % (t1, t2)   
+
+if __name__ == '__main__':
+    # Runs all the test associated with this class/file
+    test_basic()
+    test_PTR()
+    test_pack_name()
+    test_deprecated_methods()
+    test_deprecated_method_performance()
+    print 'Tests Successful...'
diff --git a/dpkt/http.py b/dpkt/http.py
@@ -174,70 +174,73 @@ def __str__(self):
                                     self.reason) + Message.__str__(self)
 
 
+def test_parse_request():
+    s = """POST /main/redirect/ab/1,295,,00.html HTTP/1.0\r\nReferer: http://www.email.com/login/snap/login.jhtml\r\nConnection: Keep-Alive\r\nUser-Agent: Mozilla/4.75 [en] (X11; U; OpenBSD 2.8 i386; Nav)\r\nHost: ltd.snap.com\r\nAccept: image/gif, image/x-xbitmap, image/jpeg, image/pjpeg, image/png, */*\r\nAccept-Encoding: gzip\r\nAccept-Language: en\r\nAccept-Charset: iso-8859-1,*,utf-8\r\nContent-type: application/x-www-form-urlencoded\r\nContent-length: 61\r\n\r\nsn=em&mn=dtest4&pw=this+is+atest&fr=true&login=Sign+in&od=www"""
+    r = Request(s)
+    assert r.method == 'POST'
+    assert r.uri == '/main/redirect/ab/1,295,,00.html'
+    assert r.body == 'sn=em&mn=dtest4&pw=this+is+atest&fr=true&login=Sign+in&od=www'
+    assert r.headers['content-type'] == 'application/x-www-form-urlencoded'
+    try:
+        Request(s[:60])
+        assert 'invalid headers parsed!'
+    except dpkt.UnpackError:
+        pass
+
+def test_format_request():
+    r = Request()
+    assert str(r) == 'GET / HTTP/1.0\r\n\r\n'
+    r.method = 'POST'
+    r.uri = '/foo/bar/baz.html'
+    r.headers['content-type'] = 'text/plain'
+    r.headers['content-length'] = '5'
+    r.body = 'hello'
+    s = str(r)
+    assert s.startswith('POST /foo/bar/baz.html HTTP/1.0\r\n')
+    assert s.endswith('\r\n\r\nhello')
+    assert '\r\ncontent-length: 5\r\n' in s
+    assert '\r\ncontent-type: text/plain\r\n' in s
+    r = Request(str(r))
+    assert str(r) == s
+
+def test_chunked_response():
+    s = """HTTP/1.1 200 OK\r\nCache-control: no-cache\r\nPragma: no-cache\r\nContent-Type: text/javascript; charset=utf-8\r\nContent-Encoding: gzip\r\nTransfer-Encoding: chunked\r\nSet-Cookie: S=gmail=agg:gmail_yj=v2s:gmproxy=JkU; Domain=.google.com; Path=/\r\nServer: GFE/1.3\r\nDate: Mon, 12 Dec 2005 22:33:23 GMT\r\n\r\na\r\n\x1f\x8b\x08\x00\x00\x00\x00\x00\x00\x00\r\n152\r\nm\x91MO\xc4 \x10\x86\xef\xfe\n\x82\xc9\x9eXJK\xe9\xb6\xee\xc1\xe8\x1e6\x9e4\xf1\xe0a5\x86R\xda\x12Yh\x80\xba\xfa\xef\x85\xee\x1a/\xf21\x99\x0c\xef0<\xc3\x81\xa0\xc3\x01\xe6\x10\xc1<\xa7eYT5\xa1\xa4\xac\xe1\xdb\x15:\xa4\x9d\x0c\xfa5K\x00\xf6.\xaa\xeb\x86\xd5y\xcdHY\x954\x8e\xbc*h\x8c\x8e!L7Y\xe6\'\xeb\x82WZ\xcf>8\x1ed\x87\x851X\xd8c\xe6\xbc\x17Z\x89\x8f\xac \x84e\xde\n!]\x96\x17i\xb5\x02{{\xc2z0\x1e\x0f#7\x9cw3v\x992\x9d\xfc\xc2c8\xea[/EP\xd6\xbc\xce\x84\xd0\xce\xab\xf7`\'\x1f\xacS\xd2\xc7\xd2\xfb\x94\x02N\xdc\x04\x0f\xee\xba\x19X\x03TtW\xd7\xb4\xd9\x92\n\xbcX\xa7;\xb0\x9b\'\x10$?F\xfd\xf3CzPt\x8aU\xef\xb8\xc8\x8b-\x18\xed\xec<\xe0\x83\x85\x08!\xf8"[\xb0\xd3j\x82h\x93\xb8\xcf\xd8\x9b\xba\xda\xd0\x92\x14\xa4a\rc\reM\xfd\x87=X;h\xd9j;\xe0db\x17\xc2\x02\xbd\xb0F\xc2in#\xfb:\xb6\xc4x\x15\xd6\x9f\x8a\xaf\xcf)\x0b^\xbc\xe7i\x11\x80\x8b\x00D\x01\xd8/\x82x\xf6\xd8\xf7J(\xae/\x11p\x1f+\xc4p\t:\xfe\xfd\xdf\xa3Y\xfa\xae4\x7f\x00\xc5\xa5\x95\xa1\xe2\x01\x00\x00\r\n0\r\n\r\n"""
+    r = Response(s)
+    assert r.version == '1.1'
+    assert r.status == '200'
+    assert r.reason == 'OK'
+
+def test_multicookie_response():
+    s = """HTTP/1.x 200 OK\r\nSet-Cookie: first_cookie=cookie1; path=/; domain=.example.com\r\nSet-Cookie: second_cookie=cookie2; path=/; domain=.example.com\r\nContent-Length: 0\r\n\r\n"""
+    r = Response(s)
+    assert type(r.headers['set-cookie']) is list
+    assert len(r.headers['set-cookie']) == 2
+
+def test_request_version():
+    s = """GET / HTTP/1.0\r\n\r\n"""
+    r = Request(s)
+    assert r.method == 'GET'
+    assert r.uri == '/'
+    assert r.version == '1.0'
+
+    s = """GET /\r\n\r\n"""
+    r = Request(s)
+    assert r.method == 'GET'
+    assert r.uri == '/'
+    assert r.version == '0.9'
+
+    s = """GET / CHEESE/1.0\r\n\r\n"""
+    try:
+        Request(s)
+        assert "invalid protocol version parsed!"
+    except:
+        pass
+
 if __name__ == '__main__':
-    import unittest
-
-    class HTTPTest(unittest.TestCase):
-        def test_parse_request(self):
-            s = """POST /main/redirect/ab/1,295,,00.html HTTP/1.0\r\nReferer: http://www.email.com/login/snap/login.jhtml\r\nConnection: Keep-Alive\r\nUser-Agent: Mozilla/4.75 [en] (X11; U; OpenBSD 2.8 i386; Nav)\r\nHost: ltd.snap.com\r\nAccept: image/gif, image/x-xbitmap, image/jpeg, image/pjpeg, image/png, */*\r\nAccept-Encoding: gzip\r\nAccept-Language: en\r\nAccept-Charset: iso-8859-1,*,utf-8\r\nContent-type: application/x-www-form-urlencoded\r\nContent-length: 61\r\n\r\nsn=em&mn=dtest4&pw=this+is+atest&fr=true&login=Sign+in&od=www"""
-            r = Request(s)
-            assert r.method == 'POST'
-            assert r.uri == '/main/redirect/ab/1,295,,00.html'
-            assert r.body == 'sn=em&mn=dtest4&pw=this+is+atest&fr=true&login=Sign+in&od=www'
-            assert r.headers['content-type'] == 'application/x-www-form-urlencoded'
-            try:
-                Request(s[:60])
-                assert 'invalid headers parsed!'
-            except dpkt.UnpackError:
-                pass
-
-        def test_format_request(self):
-            r = Request()
-            assert str(r) == 'GET / HTTP/1.0\r\n\r\n'
-            r.method = 'POST'
-            r.uri = '/foo/bar/baz.html'
-            r.headers['content-type'] = 'text/plain'
-            r.headers['content-length'] = '5'
-            r.body = 'hello'
-            s = str(r)
-            assert s.startswith('POST /foo/bar/baz.html HTTP/1.0\r\n')
-            assert s.endswith('\r\n\r\nhello')
-            assert '\r\ncontent-length: 5\r\n' in s
-            assert '\r\ncontent-type: text/plain\r\n' in s
-            r = Request(str(r))
-            assert str(r) == s
-
-        def test_chunked_response(self):
-            s = """HTTP/1.1 200 OK\r\nCache-control: no-cache\r\nPragma: no-cache\r\nContent-Type: text/javascript; charset=utf-8\r\nContent-Encoding: gzip\r\nTransfer-Encoding: chunked\r\nSet-Cookie: S=gmail=agg:gmail_yj=v2s:gmproxy=JkU; Domain=.google.com; Path=/\r\nServer: GFE/1.3\r\nDate: Mon, 12 Dec 2005 22:33:23 GMT\r\n\r\na\r\n\x1f\x8b\x08\x00\x00\x00\x00\x00\x00\x00\r\n152\r\nm\x91MO\xc4 \x10\x86\xef\xfe\n\x82\xc9\x9eXJK\xe9\xb6\xee\xc1\xe8\x1e6\x9e4\xf1\xe0a5\x86R\xda\x12Yh\x80\xba\xfa\xef\x85\xee\x1a/\xf21\x99\x0c\xef0<\xc3\x81\xa0\xc3\x01\xe6\x10\xc1<\xa7eYT5\xa1\xa4\xac\xe1\xdb\x15:\xa4\x9d\x0c\xfa5K\x00\xf6.\xaa\xeb\x86\xd5y\xcdHY\x954\x8e\xbc*h\x8c\x8e!L7Y\xe6\'\xeb\x82WZ\xcf>8\x1ed\x87\x851X\xd8c\xe6\xbc\x17Z\x89\x8f\xac \x84e\xde\n!]\x96\x17i\xb5\x02{{\xc2z0\x1e\x0f#7\x9cw3v\x992\x9d\xfc\xc2c8\xea[/EP\xd6\xbc\xce\x84\xd0\xce\xab\xf7`\'\x1f\xacS\xd2\xc7\xd2\xfb\x94\x02N\xdc\x04\x0f\xee\xba\x19X\x03TtW\xd7\xb4\xd9\x92\n\xbcX\xa7;\xb0\x9b\'\x10$?F\xfd\xf3CzPt\x8aU\xef\xb8\xc8\x8b-\x18\xed\xec<\xe0\x83\x85\x08!\xf8"[\xb0\xd3j\x82h\x93\xb8\xcf\xd8\x9b\xba\xda\xd0\x92\x14\xa4a\rc\reM\xfd\x87=X;h\xd9j;\xe0db\x17\xc2\x02\xbd\xb0F\xc2in#\xfb:\xb6\xc4x\x15\xd6\x9f\x8a\xaf\xcf)\x0b^\xbc\xe7i\x11\x80\x8b\x00D\x01\xd8/\x82x\xf6\xd8\xf7J(\xae/\x11p\x1f+\xc4p\t:\xfe\xfd\xdf\xa3Y\xfa\xae4\x7f\x00\xc5\xa5\x95\xa1\xe2\x01\x00\x00\r\n0\r\n\r\n"""
-            r = Response(s)
-            assert r.version == '1.1'
-            assert r.status == '200'
-            assert r.reason == 'OK'
-
-        def test_multicookie_response(self):
-            s = """HTTP/1.x 200 OK\r\nSet-Cookie: first_cookie=cookie1; path=/; domain=.example.com\r\nSet-Cookie: second_cookie=cookie2; path=/; domain=.example.com\r\nContent-Length: 0\r\n\r\n"""
-            r = Response(s)
-            assert type(r.headers['set-cookie']) is list
-            assert len(r.headers['set-cookie']) == 2
-
-        def test_request_version(self):
-            s = """GET / HTTP/1.0\r\n\r\n"""
-            r = Request(s)
-            assert r.method == 'GET'
-            assert r.uri == '/'
-            assert r.version == '1.0'
-
-            s = """GET /\r\n\r\n"""
-            r = Request(s)
-            assert r.method == 'GET'
-            assert r.uri == '/'
-            assert r.version == '0.9'
-
-            s = """GET / CHEESE/1.0\r\n\r\n"""
-            try:
-                Request(s)
-                assert "invalid protocol version parsed!"
-            except:
-                pass
-
-    unittest.main()
+    # Runs all the test associated with this class/file
+    test_parse_request()
+    test_format_request()
+    test_chunked_response()
+    test_multicookie_response()
+    test_request_version()
+    print 'Tests Successful...'