tests: port test_git_import_export.rs to TestWorkDir API

This commit is contained in:
Yuya Nishihara 2025-03-20 00:18:46 +09:00
parent 569a405cea
commit ca616fade2

View File

@ -11,7 +11,6 @@
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
use std::path::Path;
use itertools::Itertools as _;
use jj_lib::backend::CommitId;
@ -19,27 +18,28 @@ use testutils::git;
use crate::common::CommandOutput;
use crate::common::TestEnvironment;
use crate::common::TestWorkDir;
#[test]
fn test_resolution_of_git_tracking_bookmarks() {
let test_env = TestEnvironment::default();
test_env.run_jj_in(".", ["git", "init", "repo"]).success();
let repo_path = test_env.env_root().join("repo");
test_env
.run_jj_in(&repo_path, ["bookmark", "create", "-r@", "main"])
let work_dir = test_env.work_dir("repo");
work_dir
.run_jj(["bookmark", "create", "-r@", "main"])
.success();
test_env
.run_jj_in(&repo_path, ["describe", "-r", "main", "-m", "old_message"])
work_dir
.run_jj(["describe", "-r", "main", "-m", "old_message"])
.success();
// Create local-git tracking bookmark
let output = test_env.run_jj_in(&repo_path, ["git", "export"]);
let output = work_dir.run_jj(["git", "export"]);
insta::assert_snapshot!(output, @"");
// Move the local bookmark somewhere else
test_env
.run_jj_in(&repo_path, ["describe", "-r", "main", "-m", "new_message"])
work_dir
.run_jj(["describe", "-r", "main", "-m", "new_message"])
.success();
insta::assert_snapshot!(get_bookmark_output(&test_env, &repo_path), @r"
insta::assert_snapshot!(get_bookmark_output(&work_dir), @r"
main: qpvuntsm b61d21b6 (empty) new_message
@git (ahead by 1 commits, behind by 1 commits): qpvuntsm hidden 03757d22 (empty) old_message
[EOF]
@ -48,10 +48,7 @@ fn test_resolution_of_git_tracking_bookmarks() {
// Test that we can address both revisions
let query = |expr| {
let template = r#"commit_id ++ " " ++ description"#;
test_env.run_jj_in(
&repo_path,
["log", "-r", expr, "-T", template, "--no-graph"],
)
work_dir.run_jj(["log", "-r", expr, "-T", template, "--no-graph"])
};
insta::assert_snapshot!(query("main"), @r"
b61d21b660c17a7191f3f73873bfe7d3f7938628 new_message
@ -69,15 +66,15 @@ fn test_resolution_of_git_tracking_bookmarks() {
fn test_git_export_conflicting_git_refs() {
let test_env = TestEnvironment::default();
test_env.run_jj_in(".", ["git", "init", "repo"]).success();
let repo_path = test_env.env_root().join("repo");
let work_dir = test_env.work_dir("repo");
test_env
.run_jj_in(&repo_path, ["bookmark", "create", "-r@", "main"])
work_dir
.run_jj(["bookmark", "create", "-r@", "main"])
.success();
test_env
.run_jj_in(&repo_path, ["bookmark", "create", "-r@", "main/sub"])
work_dir
.run_jj(["bookmark", "create", "-r@", "main/sub"])
.success();
let output = test_env.run_jj_in(&repo_path, ["git", "export"]);
let output = work_dir.run_jj(["git", "export"]);
insta::with_settings!({filters => vec![("Failed to set: .*", "Failed to set: ...")]}, {
insta::assert_snapshot!(output, @r#"
------- stderr -------
@ -95,19 +92,19 @@ fn test_git_export_conflicting_git_refs() {
fn test_git_export_undo() {
let test_env = TestEnvironment::default();
test_env.run_jj_in(".", ["git", "init", "repo"]).success();
let repo_path = test_env.env_root().join("repo");
let git_repo = git::open(repo_path.join(".jj/repo/store/git"));
let work_dir = test_env.work_dir("repo");
let git_repo = git::open(work_dir.root().join(".jj/repo/store/git"));
test_env
.run_jj_in(&repo_path, ["bookmark", "create", "-r@", "a"])
work_dir
.run_jj(["bookmark", "create", "-r@", "a"])
.success();
insta::assert_snapshot!(get_bookmark_output(&test_env, &repo_path), @r"
insta::assert_snapshot!(get_bookmark_output(&work_dir), @r"
a: qpvuntsm 230dd059 (empty) (no description set)
[EOF]
");
let output = test_env.run_jj_in(&repo_path, ["git", "export"]);
let output = work_dir.run_jj(["git", "export"]);
insta::assert_snapshot!(output, @"");
insta::assert_snapshot!(test_env.run_jj_in(&repo_path, ["log", "-ra@git"]), @r"
insta::assert_snapshot!(work_dir.run_jj(["log", "-ra@git"]), @r"
@ qpvuntsm test.user@example.com 2001-02-03 08:05:07 a 230dd059
(empty) (no description set)
~
@ -116,7 +113,7 @@ fn test_git_export_undo() {
// Exported refs won't be removed by undoing the export, but the git-tracking
// bookmark is. This is the same as remote-tracking bookmarks.
let output = test_env.run_jj_in(&repo_path, ["op", "undo"]);
let output = work_dir.run_jj(["op", "undo"]);
insta::assert_snapshot!(output, @r"
------- stderr -------
Undid operation: edb40232c741 (2001-02-03 08:05:10) export git refs
@ -132,7 +129,7 @@ fn test_git_export_undo() {
),
]
"#);
insta::assert_snapshot!(test_env.run_jj_in(&repo_path, ["log", "-ra@git"]), @r"
insta::assert_snapshot!(work_dir.run_jj(["log", "-ra@git"]), @r"
------- stderr -------
Error: Revision `a@git` doesn't exist
Hint: Did you mean `a`?
@ -141,9 +138,9 @@ fn test_git_export_undo() {
");
// This would re-export bookmark "a" and create git-tracking bookmark.
let output = test_env.run_jj_in(&repo_path, ["git", "export"]);
let output = work_dir.run_jj(["git", "export"]);
insta::assert_snapshot!(output, @"");
insta::assert_snapshot!(test_env.run_jj_in(&repo_path, ["log", "-ra@git"]), @r"
insta::assert_snapshot!(work_dir.run_jj(["log", "-ra@git"]), @r"
@ qpvuntsm test.user@example.com 2001-02-03 08:05:07 a 230dd059
(empty) (no description set)
~
@ -155,12 +152,12 @@ fn test_git_export_undo() {
fn test_git_import_undo() {
let test_env = TestEnvironment::default();
test_env.run_jj_in(".", ["git", "init", "repo"]).success();
let repo_path = test_env.env_root().join("repo");
let git_repo = git::open(repo_path.join(".jj/repo/store/git"));
let work_dir = test_env.work_dir("repo");
let git_repo = git::open(work_dir.root().join(".jj/repo/store/git"));
// Create bookmark "a" in git repo
let commit_id = test_env
.run_jj_in(&repo_path, &["log", "-Tcommit_id", "--no-graph", "-r@"])
let commit_id = work_dir
.run_jj(&["log", "-Tcommit_id", "--no-graph", "-r@"])
.success()
.stdout
.into_raw();
@ -175,37 +172,37 @@ fn test_git_import_undo() {
.unwrap();
// Initial state we will return to after `undo`. There are no bookmarks.
insta::assert_snapshot!(get_bookmark_output(&test_env, &repo_path), @"");
let base_operation_id = test_env.work_dir(&repo_path).current_operation_id();
insta::assert_snapshot!(get_bookmark_output(&work_dir), @"");
let base_operation_id = work_dir.current_operation_id();
let output = test_env.run_jj_in(&repo_path, ["git", "import"]);
let output = work_dir.run_jj(["git", "import"]);
insta::assert_snapshot!(output, @r"
------- stderr -------
bookmark: a@git [new] tracked
[EOF]
");
insta::assert_snapshot!(get_bookmark_output(&test_env, &repo_path), @r"
insta::assert_snapshot!(get_bookmark_output(&work_dir), @r"
a: qpvuntsm 230dd059 (empty) (no description set)
@git: qpvuntsm 230dd059 (empty) (no description set)
[EOF]
");
// "git import" can be undone by default.
let output = test_env.run_jj_in(&repo_path, ["op", "restore", &base_operation_id]);
let output = work_dir.run_jj(["op", "restore", &base_operation_id]);
insta::assert_snapshot!(output, @r"
------- stderr -------
Restored to operation: eac759b9ab75 (2001-02-03 08:05:07) add workspace 'default'
[EOF]
");
insta::assert_snapshot!(get_bookmark_output(&test_env, &repo_path), @"");
insta::assert_snapshot!(get_bookmark_output(&work_dir), @"");
// Try "git import" again, which should re-import the bookmark "a".
let output = test_env.run_jj_in(&repo_path, ["git", "import"]);
let output = work_dir.run_jj(["git", "import"]);
insta::assert_snapshot!(output, @r"
------- stderr -------
bookmark: a@git [new] tracked
[EOF]
");
insta::assert_snapshot!(get_bookmark_output(&test_env, &repo_path), @r"
insta::assert_snapshot!(get_bookmark_output(&work_dir), @r"
a: qpvuntsm 230dd059 (empty) (no description set)
@git: qpvuntsm 230dd059 (empty) (no description set)
[EOF]
@ -216,12 +213,12 @@ fn test_git_import_undo() {
fn test_git_import_move_export_with_default_undo() {
let test_env = TestEnvironment::default();
test_env.run_jj_in(".", ["git", "init", "repo"]).success();
let repo_path = test_env.env_root().join("repo");
let git_repo = git::open(repo_path.join(".jj/repo/store/git"));
let work_dir = test_env.work_dir("repo");
let git_repo = git::open(work_dir.root().join(".jj/repo/store/git"));
// Create bookmark "a" in git repo
let commit_id = test_env
.run_jj_in(&repo_path, &["log", "-Tcommit_id", "--no-graph", "-r@"])
let commit_id = work_dir
.run_jj(&["log", "-Tcommit_id", "--no-graph", "-r@"])
.success()
.stdout
.into_raw();
@ -237,34 +234,34 @@ fn test_git_import_move_export_with_default_undo() {
// Initial state we will try to return to after `op restore`. There are no
// bookmarks.
insta::assert_snapshot!(get_bookmark_output(&test_env, &repo_path), @"");
let base_operation_id = test_env.work_dir(&repo_path).current_operation_id();
insta::assert_snapshot!(get_bookmark_output(&work_dir), @"");
let base_operation_id = work_dir.current_operation_id();
let output = test_env.run_jj_in(&repo_path, ["git", "import"]);
let output = work_dir.run_jj(["git", "import"]);
insta::assert_snapshot!(output, @r"
------- stderr -------
bookmark: a@git [new] tracked
[EOF]
");
insta::assert_snapshot!(get_bookmark_output(&test_env, &repo_path), @r"
insta::assert_snapshot!(get_bookmark_output(&work_dir), @r"
a: qpvuntsm 230dd059 (empty) (no description set)
@git: qpvuntsm 230dd059 (empty) (no description set)
[EOF]
");
// Move bookmark "a" and export to git repo
test_env.run_jj_in(&repo_path, ["new"]).success();
test_env
.run_jj_in(&repo_path, ["bookmark", "set", "a", "--to=@"])
work_dir.run_jj(["new"]).success();
work_dir
.run_jj(["bookmark", "set", "a", "--to=@"])
.success();
insta::assert_snapshot!(get_bookmark_output(&test_env, &repo_path), @r"
insta::assert_snapshot!(get_bookmark_output(&work_dir), @r"
a: yqosqzyt 096dc80d (empty) (no description set)
@git (behind by 1 commits): qpvuntsm 230dd059 (empty) (no description set)
[EOF]
");
let output = test_env.run_jj_in(&repo_path, ["git", "export"]);
let output = work_dir.run_jj(["git", "export"]);
insta::assert_snapshot!(output, @"");
insta::assert_snapshot!(get_bookmark_output(&test_env, &repo_path), @r"
insta::assert_snapshot!(get_bookmark_output(&work_dir), @r"
a: yqosqzyt 096dc80d (empty) (no description set)
@git: yqosqzyt 096dc80d (empty) (no description set)
[EOF]
@ -273,7 +270,7 @@ fn test_git_import_move_export_with_default_undo() {
// "git import" can be undone with the default `restore` behavior, as shown in
// the previous test. However, "git export" can't: the bookmarks in the git
// repo stay where they were.
let output = test_env.run_jj_in(&repo_path, ["op", "restore", &base_operation_id]);
let output = work_dir.run_jj(["op", "restore", &base_operation_id]);
insta::assert_snapshot!(output, @r"
------- stderr -------
Restored to operation: eac759b9ab75 (2001-02-03 08:05:07) add workspace 'default'
@ -281,7 +278,7 @@ fn test_git_import_move_export_with_default_undo() {
Parent commit : zzzzzzzz 00000000 (empty) (no description set)
[EOF]
");
insta::assert_snapshot!(get_bookmark_output(&test_env, &repo_path), @"");
insta::assert_snapshot!(get_bookmark_output(&work_dir), @"");
insta::assert_debug_snapshot!(get_git_repo_refs(&git_repo), @r#"
[
(
@ -295,13 +292,13 @@ fn test_git_import_move_export_with_default_undo() {
// The last bookmark "a" state is imported from git. No idea what's the most
// intuitive result here.
let output = test_env.run_jj_in(&repo_path, ["git", "import"]);
let output = work_dir.run_jj(["git", "import"]);
insta::assert_snapshot!(output, @r"
------- stderr -------
bookmark: a@git [new] tracked
[EOF]
");
insta::assert_snapshot!(get_bookmark_output(&test_env, &repo_path), @r"
insta::assert_snapshot!(get_bookmark_output(&work_dir), @r"
a: yqosqzyt 096dc80d (empty) (no description set)
@git: yqosqzyt 096dc80d (empty) (no description set)
[EOF]
@ -309,8 +306,8 @@ fn test_git_import_move_export_with_default_undo() {
}
#[must_use]
fn get_bookmark_output(test_env: &TestEnvironment, repo_path: &Path) -> CommandOutput {
test_env.run_jj_in(repo_path, ["bookmark", "list", "--all-remotes"])
fn get_bookmark_output(work_dir: &TestWorkDir) -> CommandOutput {
work_dir.run_jj(["bookmark", "list", "--all-remotes"])
}
fn get_git_repo_refs(git_repo: &gix::Repository) -> Vec<(bstr::BString, CommitId)> {