about summary refs log tree commit diff
diff options
context:
space:
mode:
authorFlorian Klink <flokli@flokli.de>2024-03-27T11·12+0100
committerflokli <flokli@flokli.de>2024-03-28T07·58+0000
commit3cb943460580cffb83b46c0ae3e7957eaf9170ef (patch)
tree94ed7b40805e00f77b4ee2760a9592960131ff4d
parent07a51c7dc9dd37205368eb0653e23e33cc04406f (diff)
refactor(tvix/store/nar/import): use rstest fixtures from castore r/7789
This stops using the gen_{blob,directory}_service() functions from
tvix_castore::utils in favor of the rstest-based fixtures.

Change-Id: I4c2fff2926f34af63d3bd5c29c8e939dbdf7e062
Reviewed-on: https://cl.tvl.fyi/c/depot/+/11274
Reviewed-by: Connor Brewster <cbrewster@hey.com>
Tested-by: BuildkiteCI
-rw-r--r--tvix/store/src/nar/import.rs32
1 files changed, 18 insertions, 14 deletions
diff --git a/tvix/store/src/nar/import.rs b/tvix/store/src/nar/import.rs
index 20b922a61f..6f4dcdea5d 100644
--- a/tvix/store/src/nar/import.rs
+++ b/tvix/store/src/nar/import.rs
@@ -223,6 +223,7 @@ mod test {
     use std::io::Cursor;
     use std::sync::Arc;
 
+    use rstest::*;
     use tokio_stream::StreamExt;
     use tvix_castore::blobservice::BlobService;
     use tvix_castore::directoryservice::DirectoryService;
@@ -231,17 +232,18 @@ mod test {
         HELLOWORLD_BLOB_DIGEST,
     };
     use tvix_castore::proto as castorepb;
-    use tvix_castore::utils::{gen_blob_service, gen_directory_service};
 
     use crate::tests::fixtures::{
-        NAR_CONTENTS_COMPLICATED, NAR_CONTENTS_HELLOWORLD, NAR_CONTENTS_SYMLINK,
+        blob_service, directory_service, NAR_CONTENTS_COMPLICATED, NAR_CONTENTS_HELLOWORLD,
+        NAR_CONTENTS_SYMLINK,
     };
 
+    #[rstest]
     #[tokio::test]
-    async fn single_symlink() {
-        let blob_service = gen_blob_service();
-        let directory_service = gen_directory_service();
-
+    async fn single_symlink(
+        blob_service: Arc<dyn BlobService>,
+        directory_service: Arc<dyn DirectoryService>,
+    ) {
         let handle = tokio::runtime::Handle::current();
 
         let root_node = handle
@@ -265,11 +267,12 @@ mod test {
         );
     }
 
+    #[rstest]
     #[tokio::test]
-    async fn single_file() {
-        let blob_service: Arc<dyn BlobService> = gen_blob_service().into();
-        let directory_service = gen_directory_service();
-
+    async fn single_file(
+        blob_service: Arc<dyn BlobService>,
+        directory_service: Arc<dyn DirectoryService>,
+    ) {
         let handle = tokio::runtime::Handle::current();
 
         let root_node = handle
@@ -301,11 +304,12 @@ mod test {
         assert!(blob_service.has(&HELLOWORLD_BLOB_DIGEST).await.unwrap());
     }
 
+    #[rstest]
     #[tokio::test]
-    async fn complicated() {
-        let blob_service: Arc<dyn BlobService> = gen_blob_service().into();
-        let directory_service: Arc<dyn DirectoryService> = gen_directory_service().into();
-
+    async fn complicated(
+        blob_service: Arc<dyn BlobService>,
+        directory_service: Arc<dyn DirectoryService>,
+    ) {
         let handle = tokio::runtime::Handle::current();
 
         let root_node = handle