def test_server_complete(): """ Test ParadropAPIServer complete method """ from paradrop.backend.pdfcd.server import ParadropAPIServer reactor = MagicMock() reactor.callFromThread = call server = ParadropAPIServer(reactor) update = MagicMock() server.complete(update) # Test the private _complete method. update = MagicMock() update.result = {} server._complete(update) assert update.pkg.request.write.called assert update.pkg.request.finish.called update.pkg.request.write.side_effect = Exception("Boom!") server._complete(update)