From 50d3a8e6dae8ab756ca6d0c483103440f0384c89 Mon Sep 17 00:00:00 2001 From: Val Lorentz Date: Sat, 2 Sep 2023 13:43:35 +0200 Subject: [PATCH] echo_message: Simplify code (#219) --- irctest/server_tests/echo_message.py | 20 +++++++------------- 1 file changed, 7 insertions(+), 13 deletions(-) diff --git a/irctest/server_tests/echo_message.py b/irctest/server_tests/echo_message.py index 063028a..2eb7119 100644 --- a/irctest/server_tests/echo_message.py +++ b/irctest/server_tests/echo_message.py @@ -22,23 +22,17 @@ class EchoMessageTestCase(cases.BaseServerTestCase): @cases.mark_capabilities("echo-message") def testEchoMessage(self, command, solo, server_time): """""" - if server_time: - self.connectClient( - "baz", - capabilities=["echo-message", "server-time"], - skip_if_cap_nak=True, - ) - else: - self.connectClient( - "baz", - capabilities=["echo-message", "server-time"], - skip_if_cap_nak=True, - ) + capabilities = ["server-time"] if server_time else [] + + self.connectClient( + "baz", + capabilities=["echo-message", *capabilities], + skip_if_cap_nak=True, + ) self.sendLine(1, "JOIN #chan") if not solo: - capabilities = ["server-time"] if server_time else None self.connectClient("qux", capabilities=capabilities) self.sendLine(2, "JOIN #chan")