Add a main() function to all custom servers based on isctest.asyncserver
and move server startup code there. This prevents redefining variables
from outer scope in custom server code as it evolves.
(cherry picked from commit
8cb51d4c2b79795124b1a9e9e1a08e3008d65260)
yield send_delegation(qctx, b_stale, "10.53.0.4")
-if __name__ == "__main__":
+def main() -> None:
server = AsyncDnsServer()
server.install_response_handler(QueryLogger())
server.install_response_handler(BadHandler())
server.install_response_handler(SlowHandler())
server.install_response_handler(StaleHandler())
server.run()
+
+
+if __name__ == "__main__":
+ main()
delay = 0.4
-if __name__ == "__main__":
+def main() -> None:
server = AsyncDnsServer()
server.install_response_handler(QueryLogger())
server.install_response_handler(ZoopBoingBadHandler())
server.install_response_handler(ZoopBoingUglyHandler())
server.install_response_handler(ZoopBoingSlowHandler())
server.run()
+
+
+if __name__ == "__main__":
+ main()
delay = 0.4
-if __name__ == "__main__":
+def main() -> None:
server = AsyncDnsServer()
server.install_response_handler(QueryLogger())
server.install_response_handler(StaleHandler())
server.install_response_handler(IckyPtangZoopBoingUglyHandler())
server.install_response_handler(IckyPtangZoopBoingSlowHandler())
server.run()
+
+
+if __name__ == "__main__":
+ main()
from isctest.asyncserver import AsyncDnsServer, IgnoreAllQueries
-if __name__ == "__main__":
+def main() -> None:
server = AsyncDnsServer()
server.install_response_handler(IgnoreAllQueries())
server.run()
+
+
+if __name__ == "__main__":
+ main()