From fe9f8654a3f998bb0c01bc4028fe1ca6fc4ebf9d Mon Sep 17 00:00:00 2001
From: Peter Rowlands <peter@pmrowla.com>
Date: Tue, 31 Jan 2023 14:25:31 +0900
Subject: [PATCH] testing: add test for worktree update

---
 dvc/testing/remote_tests.py | 23 +++++++++++++++++++++++
 1 file changed, 23 insertions(+)

diff --git a/dvc/testing/remote_tests.py b/dvc/testing/remote_tests.py
index 5bfe8fc42..30b189446 100644
--- a/dvc/testing/remote_tests.py
+++ b/dvc/testing/remote_tests.py
@@ -243,3 +243,26 @@ class TestRemoteWorktree:
         scm.checkout(v1)
         dvc.pull()
         assert (tmp_dir / "data_dir" / "data").read_text() == "data"
+
+    def test_update(
+        self, tmp_dir, dvc, remote_worktree  # pylint: disable=W0613
+    ):
+        tmp_dir.dvc_gen("foo", "foo")
+        tmp_dir.dvc_gen(
+            {
+                "data_dir": {
+                    "data_sub_dir": {"data_sub": "data_sub"},
+                    "data": "data",
+                    "empty": "",
+                }
+            }
+        )
+        dvc.push()
+        (remote_worktree / "foo").write_text("bar")
+        (remote_worktree / "data_dir" / "data").write_text("modified")
+        (remote_worktree / "data_dir" / "new_data").write_text("new data")
+
+        dvc.update([str(tmp_dir / "foo.dvc"), str(tmp_dir / "data_dir.dvc")])
+        assert (tmp_dir / "foo").read_text() == "bar"
+        assert (tmp_dir / "data_dir" / "data").read_text() == "modified"
+        assert (tmp_dir / "data_dir" / "new_data").read_text() == "new data"