From a29e46c17a9d33fb6cddb2f43ecc9d3ca30e4e7c Mon Sep 17 00:00:00 2001
From: Valentin Lorentz <progval+git@progval.net>
Date: Sat, 26 Jun 2021 18:44:55 +0200
Subject: [PATCH] Add test for account tag on INVITE messages.

This will catch issues like https://github.com/solanum-ircd/solanum/issues/166
(when we have registration support for solanum)
---
 irctest/server_tests/test_account_tag.py | 18 +++++++++++++++++-
 irctest/specifications.py                |  1 +
 pytest.ini                               |  1 +
 3 files changed, 19 insertions(+), 1 deletion(-)

diff --git a/irctest/server_tests/test_account_tag.py b/irctest/server_tests/test_account_tag.py
index b6e1897..321783a 100644
--- a/irctest/server_tests/test_account_tag.py
+++ b/irctest/server_tests/test_account_tag.py
@@ -32,7 +32,7 @@ class AccountTagTestCase(cases.BaseServerTestCase, cases.OptionalityHelper):
         self.sendLine(2, "CAP END")
         self.skipToWelcome(2)
 
-    @cases.mark_specifications("IRCv3")
+    @cases.mark_capabilities("account-tag")
     @cases.OptionalityHelper.skipUnlessHasMechanism("PLAIN")
     def testPrivmsg(self):
         self.connectClient("foo", capabilities=["account-tag"], skip_if_cap_nak=True)
@@ -45,3 +45,19 @@ class AccountTagTestCase(cases.BaseServerTestCase, cases.OptionalityHelper):
         self.assertMessageMatch(
             m, command="PRIVMSG", params=["foo", "hi"], tags={"account": "jilles"}
         )
+
+    @cases.mark_capabilities("account-tag")
+    @cases.OptionalityHelper.skipUnlessHasMechanism("PLAIN")
+    def testInvite(self):
+        self.connectClient("foo", capabilities=["account-tag"], skip_if_cap_nak=True)
+        self.getMessages(1)
+        self.controller.registerUser(self, "jilles", "sesame")
+        self.connectRegisteredClient("bar")
+        self.sendLine(2, "JOIN #chan")
+        self.getMessages(2)
+        self.sendLine(2, "INVITE foo #chan")
+        self.getMessages(2)
+        m = self.getMessage(1)
+        self.assertMessageMatch(
+            m, command="INVITE", params=["foo", "#chan"], tags={"account": "jilles"}
+        )
diff --git a/irctest/specifications.py b/irctest/specifications.py
index f2011a7..4680744 100644
--- a/irctest/specifications.py
+++ b/irctest/specifications.py
@@ -27,6 +27,7 @@ class Specifications(enum.Enum):
 
 @enum.unique
 class Capabilities(enum.Enum):
+    ACCOUNT_TAG = "account-tag"
     AWAY_NOTIFY = "away-notify"
     BATCH = "batch"
     ECHO_MESSAGE = "echo-message"
diff --git a/pytest.ini b/pytest.ini
index 5c218ba..2b41119 100644
--- a/pytest.ini
+++ b/pytest.ini
@@ -13,6 +13,7 @@ markers =
     deprecated
 
     # capabilities
+    account-tag
     away-notify
     batch
     echo-message