testCapRemovalByClient: Support multiple CAP LS responses

progval · Sep 2, 2023 · 9f7876c · 9f7876c
1 parent 1c6a718
commit 9f7876c
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/irctest/server_tests/cap.py b/irctest/server_tests/cap.py
@@ -347,8 +347,13 @@ def testCapRemovalByClient(self):
         self.addClient(1)
         self.connectClient("sender")
         self.sendLine(1, "CAP LS 302")
-        m = self.getRegistrationMessage(1)
-        if not ({cap1, cap2} <= set(m.params[2].split())):
+        caps = set()
+        while True:
+            m = self.getRegistrationMessage(1)
+            caps.update(m.params[-1].split())
+            if m.params[2] != "*":
+                break
+        if not ({cap1, cap2} <= caps):
             raise CapabilityNotSupported(f"{cap1} or {cap2}")
         self.sendLine(1, f"CAP REQ :{cap1} {cap2}")
         self.sendLine(1, "nick bar")