def test_connect_sets_stream(self, IOStream):
IOStream.return_value.connect.return_value = self.future_value(None)
conn = Connection("localhost", 1234)
self.assertEqual(conn.stream, None)
yield conn.connect()
self.assertEqual(conn.stream, IOStream.return_value)
IOStream.return_value.connect.assert_called_once_with(
("localhost", 1234))
def test_connect_sets_stream(self, IOStream):
IOStream.return_value.connect.return_value = self.future_value(None)
conn = Connection("localhost", 1234)
self.assertEqual(conn.stream, None)
yield conn.connect()
self.assertEqual(conn.stream, IOStream.return_value)
IOStream.return_value.connect.assert_called_once_with(
("localhost", 1234)
)
