utils: adjusts aa-notify tests to handle Python 3.13+

zyga · zyga · commit 6336465edfc9 · 2025-01-28T20:49:16.000+01:00
Python 3.13 changes the formatting of long-short option pairs that use a meta-variable. Up until 3.13 the meta-variable was repeated. Since Python change [1] the meta-var is only printed once. [1] python/cpython#103372 Signed-off-by: Zygmunt Krynicki <zygmunt.krynicki@canonical.com>
diff --git a/utils/test/test-aa-notify.py b/utils/test/test-aa-notify.py
@@ -209,14 +209,13 @@ def test_help_contents(self):
   -p, --poll            poll AppArmor logs and display notifications
   --display DISPLAY     set the DISPLAY environment variable (might be needed if
                         sudo resets $DISPLAY)
-  -f FILE, --file FILE  search FILE for AppArmor messages
+  -f, --file FILE       search FILE for AppArmor messages
   -l, --since-last      display stats since last login
-  -s NUM, --since-days NUM
-                        show stats for last NUM days (can be used alone or with
+  -s, --since-days NUM  show stats for last NUM days (can be used alone or with
                         -p)
   -v, --verbose         show messages with stats
-  -u USER, --user USER  user to drop privileges to when not using sudo
-  -w NUM, --wait NUM    wait NUM seconds before displaying notifications (with
+  -u, --user USER       user to drop privileges to when not using sudo
+  -w, --wait NUM        wait NUM seconds before displaying notifications (with
                         -p)
   -m, --merge-notifications
                         Merge notification for improved readability (with -p)
@@ -240,6 +239,28 @@ def test_help_contents(self):
                         regular expression to match the network socket type
 '''  # noqa: E128
 
+        if sys.version_info[:2] < (3, 13):
+            # Python 3.13 tweaked argparse output [1]. When running on older
+            # Python versions, we adapt the expected output to match.
+            #
+            # https://github.com/python/cpython/pull/103372
+            patches = [(
+                '  -f, --file FILE       search FILE for AppArmor messages',
+                '  -f FILE, --file FILE  search FILE for AppArmor messages',
+            ), (
+                '  -s, --since-days NUM  show stats for last NUM days (can be used alone or with',
+                '  -s NUM, --since-days NUM\n'
+                + '                        show stats for last NUM days (can be used alone or with',
+            ), (
+                '  -u, --user USER       user to drop privileges to when not using sudo',
+                '  -u USER, --user USER  user to drop privileges to when not using sudo',
+            ), (
+                '  -w, --wait NUM        wait NUM seconds before displaying notifications (with',
+                '  -w NUM, --wait NUM    wait NUM seconds before displaying notifications (with',
+            )]
+            for patch in patches:
+                expected_output_2 = expected_output_2.replace(patch[0], patch[1])
+
         return_code, output = cmd(aanotify_bin + ['--help'])
         result = 'Got return code {}, expected {}\n'.format(return_code, expected_return_code)
         self.assertEqual(expected_return_code, return_code, result + output)
