fix(git): Fix update_repo when there are untracked files

jfpedroza · jfpedroza · commit caf47b55965e · 2022-09-24T23:18:25.000-05:00
diff --git a/libvcs/projects/git.py b/libvcs/projects/git.py
@@ -435,9 +435,9 @@ def update_repo(self, set_remotes: bool = False, *args, **kwargs):
             # to be able to perform git pull --rebase
             if need_stash:
                 # If Git < 1.7.6, uses --quiet --all
-                git_stash_save_options = "--quiet"
+                git_stash_save_options = ["--quiet", "--include-untracked"]
                 try:
-                    process = self.run(["stash", "save", git_stash_save_options])
+                    process = self.run(["stash", "save"] + git_stash_save_options)
                 except exc.CommandError:
                     self.log.error("Failed to stash changes")
 
diff --git a/tests/projects/test_git.py b/tests/projects/test_git.py
@@ -108,21 +108,39 @@ def test_repo_git_obtain_full(
 
 @pytest.mark.parametrize(
     # Postpone evaluation of options so fixture variables can interpolate
-    "constructor,lazy_constructor_options",
+    "constructor,lazy_constructor_options,has_untracked_files",
     [
         [
             GitProject,
             lambda git_remote_repo, tmp_path, **kwargs: {
                 "url": f"file://{git_remote_repo}",
                 "dir": tmp_path / "myrepo",
             },
+            False
+        ],
+        [
+            GitProject,
+            lambda git_remote_repo, tmp_path, **kwargs: {
+                "url": f"file://{git_remote_repo}",
+                "dir": tmp_path / "myrepo",
+            },
+            True
         ],
         [
             create_project_from_pip_url,
             lambda git_remote_repo, tmp_path, **kwargs: {
                 "pip_url": f"git+file://{git_remote_repo}",
                 "dir": tmp_path / "myrepo",
             },
+            False
+        ],
+        [
+            create_project_from_pip_url,
+            lambda git_remote_repo, tmp_path, **kwargs: {
+                "pip_url": f"git+file://{git_remote_repo}",
+                "dir": tmp_path / "myrepo",
+            },
+            True
         ],
     ],
 )
@@ -132,9 +150,16 @@ def test_repo_update_handle_cases(
     mocker: MockerFixture,
     constructor: ProjectTestFactory,
     lazy_constructor_options: ProjectTestFactoryLazyKwargs,
+    has_untracked_files: bool,
 ):
     git_repo: GitProject = constructor(**lazy_constructor_options(**locals()))
     git_repo.obtain()  # clone initial repo
+
+    if has_untracked_files:
+        some_file = git_repo.dir.joinpath("some_file")
+        with open(some_file, "w") as untracked_file:
+            untracked_file.write("some content")
+
     mocka = mocker.spy(git_repo, "run")
     git_repo.update_repo()
 
