From c9dfd644744e77783abe4675ed0817cfdb080d07 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Philipp=20Kr=C3=BCger?= <philipp.krueger1@gmail.com>
Date: Mon, 4 Mar 2024 11:17:06 +0100
Subject: [PATCH] chore: Test more lines in `messages.rs`: `indicates_finished`

---
 car-mirror/src/messages.rs | 48 +++++++++++++++++++++++++++++++++++---
 1 file changed, 45 insertions(+), 3 deletions(-)

diff --git a/car-mirror/src/messages.rs b/car-mirror/src/messages.rs
index 2d522a7..d1a2c04 100644
--- a/car-mirror/src/messages.rs
+++ b/car-mirror/src/messages.rs
@@ -93,7 +93,21 @@ mod test {
     use wnfs_common::MemoryBlockStore;
     use wnfs_unixfs_file::builder::FileBuilder;
 
-    async fn example_receiver_state() -> Result<ReceiverState> {
+    async fn loaded_receiver_state() -> Result<ReceiverState> {
+        let store = &MemoryBlockStore::new();
+
+        let root_cid = FileBuilder::new()
+            .content_bytes(vec![42; 500_000])
+            .build()?
+            .store(store)
+            .await?;
+
+        let dag = IncrementalDagVerification::new([root_cid], store, &NoCache).await?;
+
+        Ok(dag.into_receiver_state(Config::default().bloom_fpr))
+    }
+
+    async fn partial_receiver_state() -> Result<ReceiverState> {
         let store = &MemoryBlockStore::new();
         let store2 = &MemoryBlockStore::new();
 
@@ -118,7 +132,7 @@ mod test {
 
     #[test_log::test(async_std::test)]
     async fn test_encoding_format_json_concise() -> TestResult {
-        let receiver_state = example_receiver_state().await?;
+        let receiver_state = partial_receiver_state().await?;
         let pull_request: PullRequest = receiver_state.clone().into();
         let push_response: PushResponse = receiver_state.into();
 
@@ -132,7 +146,7 @@ mod test {
 
     #[test_log::test(async_std::test)]
     async fn test_dag_cbor_roundtrip() -> TestResult {
-        let receiver_state = example_receiver_state().await?;
+        let receiver_state = partial_receiver_state().await?;
         let pull_request: PullRequest = receiver_state.clone().into();
         let push_response: PushResponse = receiver_state.into();
 
@@ -144,4 +158,32 @@ mod test {
 
         Ok(())
     }
+
+    #[test_log::test(async_std::test)]
+    async fn test_pull_request_have_everything_indicates_finished() -> TestResult {
+        let pull_request: PullRequest = loaded_receiver_state().await?.into();
+        assert!(pull_request.indicates_finished());
+        Ok(())
+    }
+
+    #[test_log::test(async_std::test)]
+    async fn test_push_response_have_everything_indicates_finished() -> TestResult {
+        let push_response: PushResponse = loaded_receiver_state().await?.into();
+        assert!(push_response.indicates_finished());
+        Ok(())
+    }
+
+    #[test_log::test(async_std::test)]
+    async fn test_pull_request_partial_indicates_not_finished() -> TestResult {
+        let pull_request: PullRequest = partial_receiver_state().await?.into();
+        assert!(!pull_request.indicates_finished());
+        Ok(())
+    }
+
+    #[test_log::test(async_std::test)]
+    async fn test_push_response_partial_indicates_not_finished() -> TestResult {
+        let push_response: PushResponse = partial_receiver_state().await?.into();
+        assert!(!push_response.indicates_finished());
+        Ok(())
+    }
 }