[tor-commits] [flashproxy/master] Add a test for connector connection limits.

[dcf at torproject.org]

commit 1f4535bfefbf6a8cd6f7b9f26b1d76e8f7bec55c
Author: David Fifield <david at bamsoftware.com>
Date:   Fri Apr 6 11:42:42 2012 -0700

    Add a test for connector connection limits.
---
 connector-test.py |   22 ++++++++++++++++++++++
 1 files changed, 22 insertions(+), 0 deletions(-)

diff --git a/connector-test.py b/connector-test.py
index 36c9208..d1a6bda 100755
--- a/connector-test.py
+++ b/connector-test.py
@@ -1,9 +1,14 @@
 #!/usr/bin/env python
 # -*- coding: utf-8 -*-
 
+import socket
+import subprocess
 import unittest
 from connector import WebSocketDecoder, WebSocketEncoder
 
+LOCAL_ADDRESS = ("127.0.0.1", 40000)
+REMOTE_ADDRESS = ("127.0.0.1", 40001)
+
 def read_frames(dec):
     frames = []
     while True:
@@ -176,5 +181,22 @@ class TestWebSocketEncoder(unittest.TestCase):
                 dec.feed(enc_message)
                 self.assertEqual(read_messages(dec), [(opcode, payload)])
 
+def format_address(addr):
+    return "%s:%d" % addr
+
+class TestConnectionLimit(unittest.TestCase):
+    def setUp(self):
+        self.p = subprocess.Popen(["./connector.py", format_address(LOCAL_ADDRESS), format_address(REMOTE_ADDRESS)])
+
+    def tearDown(self):
+        self.p.terminate()
+
+    def test_remote_limit(self):
+        """Test that the connector limits the number of remote connections that
+        it will accept."""
+        for i in range(5):
+            s = socket.create_connection(REMOTE_ADDRESS, 2)
+        self.assertRaises(socket.error, socket.create_connection, REMOTE_ADDRESS)
+
 if __name__ == "__main__":
     unittest.main()