class TestNailgunStreamWriter(unittest.TestCase):
def setUp(self):
self.in_fd = -1
self.mock_socket = mock.Mock()
self.writer = NailgunStreamWriter(
(self.in_fd,),
self.mock_socket,
(ChunkType.STDIN,),
ChunkType.STDIN_EOF
)
def test_stop(self):
self.assertFalse(self.writer.is_stopped)
self.writer.stop()
self.assertTrue(self.writer.is_stopped)
self.writer.run()
def test_startable(self):
self.assertTrue(inspect.ismethod(self.writer.start))
@mock.patch('select.select')
def test_run_stop_on_error(self, mock_select):
mock_select.return_value = ([], [], [self.in_fd])
self.writer.run()
self.assertFalse(self.writer.is_alive())
self.assertEquals(mock_select.call_count, 1)
@mock.patch('os.read')
@mock.patch('select.select')
@mock.patch.object(NailgunProtocol, 'write_chunk')
def test_run_read_write(self, mock_writer, mock_select, mock_read):
mock_select.side_effect = [
([self.in_fd], [], []),
([self.in_fd], [], [])
]
mock_read.side_effect = [
b'A' * 300,
b'' # Simulate EOF.
]
# Exercise NailgunStreamWriter.running() and .run() simultaneously.
inc = 0
with self.writer.running():
while self.writer.is_alive():
time.sleep(0.01)
inc += 1
if inc >= 1000:
raise Exception('waited too long.')
self.assertFalse(self.writer.is_alive())
mock_read.assert_called_with(-1, io.DEFAULT_BUFFER_SIZE)
self.assertEquals(mock_read.call_count, 2)
mock_writer.assert_has_calls([
mock.call(mock.ANY, ChunkType.STDIN, b'A' * 300),
mock.call(mock.ANY, ChunkType.STDIN_EOF)
])
class TestNailgunStreamWriter(unittest.TestCase):
def setUp(self):
self.in_fd = -1
self.mock_socket = unittest.mock.Mock()
self.writer = NailgunStreamWriter(
(self.in_fd,),
self.mock_socket,
(ChunkType.STDIN,),
ChunkType.STDIN_EOF
)
def test_stop(self):
self.assertFalse(self.writer.is_stopped)
self.writer.stop()
self.assertTrue(self.writer.is_stopped)
self.writer.run()
def test_startable(self):
self.assertTrue(inspect.ismethod(self.writer.start))
@unittest.mock.patch('select.select')
def test_run_stop_on_error(self, mock_select):
mock_select.return_value = ([], [], [self.in_fd])
self.writer.run()
self.assertFalse(self.writer.is_alive())
self.assertEqual(mock_select.call_count, 1)
@unittest.mock.patch('os.read')
@unittest.mock.patch('select.select')
@unittest.mock.patch.object(NailgunProtocol, 'write_chunk')
def test_run_read_write(self, mock_writer, mock_select, mock_read):
mock_select.side_effect = [
([self.in_fd], [], []),
([self.in_fd], [], [])
]
mock_read.side_effect = [
b'A' * 300,
b'' # Simulate EOF.
]
# Exercise NailgunStreamWriter.running() and .run() simultaneously.
inc = 0
with self.writer.running():
while self.writer.is_alive():
time.sleep(0.01)
inc += 1
if inc >= 1000:
raise Exception('waited too long.')
self.assertFalse(self.writer.is_alive())
mock_read.assert_called_with(-1, io.DEFAULT_BUFFER_SIZE)
self.assertEqual(mock_read.call_count, 2)
mock_writer.assert_has_calls([
unittest.mock.call(unittest.mock.ANY, ChunkType.STDIN, b'A' * 300),
unittest.mock.call(unittest.mock.ANY, ChunkType.STDIN_EOF)
])
class TestNailgunStreamWriter(unittest.TestCase):
def setUp(self):
self.in_fd = -1
self.mock_socket = unittest.mock.Mock()
self.writer = NailgunStreamWriter((self.in_fd, ), self.mock_socket,
(ChunkType.STDIN, ),
ChunkType.STDIN_EOF)
def test_stop(self):
self.assertFalse(self.writer.is_stopped)
self.writer.stop()
self.assertTrue(self.writer.is_stopped)
self.writer.run()
def test_startable(self):
self.assertTrue(inspect.ismethod(self.writer.start))
@unittest.mock.patch("select.select")
def test_run_stop_on_error(self, mock_select):
mock_select.return_value = ([], [], [self.in_fd])
self.writer.run()
self.assertFalse(self.writer.is_alive())
self.assertEqual(mock_select.call_count, 1)
@unittest.mock.patch("os.read")
@unittest.mock.patch("select.select")
@unittest.mock.patch.object(NailgunProtocol, "write_chunk")
def test_run_read_write(self, mock_writer, mock_select, mock_read):
mock_select.side_effect = [([self.in_fd], [], []),
([self.in_fd], [], [])]
mock_read.side_effect = [b"A" * 300, b""] # Simulate EOF.
# Exercise NailgunStreamWriter.running() and .run() simultaneously.
inc = 0
with self.writer.running():
while self.writer.is_alive():
time.sleep(0.01)
inc += 1
if inc >= 1000:
raise Exception("waited too long.")
self.assertFalse(self.writer.is_alive())
mock_read.assert_called_with(-1, io.DEFAULT_BUFFER_SIZE)
self.assertEqual(mock_read.call_count, 2)
mock_writer.assert_has_calls([
unittest.mock.call(unittest.mock.ANY, ChunkType.STDIN, b"A" * 300),
unittest.mock.call(unittest.mock.ANY, ChunkType.STDIN_EOF),
])
@unittest.mock.patch("os.close")
@unittest.mock.patch("os.read")
@unittest.mock.patch("select.select")
def test_run_exits_for_closed_and_errored_socket(self, mock_select,
mock_read, mock_close):
# When stdin is closed, select can indicate that it is both empty and errored.
mock_select.return_value = ([self.in_fd], [self.in_fd], [])
mock_read.return_value = b"" # EOF.
self.writer.run()
assert self.writer.is_alive() is False
assert mock_select.call_count == 1
