Implement a few more skipped tests in test/test_conn.py

1 files changed, 19 insertions, 4 deletions

diff --git a/test/test_conn.py b/test/test_conn.py
index 6213440..7392bab 100644
--- a/test/test_conn.py
+++ b/test/test_conn.py
@@ -163,14 +163,29 @@ class ConnTest(unittest2.TestCase):
         self.conn._read_bytes(len(self.config['payload']))
         self.assertEqual(socket.create_connection.call_count, 1)
 
-    @unittest2.skip("Not Implemented")
     def test_recv__failure_sets_dirty_connection(self):
-        pass
+
+        def raise_error(*args):
+            raise socket.error
+
+        # test that recv'ing attempts to reconnect
+        assert self.conn._dirty is False
+        assert isinstance(self.conn._sock, mock.Mock)
+        self.conn._sock.recv.side_effect=raise_error
+        try:
+            self.conn.recv(self.config['request_id'])
+        except ConnectionError:
+            self.assertEquals(self.conn._dirty, True)
 
     @unittest2.skip("Not Implemented")
     def test_recv__doesnt_consume_extra_data_in_stream(self):
         pass
 
-    @unittest2.skip("Not Implemented")
     def test_close__object_is_reusable(self):
-        pass
+
+        # test that sending to a closed connection
+        # will re-connect and send data to the socket
+        self.conn.close()
+        self.conn.send(self.config['request_id'], self.config['payload'])
+        self.assertEqual(socket.create_connection.call_count, 1)
+        self.conn._sock.sendall.assert_called_with(self.config['payload'])