Fix linter errors in tests

Signed-off-by: Teodora Sechkova <[email protected]>
theupdateframework · Dec 15, 2021 · 2562aff · 2562aff
1 parent 26a5f81
commit 2562aff
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/tests/test_updater_consistent_snapshot.py b/tests/test_updater_consistent_snapshot.py
@@ -39,6 +39,7 @@ def setUp(self) -> None:
         self.targets_dir = os.path.join(self.temp_dir.name, "targets")
         os.mkdir(self.metadata_dir)
         os.mkdir(self.targets_dir)
+        self.sim: RepositorySimulator
 
     def tearDown(self) -> None:
         self.temp_dir.cleanup()
@@ -55,7 +56,7 @@ def setup_subtest(
             self.sim.dump_dir = os.path.join(self.dump_dir, name)
             os.mkdir(self.sim.dump_dir)
 
-    def teardown_subtest(self):
+    def teardown_subtest(self) -> None:
         if self.dump_dir is not None:
             self.sim.write()
 

diff --git a/tests/test_updater_delegation_graphs.py b/tests/test_updater_delegation_graphs.py
@@ -52,6 +52,7 @@ class TestDelegationsGraphs(unittest.TestCase):
     dump_dir: Optional[str] = None
 
     def setUp(self) -> None:
+        # pylint: disable=consider-using-with
         self.subtest_count = 0
         self.temp_dir = tempfile.TemporaryDirectory()
         self.metadata_dir = os.path.join(self.temp_dir.name, "metadata")