1 files changed, 71 insertions, 0 deletions

diff --git a/tests/git_remote_helper.rs b/tests/git_remote_helper.rs
index d473214..ab29726 100644
--- a/tests/git_remote_helper.rs
+++ b/tests/git_remote_helper.rs
@@ -177,3 +177,74 @@ mod list {
         async_run_test().await
     }
 }
+
+mod fetch {
+
+    use super::*;
+
+    async fn async_run_test() -> Result<()> {
+        let source_git_repo = prep_git_repo()?;
+        std::fs::write(source_git_repo.dir.join("commit.md"), "some content")?;
+        let main_commit_id = source_git_repo.stage_and_commit("commit.md")?;
+
+        source_git_repo.create_branch("vnext")?;
+        source_git_repo.checkout("vnext")?;
+        std::fs::write(source_git_repo.dir.join("vnext.md"), "some content")?;
+        let vnext_commit_id = source_git_repo.stage_and_commit("vnext.md")?;
+
+        let git_repo = prep_git_repo()?;
+        let events = vec![
+            generate_test_key_1_metadata_event("fred"),
+            generate_test_key_1_relay_list_event(),
+            generate_repo_ref_event_with_git_server(source_git_repo.dir.to_str().unwrap()),
+        ];
+        // fallback (51,52) user write (53, 55) repo (55, 56) blaster (57)
+        let (mut r51, mut r52, mut r53, mut r55, mut r56, mut r57) = (
+            Relay::new(8051, None, None),
+            Relay::new(8052, None, None),
+            Relay::new(8053, None, None),
+            Relay::new(8055, None, None),
+            Relay::new(8056, None, None),
+            Relay::new(8057, None, None),
+        );
+        r51.events = events.clone();
+        r55.events = events;
+
+        let cli_tester_handle = std::thread::spawn(move || -> Result<()> {
+            assert!(git_repo.git_repo.find_commit(main_commit_id).is_err());
+            assert!(git_repo.git_repo.find_commit(vnext_commit_id).is_err());
+
+            let mut p = cli_tester_after_fetch(&git_repo)?;
+            p.send_line(format!("fetch {main_commit_id} main").as_str())?;
+            p.send_line(format!("fetch {vnext_commit_id} vnext").as_str())?;
+            p.send_line("")?;
+            p.expect("\r\n")?;
+
+            assert!(git_repo.git_repo.find_commit(main_commit_id).is_ok());
+            assert!(git_repo.git_repo.find_commit(vnext_commit_id).is_ok());
+
+            p.exit()?;
+            for p in [51, 52, 53, 55, 56, 57] {
+                relay::shutdown_relay(8000 + p)?;
+            }
+            Ok(())
+        });
+        // launch relays
+        let _ = join!(
+            r51.listen_until_close(),
+            r52.listen_until_close(),
+            r53.listen_until_close(),
+            r55.listen_until_close(),
+            r56.listen_until_close(),
+            r57.listen_until_close(),
+        );
+        cli_tester_handle.join().unwrap()?;
+        Ok(())
+    }
+
+    #[tokio::test]
+    #[serial]
+    async fn fetch_downloads_speficied_branch_ref_commits_from_git_server() -> Result<()> {
+        async_run_test().await
+    }
+}