Store index data in the snapshot, and resolve it as well.

Author: Byron

crates/but-workspace/tests/fixtures/scenario/unborn-all-file-types-added-to-index.sh

@@ -0,0 +1,15 @@
+#!/usr/bin/env bash
+
+### Description
+# A newly initialized git repository with an executable, a normal file, a symlink and a fifo, added to the index.
+set -eu -o pipefail
+
+git init
+echo content > untracked
+echo exe > untracked-exe && chmod +x untracked-exe
+ln -s untracked link
+mkdir dir
+mkfifo dir/fifo-should-be-ignored
+
+git add .
+

crates/but-workspace/tests/workspace/snapshot/index_create_and_resolve.rs

@@ -0,0 +1,67 @@
+use crate::snapshot::args_for_worktree_changes;
+use crate::utils::{read_only_in_memory_scenario, visualize_index};
+use but_testsupport::visualize_tree;
+use but_workspace::snapshot;
+use gix::prelude::ObjectIdExt;
+
+#[test]
+fn unborn_added_to_index() -> anyhow::Result<()> {
+    let repo = read_only_in_memory_scenario("unborn-all-file-types-added-to-index")?;
+    let (head_tree_id, state, no_workspace_and_meta) = args_for_worktree_changes(&repo)?;
+
+    let out = snapshot::create_tree(head_tree_id, state, no_workspace_and_meta)?;
+    insta::assert_snapshot!(visualize_tree(out.snapshot_tree.attach(&repo)), @r#""#);
+    insta::assert_debug_snapshot!(out, @r"");
+
+    let res_out = snapshot::resolve_tree(
+        out.snapshot_tree.attach(&repo),
+        out.head_tree,
+        snapshot::resolve_tree::Options::default(),
+    )?;
+    let mut cherry_pick = res_out
+        .worktree_cherry_pick
+        .expect("a worktree change was applied");
+    assert_eq!(cherry_pick.tree.write()?, out.worktree.unwrap());
+    let index = res_out
+        .index
+        .expect("the index was altered with many added files");
+    insta::assert_snapshot!(visualize_index(&index), @r"");
+
+    assert!(res_out.metadata.is_none());
+    assert!(
+        res_out.workspace_references.is_none(),
+        "didn't ask to store this"
+    );
+    Ok(())
+}
+
+#[test]
+fn with_conflicts() -> anyhow::Result<()> {
+    let repo = read_only_in_memory_scenario("merge-with-two-branches-conflict")?;
+    let (head_tree_id, state, no_workspace_and_meta) = args_for_worktree_changes(&repo)?;
+
+    let out = snapshot::create_tree(head_tree_id, state, no_workspace_and_meta)?;
+    insta::assert_snapshot!(visualize_tree(out.snapshot_tree.attach(&repo)), @r#""#);
+    insta::assert_debug_snapshot!(out, @r"");
+
+    let res_out = snapshot::resolve_tree(
+        out.snapshot_tree.attach(&repo),
+        out.head_tree,
+        snapshot::resolve_tree::Options::default(),
+    )?;
+    let mut cherry_pick = res_out
+        .worktree_cherry_pick
+        .expect("a worktree change was applied");
+    assert_eq!(cherry_pick.tree.write()?, out.worktree.unwrap());
+    let index = res_out
+        .index
+        .expect("the index was altered with many added files");
+    insta::assert_snapshot!(visualize_index(&index), @r"");
+
+    assert!(res_out.metadata.is_none());
+    assert!(
+        res_out.workspace_references.is_none(),
+        "didn't ask to store this"
+    );
+    Ok(())
+}

crates/but-workspace/tests/workspace/snapshot/mod.rs

@@ -1 +1,36 @@
+mod index_create_and_resolve;
 mod worktree_create_and_resolve;
+
+mod utils {
+    use but_graph::VirtualBranchesTomlMetadata;
+    use but_workspace::snapshot;
+
+    /// Produce all args needed for creating a snapshot tree, and assure everything is selected.
+    #[allow(clippy::type_complexity)]
+    pub fn args_for_worktree_changes(
+        repo: &gix::Repository,
+    ) -> anyhow::Result<(
+        gix::Id<'_>,
+        snapshot::create_tree::State,
+        Option<(
+            &'static but_graph::projection::Workspace<'static>,
+            &'static VirtualBranchesTomlMetadata,
+        )>,
+    )> {
+        let changes = but_core::diff::worktree_changes(repo)?;
+        let state = snapshot::create_tree::State {
+            selection: changes
+                .changes
+                .iter()
+                .map(|c| c.path.clone())
+                .chain(changes.ignored_changes.iter().map(|c| c.path.clone()))
+                .collect(),
+            changes,
+            head: false,
+        };
+        let head_tree_id = repo.head_tree_id_or_empty()?;
+
+        Ok((head_tree_id, state, None))
+    }
+}
+pub use utils::args_for_worktree_changes;

crates/but-workspace/tests/workspace/snapshot/worktree_create_and_resolve.rs

@@ -1,5 +1,5 @@
+use crate::snapshot::args_for_worktree_changes;
 use crate::utils::read_only_in_memory_scenario;
-use but_graph::VirtualBranchesTomlMetadata;
 use but_testsupport::visualize_tree;
 use but_workspace::snapshot;
 use gix::prelude::ObjectIdExt;
@@ -154,31 +154,3 @@ fn worktree_all_filetypes() -> anyhow::Result<()> {
     );
     Ok(())
 }
-
-/// Produce all args needed for creating a snapshot tree, and assure everything is selected.
-#[allow(clippy::type_complexity)]
-fn args_for_worktree_changes(
-    repo: &gix::Repository,
-) -> anyhow::Result<(
-    gix::Id<'_>,
-    snapshot::create_tree::State,
-    Option<(
-        &'static but_graph::projection::Workspace<'static>,
-        &'static VirtualBranchesTomlMetadata,
-    )>,
-)> {
-    let changes = but_core::diff::worktree_changes(repo)?;
-    let state = snapshot::create_tree::State {
-        selection: changes
-            .changes
-            .iter()
-            .map(|c| c.path.clone())
-            .chain(changes.ignored_changes.iter().map(|c| c.path.clone()))
-            .collect(),
-        changes,
-        head: false,
-    };
-    let head_tree_id = repo.head_tree_id_or_empty()?;
-
-    Ok((head_tree_id, state, None))
-}