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) )