tests get documents limit, offset, attr to retrieve

tests/common/index.rs

@@ -100,8 +100,20 @@ impl Index<'_> {
         self.service.get(url).await
     }
 
-    pub async fn get_all_documents(&self, _options: GetAllDocumentsOptions) -> (Value, StatusCode) {
-        let url = format!("/indexes/{}/documents", self.uid);
+    pub async fn get_all_documents(&self, options: GetAllDocumentsOptions) -> (Value, StatusCode) {
+        let mut url = format!("/indexes/{}/documents?", self.uid);
+        if let Some(limit) = options.limit {
+            url.push_str(&format!("limit={}&", limit));
+        }
+
+        if let Some(offset) = options.offset {
+            url.push_str(&format!("offset={}&", offset));
+        }
+
+        if let Some(attributes_to_retrieve) = options.attributes_to_retrieve {
+            url.push_str(&format!("attributesToRetrieve={}&", attributes_to_retrieve.join(",")));
+        }
+
         self.service.get(url).await
     }
 }
@@ -109,4 +121,8 @@ impl Index<'_> {
 pub struct GetDocumentOptions;
 
 #[derive(Debug, Default)]
-pub struct GetAllDocumentsOptions;
+pub struct GetAllDocumentsOptions {
+    pub limit: Option<usize>,
+    pub offset: Option<usize>,
+    pub attributes_to_retrieve: Option<Vec<&'static str>>,
+}

tests/common/mod.rs

@@ -3,6 +3,7 @@ mod server;
 mod service;
 
 pub use server::Server;
+pub use index::{GetDocumentOptions, GetAllDocumentsOptions};
 
 /// Performs a search test on both post and get routes
 #[macro_export]

tests/common/service.rs

@@ -23,6 +23,24 @@ impl Service {
         (response, status_code)
     }
 
+    /// Send a test post request from a text body, with a `content-type:application/json` header.
+    pub async fn post_str(&self, url: impl AsRef<str>, body: impl AsRef<str>) -> (Value, StatusCode) {
+        let mut app =
+            test::init_service(meilisearch_http::create_app(&self.0, true).wrap(NormalizePath)).await;
+
+        let req = test::TestRequest::post()
+            .uri(url.as_ref())
+            .set_payload(body.as_ref().to_string())
+            .header("content-type", "application/json")
+            .to_request();
+        let res = test::call_service(&mut app, req).await;
+        let status_code = res.status();
+
+        let body = test::read_body(res).await;
+        let response = serde_json::from_slice(&body).unwrap_or_default();
+        (response, status_code)
+    }
+
     pub async fn get(&self, url: impl AsRef<str>) -> (Value, StatusCode) {
         let mut app =
             test::init_service(meilisearch_http::create_app(&self.0, true).wrap(NormalizePath)).await;
@@ -64,6 +82,4 @@ impl Service {
         let response = serde_json::from_slice(&body).unwrap_or_default();
         (response, status_code)
     }
-
 }
-