[tor-commits] [flashproxy/master] Add tests for parse_socks_request.

dcf at torproject.org dcf at torproject.org
Wed Jul 4 14:33:23 UTC 2012


commit b71898a781345d57c9ba7dbe1c4f02e6a7bbbba3
Author: David Fifield <david at bamsoftware.com>
Date:   Wed Jul 4 07:09:30 2012 -0700

    Add tests for parse_socks_request.
---
 connector-test.py |   20 +++++++++++++++++++-
 1 files changed, 19 insertions(+), 1 deletions(-)

diff --git a/connector-test.py b/connector-test.py
index b19fcc6..374b173 100755
--- a/connector-test.py
+++ b/connector-test.py
@@ -4,11 +4,29 @@
 import socket
 import subprocess
 import unittest
-from connector import WebSocketDecoder, WebSocketEncoder
+from connector import parse_socks_request, WebSocketDecoder, WebSocketEncoder
 
 LOCAL_ADDRESS = ("127.0.0.1", 40000)
 REMOTE_ADDRESS = ("127.0.0.1", 40001)
 
+class TestSocks(unittest.TestCase):
+    def test_parse_socks_request_empty(self):
+        self.assertRaises(ValueError, parse_socks_request, "")
+    def test_parse_socks_request_short(self):
+        self.assertRaises(ValueError, parse_socks_request, "\x04\x01\x99\x99\x01\x02\x03\x04")
+    def test_parse_socks_request_ip_userid_missing(self):
+        dest, port = parse_socks_request("\x04\x01\x99\x99\x01\x02\x03\x04\x00")
+        dest, port = parse_socks_request("\x04\x01\x99\x99\x01\x02\x03\x04\x00userid")
+        self.assertEqual((dest, port), ("1.2.3.4", 0x9999))
+    def test_parse_socks_request_ip(self):
+        dest, port = parse_socks_request("\x04\x01\x99\x99\x01\x02\x03\x04userid\x00")
+        self.assertEqual((dest, port), ("1.2.3.4", 0x9999))
+    def test_parse_socks_request_hostname_missing(self):
+        self.assertRaises(ValueError, parse_socks_request, "\x04\x01\x99\x99\x00\x00\x00\x01userid\x00")
+        self.assertRaises(ValueError, parse_socks_request, "\x04\x01\x99\x99\x00\x00\x00\x01userid\x00abc")
+    def test_parse_socks_request_hostname(self):
+        dest, port = parse_socks_request("\x04\x01\x99\x99\x00\x00\x00\x01userid\x00abc\x00")
+
 def read_frames(dec):
     frames = []
     while True:





More information about the tor-commits mailing list