Add additional assertions to test_base.py

Sineaggi · Sineaggi · commit fb6eb3d5eb56 · 2023-02-02T10:00:18.000-07:00
diff --git a/test/test_base.py b/test/test_base.py
@@ -9,6 +9,7 @@
 import tempfile
 from unittest import SkipTest, skipIf
 
+from git import Repo
 from git.objects import Blob, Tree, Commit, TagObject
 from git.compat import is_win
 from git.objects.util import get_object_type_by_name
@@ -95,15 +96,18 @@ def test_object_resolution(self):
 self.assertEqual(self.rorepo.head.reference.object, self.rorepo.active_branch.object)
 
 @with_rw_repo("HEAD", bare=True)
- def test_with_bare_rw_repo(self, bare_rw_repo):
+ def test_with_bare_rw_repo(self, bare_rw_repo: Repo):
 assert bare_rw_repo.config_reader("repository").getboolean("core", "bare")
 assert osp.isfile(osp.join(bare_rw_repo.git_dir, "HEAD"))
 assert osp.isdir(bare_rw_repo.working_dir)
+ assert bare_rw_repo.working_tree_dir is None
 
 @with_rw_repo("0.1.6")
- def test_with_rw_repo(self, rw_repo):
+ def test_with_rw_repo(self, rw_repo: Repo):
 assert not rw_repo.config_reader("repository").getboolean("core", "bare")
+ assert osp.isdir(rw_repo.working_tree_dir)
 assert osp.isdir(osp.join(rw_repo.working_tree_dir, "lib"))
+ assert osp.isdir(rw_repo.working_dir)
 
 @skipIf(HIDE_WINDOWS_FREEZE_ERRORS, "FIXME: Freezes! sometimes...")
 @with_rw_and_rw_remote_repo("0.1.6")

-Original file line number
+Diff line change
 importtempfile
 fromunittestimportSkipTest, skipIf
 +fromgitimportRepo
 fromgit.objectsimportBlob, Tree, Commit, TagObject
 fromgit.compatimportis_win
 fromgit.objects.utilimportget_object_type_by_name
 self.assertEqual(self.rorepo.head.reference.object, self.rorepo.active_branch.object)
 @with_rw_repo("HEAD", bare=True)
 -deftest_with_bare_rw_repo(self, bare_rw_repo):
 +deftest_with_bare_rw_repo(self, bare_rw_repo: Repo):
 assertbare_rw_repo.config_reader("repository").getboolean("core", "bare")
 assertosp.isfile(osp.join(bare_rw_repo.git_dir, "HEAD"))
 assertosp.isdir(bare_rw_repo.working_dir)
 +assertbare_rw_repo.working_tree_dirisNone
 @with_rw_repo("0.1.6")
 -deftest_with_rw_repo(self, rw_repo):
 +deftest_with_rw_repo(self, rw_repo: Repo):
 assertnotrw_repo.config_reader("repository").getboolean("core", "bare")
 +assertosp.isdir(rw_repo.working_tree_dir)
 assertosp.isdir(osp.join(rw_repo.working_tree_dir, "lib"))
 +assertosp.isdir(rw_repo.working_dir)
 @skipIf(HIDE_WINDOWS_FREEZE_ERRORS, "FIXME: Freezes! sometimes...")
 @with_rw_and_rw_remote_repo("0.1.6")