def setUp(self): self.client_sock, self.server_sock = socket.socketpair() self.mock_socket = mock.Mock() self.mock_server = mock.create_autospec(PailgunServer, spec_set=True) self.handler = PailgunHandler(self.server_sock, self.client_sock.getsockname()[:2], self.mock_server)
def setUp(self): self.client_sock, self.server_sock = socket.socketpair() self.mock_socket = mock.Mock() self.mock_server = mock.create_autospec(PailgunServer, spec_set=True) self.handler = PailgunHandler( self.server_sock, self.client_sock.getsockname()[:2], self.mock_server )
class TestPailgunHandler(unittest.TestCase): def setUp(self): self.client_sock, self.server_sock = socket.socketpair() self.mock_socket = unittest.mock.Mock() self.mock_server = unittest.mock.create_autospec(PailgunServer, spec_set=True) self.handler = PailgunHandler( self.server_sock, self.client_sock.getsockname()[:2], self.mock_server ) def test_handle_error(self): self.handler.handle_error() maybe_shutdown_socket = MaybeShutdownSocket(self.client_sock) last_chunk_type, last_payload = list(NailgunProtocol.iter_chunks(maybe_shutdown_socket))[-1] self.assertEqual(last_chunk_type, ChunkType.EXIT) self.assertEqual(last_payload, "1") @unittest.mock.patch.object(PailgunHandler, "_run_pants", **PATCH_OPTS) def test_handle_request(self, mock_run_pants): NailgunProtocol.send_request(self.client_sock, "/test", "./pants", "help-advanced") self.handler.handle_request() self.assertIs(mock_run_pants.called, True)
class TestPailgunHandler(unittest.TestCase): def setUp(self): self.client_sock, self.server_sock = socket.socketpair() self.mock_socket = mock.Mock() self.mock_server = mock.create_autospec(PailgunServer, spec_set=True) self.handler = PailgunHandler( self.server_sock, self.client_sock.getsockname()[:2], self.mock_server ) def test_handle_error(self): self.handler.handle_error() last_chunk_type, last_payload = list(NailgunProtocol.iter_chunks(self.client_sock))[-1] self.assertEquals(last_chunk_type, ChunkType.EXIT) self.assertEquals(bytes(last_payload), '1') @mock.patch.object(PailgunHandler, '_run_pants', **PATCH_OPTS) def test_handle_request(self, mock_run_pants): NailgunProtocol.send_request(self.client_sock, '/test', './pants', 'help-advanced') self.handler.handle_request() self.assertIs(mock_run_pants.called, True)