pkgdev mask: Test both -r and --rites

Signed-off-by: Thomas Bracht Laumann Jespersen <t@laumann.xyz>

Author: laumann

tests/scripts/test_pkgdev_mask.py

@@ -182,18 +182,20 @@ def test_existing_masks(self):
         assert self.profile.masks == frozenset([atom_cls('cat/masked'), atom_cls('=cat/pkg-0')])
 
     def test_last_rites(self):
-        with os_environ(EDITOR="sed -i '1s/$/mask comment/'"), \
-                patch('sys.argv', self.args + ['cat/pkg', '-r']), \
-                pytest.raises(SystemExit), \
-                chdir(pjoin(self.repo.path)):
-            self.script()
-
         removal_date = self.today + timedelta(days=30)
         today = self.today.strftime('%Y-%m-%d')
         removal = removal_date.strftime('%Y-%m-%d')
-        assert self.masks_path.read_text() == textwrap.dedent(f"""\
-            # First Last <first.last@email.com> ({today})
-            # mask comment
-            # Removal: {removal}
-            cat/pkg
-        """)
+        for rflag in ('-r', '--rites'):
+            with os_environ(EDITOR="sed -i '1s/$/mask comment/'"), \
+                    patch('sys.argv', self.args + ['cat/pkg', rflag]), \
+                    pytest.raises(SystemExit), \
+                    chdir(pjoin(self.repo.path)):
+                self.script()
+
+            assert self.masks_path.read_text() == textwrap.dedent(f"""\
+                # First Last <first.last@email.com> ({today})
+                # mask comment
+                # Removal: {removal}
+                cat/pkg
+            """)
+            self.masks_path.write_text("")  # Reset the contents of package.mask