[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