update unit test

pixee · Oct 17, 2023 · 732c6a1 · 732c6a1
1 parent 4320ede
commit 732c6a1
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/test_codemodder.py b/tests/test_codemodder.py
@@ -98,10 +98,9 @@ def test_reporting(self, mock_reporting, dry_run):
         args_to_reporting = mock_reporting.call_args_list[0][0]
         assert len(args_to_reporting) == 4
         _, _, _, results_by_codemod = args_to_reporting
-        # assert len(results_by_codemod) == 2
 
-        for codemod_results in results_by_codemod:
-            assert len(codemod_results["changeset"]) > 0
+        registry = load_registered_codemods()
+        assert len(results_by_codemod) == len(registry.codemods)
 
     @mock.patch("codemodder.codemods.base_codemod.semgrep_run")
     def test_no_codemods_to_run(self, mock_semgrep_run):