Patch release. (#97)

* Patch release. * Add the tests in tokio.
huggingface · Feb 26, 2025 · 357331d · 357331d
1 parent 17a5969
commit 357331d
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 3 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "hf-hub"
-version = "0.4.1"
+version = "0.4.2"
 edition = "2021"
 homepage = "https://github.com/huggingface/hf-hub"
 license = "Apache-2.0"

diff --git a/src/api/sync.rs b/src/api/sync.rs
@@ -1266,7 +1266,7 @@ mod tests {
         let headers = api.client.headers;
         assert_eq!(
             headers.get(USER_AGENT),
-            Some(&"unknown/None; hf-hub/0.4.1; rust/unknown".to_string())
+            Some(&"unknown/None; hf-hub/0.4.2; rust/unknown".to_string())
         );
     }
 
@@ -1279,7 +1279,7 @@ mod tests {
         let headers = api.client.headers;
         assert_eq!(
             headers.get(USER_AGENT),
-            Some(&"unknown/None; hf-hub/0.4.1; rust/unknown; origin/custom".to_string())
+            Some(&"unknown/None; hf-hub/0.4.2; rust/unknown; origin/custom".to_string())
         );
     }
 

diff --git a/src/api/tokio.rs b/src/api/tokio.rs
@@ -1389,6 +1389,35 @@ mod tests {
         );
     }
 
+    #[test]
+    fn headers_default() {
+        let headers = ApiBuilder::new().build_headers().unwrap();
+        assert_eq!(
+            headers.get(USER_AGENT),
+            Some(
+                &"unknown/None; hf-hub/0.4.2; rust/unknown"
+                    .try_into()
+                    .unwrap()
+            )
+        );
+    }
+
+    #[test]
+    fn headers_custom() {
+        let headers = ApiBuilder::new()
+            .with_user_agent("origin", "custom")
+            .build_headers()
+            .unwrap();
+        assert_eq!(
+            headers.get(USER_AGENT),
+            Some(
+                &"unknown/None; hf-hub/0.4.2; rust/unknown; origin/custom"
+                    .try_into()
+                    .unwrap()
+            )
+        );
+    }
+
     // #[tokio::test]
     // async fn real() {
     //     let api = Api::new().unwrap();