upgrade test

Author: thehighestmath

list.txt

@@ -0,0 +1 @@
+moevm/github_repo_commitment_calc

requirements.txt

@@ -5,4 +5,5 @@ pandas==2.2.3
 pytz==2024.2
 requests==2.32.3
 pyforgejo==2.0.0
-isodate==0.7.2
+isodate==0.7.2
+unittest-parametrize==1.6.0

test_token_usage.py

@@ -5,6 +5,10 @@
 import git_logger
 from main import run
 
+from unittest_parametrize import parametrize
+from unittest_parametrize import ParametrizedTestCase
+from unittest_parametrize import param
+
 
 def parse_args(args):
     parser = argparse.ArgumentParser()
@@ -24,7 +28,7 @@ def parse_args(args):
     return parser.parse_args(args)
 
 
-class TestTokenUsage(unittest.TestCase):
+class TestTokenUsage(ParametrizedTestCase):
     def setUp(self):
         test_args = parse_args(sys.argv[1:])
         self.tokens = [test_args.tt1, test_args.tt2]
@@ -78,60 +82,21 @@ def _get_usage(self, binded_repos, clients):
 
         return limit_start, limit_finish
 
-    def test_commits_parser(self):
-        self.args.commits = True
-        for i in range(2):
-            clients = git_logger.Clients(
-                "github", self._change_tokens_order(self.tokens, i)
-            )
-            binded_repos = git_logger.get_next_binded_repo(clients, [self.repo])
-
-            limit_start, limit_finish = self._get_usage(binded_repos, clients)
-
-            self.assertTrue(self._is_only_one_token_used(limit_start, limit_finish))
-            self.assertTrue(self._is_max_token_used(limit_start, limit_finish))
-
-    def test_contributors_parser(self):
-        self.args.contributors = True
-        for i in range(2):
-            clients = git_logger.Clients(
-                "github", self._change_tokens_order(self.tokens, i)
-            )
-            binded_repos = git_logger.get_next_binded_repo(clients, [self.repo])
-
-            limit_start, limit_finish = self._get_usage(binded_repos, clients)
-
-            self.assertTrue(self._is_only_one_token_used(limit_start, limit_finish))
-            self.assertTrue(self._is_max_token_used(limit_start, limit_finish))
-
-    def test_issues_parser(self):
-        self.args.issues = True
-        for i in range(2):
-            clients = git_logger.Clients(
-                "github", self._change_tokens_order(self.tokens, i)
-            )
-            binded_repos = git_logger.get_next_binded_repo(clients, [self.repo])
-
-            limit_start, limit_finish = self._get_usage(binded_repos, clients)
-
-            self.assertTrue(self._is_only_one_token_used(limit_start, limit_finish))
-            self.assertTrue(self._is_max_token_used(limit_start, limit_finish))
-
-    def test_invites_parser(self):
-        self.args.invites = True
-        for i in range(2):
-            clients = git_logger.Clients(
-                "github", self._change_tokens_order(self.tokens, i)
-            )
-            binded_repos = git_logger.get_next_binded_repo(clients, [self.repo])
-
-            limit_start, limit_finish = self._get_usage(binded_repos, clients)
-
-            self.assertTrue(self._is_only_one_token_used(limit_start, limit_finish))
-            self.assertTrue(self._is_max_token_used(limit_start, limit_finish))
+    @parametrize(
+        'args',
+        [
+            param({'commits': True}, id='commits'),
+            param({'contributors': True}, id='contributors'),
+            param({'issues': True}, id='issues'),
+            param({'invites': True}, id='invites'),
+            param({'pull_requests': True}, id='pull_requests'),
+        ],
+    )
+    def test_commits_parser(self, args: dict[str, bool]):
+        # patch args
+        for k, v in args.items():
+            setattr(self.args, k, v)
 
-    def test_pull_requests_parser(self):
-        self.args.pull_requests = True
         for i in range(2):
             clients = git_logger.Clients(
                 "github", self._change_tokens_order(self.tokens, i)