Я немного поиграл с веб-сервером Tornado и пришел к тому, что хочу остановить веб-сервер (например, во время модульного тестирования). Следующий простой пример существует на веб-странице Tornado:
import tornado.ioloop
import tornado.web
class MainHandler(tornado.web.RequestHandler):
def get(self):
self.write("Hello, world")
application = tornado.web.Application([
(r"/", MainHandler),
])
if __name__ == "__main__":
application.listen(8888)
tornado.ioloop.IOLoop.instance().start()
Как только tornado.ioloop.IOLoop.instance().start()
вызывается, он блокирует программу (или текущий поток). Чтение исходного кода для объекта IOLoop
дает этот пример в документации для функции stop
:
To use asynchronous methods from otherwise-synchronous code (such as
unit tests), you can start and stop the event loop like this:
ioloop = IOLoop()
async_method(ioloop=ioloop, callback=ioloop.stop)
ioloop.start()
ioloop.start() will return after async_method has run its callback,
whether that callback was invoked before or after ioloop.start.
Однако я не знаю, как интегрировать это в мою программу. На самом деле у меня есть класс, который инкапсулирует веб-сервер (имея собственные функции start
и stop
), но как только я начну звонить, программа (или тесты), конечно же, заблокирует.
Я попытался запустить веб-сервер в другом процессе (используя пакет multiprocessing
). Это класс, который обертывает веб-сервер:
class Server:
def __init__(self, port=8888):
self.application = tornado.web.Application([ (r"/", Handler) ])
def server_thread(application, port):
http_server = tornado.httpserver.HTTPServer(application)
http_server.listen(port)
tornado.ioloop.IOLoop.instance().start()
self.process = Process(target=server_thread,
args=(self.application, port,))
def start(self):
self.process.start()
def stop(self):
ioloop = tornado.ioloop.IOLoop.instance()
ioloop.add_callback(ioloop.stop)
Однако остановка, похоже, не полностью останавливает веб-сервер, поскольку она все еще работает в следующем тесте, даже с этой тестовой настройкой:
def setup_method(self, _function):
self.server = Server()
self.server.start()
time.sleep(0.5) # Wait for web server to start
def teardown_method(self, _function):
self.kstore.stop()
time.sleep(0.5)
Как я могу запустить и остановить веб-сервер Tornado из программы Python?