Fix the authorization tests

This commit is contained in:
Kerollmops 2022-06-02 12:16:46 +02:00
parent 10a71fdb10
commit bcb51905d7
No known key found for this signature in database
GPG Key ID: 92ADA4E935E71FA4
5 changed files with 70 additions and 48 deletions

1
Cargo.lock generated
View File

@ -2036,7 +2036,6 @@ dependencies = [
"obkv", "obkv",
"once_cell", "once_cell",
"parking_lot", "parking_lot",
"paste",
"pin-project-lite", "pin-project-lite",
"platform-dirs", "platform-dirs",
"rand", "rand",

View File

@ -83,7 +83,6 @@ actix-rt = "2.7.0"
assert-json-diff = "2.0.1" assert-json-diff = "2.0.1"
manifest-dir-macros = "0.1.14" manifest-dir-macros = "0.1.14"
maplit = "1.0.2" maplit = "1.0.2"
paste = "1.0.6"
serde_url_params = "0.2.1" serde_url_params = "0.2.1"
urlencoding = "2.1.0" urlencoding = "2.1.0"

View File

@ -18,7 +18,7 @@ pub static AUTHORIZATIONS: Lazy<HashMap<(&'static str, &'static str), HashSet<&'
("GET", "/tasks") => hashset!{"tasks.get", "*"}, ("GET", "/tasks") => hashset!{"tasks.get", "*"},
("GET", "/tasks?indexUid=products") => hashset!{"tasks.get", "*"}, ("GET", "/tasks?indexUid=products") => hashset!{"tasks.get", "*"},
("GET", "/tasks/0") => hashset!{"tasks.get", "*"}, ("GET", "/tasks/0") => hashset!{"tasks.get", "*"},
("PUT", "/indexes/products/") => hashset!{"indexes.update", "*"}, ("PATCH", "/indexes/products/") => hashset!{"indexes.update", "*"},
("GET", "/indexes/products/") => hashset!{"indexes.get", "*"}, ("GET", "/indexes/products/") => hashset!{"indexes.get", "*"},
("DELETE", "/indexes/products/") => hashset!{"indexes.delete", "*"}, ("DELETE", "/indexes/products/") => hashset!{"indexes.delete", "*"},
("POST", "/indexes") => hashset!{"indexes.create", "*"}, ("POST", "/indexes") => hashset!{"indexes.create", "*"},
@ -33,15 +33,15 @@ pub static AUTHORIZATIONS: Lazy<HashMap<(&'static str, &'static str), HashSet<&'
("GET", "/indexes/products/settings/stop-words") => hashset!{"settings.get", "*"}, ("GET", "/indexes/products/settings/stop-words") => hashset!{"settings.get", "*"},
("GET", "/indexes/products/settings/synonyms") => hashset!{"settings.get", "*"}, ("GET", "/indexes/products/settings/synonyms") => hashset!{"settings.get", "*"},
("DELETE", "/indexes/products/settings") => hashset!{"settings.update", "*"}, ("DELETE", "/indexes/products/settings") => hashset!{"settings.update", "*"},
("POST", "/indexes/products/settings") => hashset!{"settings.update", "*"}, ("PATCH", "/indexes/products/settings") => hashset!{"settings.update", "*"},
("POST", "/indexes/products/settings/displayed-attributes") => hashset!{"settings.update", "*"}, ("PUT", "/indexes/products/settings/displayed-attributes") => hashset!{"settings.update", "*"},
("POST", "/indexes/products/settings/distinct-attribute") => hashset!{"settings.update", "*"}, ("PUT", "/indexes/products/settings/distinct-attribute") => hashset!{"settings.update", "*"},
("POST", "/indexes/products/settings/filterable-attributes") => hashset!{"settings.update", "*"}, ("PUT", "/indexes/products/settings/filterable-attributes") => hashset!{"settings.update", "*"},
("POST", "/indexes/products/settings/ranking-rules") => hashset!{"settings.update", "*"}, ("PUT", "/indexes/products/settings/ranking-rules") => hashset!{"settings.update", "*"},
("POST", "/indexes/products/settings/searchable-attributes") => hashset!{"settings.update", "*"}, ("PUT", "/indexes/products/settings/searchable-attributes") => hashset!{"settings.update", "*"},
("POST", "/indexes/products/settings/sortable-attributes") => hashset!{"settings.update", "*"}, ("PUT", "/indexes/products/settings/sortable-attributes") => hashset!{"settings.update", "*"},
("POST", "/indexes/products/settings/stop-words") => hashset!{"settings.update", "*"}, ("PUT", "/indexes/products/settings/stop-words") => hashset!{"settings.update", "*"},
("POST", "/indexes/products/settings/synonyms") => hashset!{"settings.update", "*"}, ("PUT", "/indexes/products/settings/synonyms") => hashset!{"settings.update", "*"},
("GET", "/indexes/products/stats") => hashset!{"stats.get", "*"}, ("GET", "/indexes/products/stats") => hashset!{"stats.get", "*"},
("GET", "/stats") => hashset!{"stats.get", "*"}, ("GET", "/stats") => hashset!{"stats.get", "*"},
("POST", "/dumps") => hashset!{"dumps.create", "*"}, ("POST", "/dumps") => hashset!{"dumps.create", "*"},

View File

@ -4,29 +4,12 @@ use std::{
}; };
use actix_web::http::StatusCode; use actix_web::http::StatusCode;
use paste::paste;
use serde_json::{json, Value}; use serde_json::{json, Value};
use tokio::time::sleep; use tokio::time::sleep;
use urlencoding::encode; use urlencoding::encode;
use super::service::Service; use super::service::Service;
macro_rules! make_settings_test_routes {
($($name:ident),+) => {
$(paste! {
pub async fn [<update_$name>](&self, value: Value) -> (Value, StatusCode) {
let url = format!("/indexes/{}/settings/{}", encode(self.uid.as_ref()).to_string(), stringify!($name).replace("_", "-"));
self.service.post(url, value).await
}
pub async fn [<get_$name>](&self) -> (Value, StatusCode) {
let url = format!("/indexes/{}/settings/{}", encode(self.uid.as_ref()).to_string(), stringify!($name).replace("_", "-"));
self.service.get(url).await
}
})*
};
}
pub struct Index<'a> { pub struct Index<'a> {
pub uid: String, pub uid: String,
pub service: &'a Service, pub service: &'a Service,
@ -198,7 +181,7 @@ impl Index<'_> {
pub async fn update_settings(&self, settings: Value) -> (Value, StatusCode) { pub async fn update_settings(&self, settings: Value) -> (Value, StatusCode) {
let url = format!("/indexes/{}/settings", encode(self.uid.as_ref())); let url = format!("/indexes/{}/settings", encode(self.uid.as_ref()));
self.service.post(url, settings).await self.service.patch(url, settings).await
} }
pub async fn delete_settings(&self) -> (Value, StatusCode) { pub async fn delete_settings(&self) -> (Value, StatusCode) {
@ -242,7 +225,23 @@ impl Index<'_> {
self.service.get(url).await self.service.get(url).await
} }
make_settings_test_routes!(distinct_attribute); pub async fn update_distinct_attribute(&self, value: Value) -> (Value, StatusCode) {
let url = format!(
"/indexes/{}/settings/{}",
encode(self.uid.as_ref()).to_string(),
"distinct-attribute"
);
self.service.put(url, value).await
}
pub async fn get_distinct_attribute(&self) -> (Value, StatusCode) {
let url = format!(
"/indexes/{}/settings/{}",
encode(self.uid.as_ref()).to_string(),
"distinct-attribute"
);
self.service.get(url).await
}
} }
pub struct GetDocumentOptions { pub struct GetDocumentOptions {

View File

@ -7,23 +7,45 @@ use actix_web::test;
use meilisearch_http::{analytics, create_app}; use meilisearch_http::{analytics, create_app};
use serde_json::{json, Value}; use serde_json::{json, Value};
enum HttpVerb {
Put,
Patch,
Post,
Get,
Delete,
}
impl HttpVerb {
fn test_request(&self) -> test::TestRequest {
match self {
HttpVerb::Put => test::TestRequest::put(),
HttpVerb::Patch => test::TestRequest::patch(),
HttpVerb::Post => test::TestRequest::post(),
HttpVerb::Get => test::TestRequest::get(),
HttpVerb::Delete => test::TestRequest::delete(),
}
}
}
#[actix_rt::test] #[actix_rt::test]
async fn error_json_bad_content_type() { async fn error_json_bad_content_type() {
use HttpVerb::{Patch, Post, Put};
let routes = [ let routes = [
// all the POST routes except the dumps that can be created without any body or content-type // all the routes except the dumps that can be created without any body or content-type
// and the search that is not a strict json // and the search that is not a strict json
"/indexes", (Post, "/indexes"),
"/indexes/doggo/documents/delete-batch", (Post, "/indexes/doggo/documents/delete-batch"),
"/indexes/doggo/search", (Post, "/indexes/doggo/search"),
"/indexes/doggo/settings", (Patch, "/indexes/doggo/settings"),
"/indexes/doggo/settings/displayed-attributes", (Put, "/indexes/doggo/settings/displayed-attributes"),
"/indexes/doggo/settings/distinct-attribute", (Put, "/indexes/doggo/settings/distinct-attribute"),
"/indexes/doggo/settings/filterable-attributes", (Put, "/indexes/doggo/settings/filterable-attributes"),
"/indexes/doggo/settings/ranking-rules", (Put, "/indexes/doggo/settings/ranking-rules"),
"/indexes/doggo/settings/searchable-attributes", (Put, "/indexes/doggo/settings/searchable-attributes"),
"/indexes/doggo/settings/sortable-attributes", (Put, "/indexes/doggo/settings/sortable-attributes"),
"/indexes/doggo/settings/stop-words", (Put, "/indexes/doggo/settings/stop-words"),
"/indexes/doggo/settings/synonyms", (Put, "/indexes/doggo/settings/synonyms"),
]; ];
let bad_content_types = [ let bad_content_types = [
"application/csv", "application/csv",
@ -45,10 +67,11 @@ async fn error_json_bad_content_type() {
analytics::MockAnalytics::new(&server.service.options).0 analytics::MockAnalytics::new(&server.service.options).0
)) ))
.await; .await;
for route in routes { for (verb, route) in routes {
// Good content-type, we probably have an error since we didn't send anything in the json // Good content-type, we probably have an error since we didn't send anything in the json
// so we only ensure we didn't get a bad media type error. // so we only ensure we didn't get a bad media type error.
let req = test::TestRequest::post() let req = verb
.test_request()
.uri(route) .uri(route)
.set_payload(document) .set_payload(document)
.insert_header(("content-type", "application/json")) .insert_header(("content-type", "application/json"))
@ -59,7 +82,8 @@ async fn error_json_bad_content_type() {
"calling the route `{}` with a content-type of json isn't supposed to throw a bad media type error", route); "calling the route `{}` with a content-type of json isn't supposed to throw a bad media type error", route);
// No content-type. // No content-type.
let req = test::TestRequest::post() let req = verb
.test_request()
.uri(route) .uri(route)
.set_payload(document) .set_payload(document)
.to_request(); .to_request();
@ -82,7 +106,8 @@ async fn error_json_bad_content_type() {
for bad_content_type in bad_content_types { for bad_content_type in bad_content_types {
// Always bad content-type // Always bad content-type
let req = test::TestRequest::post() let req = verb
.test_request()
.uri(route) .uri(route)
.set_payload(document.to_string()) .set_payload(document.to_string())
.insert_header(("content-type", bad_content_type)) .insert_header(("content-type", bad_content_type))