diff --git a/tests/common/client.rs b/tests/common/client.rs index c9c22d2e..1351493a 100644 --- a/tests/common/client.rs +++ b/tests/common/client.rs @@ -177,6 +177,12 @@ impl Client { pub async fn ban_user(&self, username: Username) -> TextResponse { self.http_client.delete(&format!("/user/ban/{}", &username.value)).await } + + // Context: proxy + + pub async fn get_image_by_url(&self, url: &str) -> TextResponse { + self.http_client.get(&format!("/proxy/image/{url}"), Query::empty()).await + } } /// Generic HTTP Client diff --git a/tests/e2e/web/api/v1/contexts/proxy/contract.rs b/tests/e2e/web/api/v1/contexts/proxy/contract.rs index 0b63dfc4..b2de1433 100644 --- a/tests/e2e/web/api/v1/contexts/proxy/contract.rs +++ b/tests/e2e/web/api/v1/contexts/proxy/contract.rs @@ -1,3 +1,103 @@ //! API contract for `proxy` context. -// todo +mod authorization { + use torrust_index::web::api; + + use crate::common::client::Client; + use crate::common::contexts::torrent::fixtures::random_torrent; + use crate::e2e::environment::TestEnv; + use crate::e2e::web::api::v1::contexts::torrent::steps::upload_torrent; + use crate::e2e::web::api::v1::contexts::user::steps::{new_logged_in_admin, new_logged_in_user}; + + #[tokio::test] + async fn it_should_return_an_ok_response_when_a_guest_user_tries_to_get_a_proxied_image_by_url() { + let mut env = TestEnv::new(); + + env.start(api::Version::V1).await; + + if !env.provides_a_tracker() { + println!("test skipped. It requires a tracker to be running."); + return; + } + + let mut test_torrent = random_torrent(); + + test_torrent.index_info.description = String::from( + "![image info](https://upload.wikimedia.org/wikipedia/commons/thumb/7/71/Zaadpluizen_van_een_Clematis_texensis_%27Princess_Diana%27._18-07-2023_%28actm.%29_02.jpg/1024px-Zaadpluizen_van_een_Clematis_texensis_%27Princess_Diana%27._18-07-2023_%28actm.%29_02.jpg)", + ); + + let torrent_uploader = new_logged_in_user(&env).await; + + upload_torrent(&torrent_uploader, &test_torrent.index_info, &env).await; + + let client = Client::unauthenticated(&env.server_socket_addr().unwrap()); + + let url = "https%3A%2F%2Fupload.wikimedia.org%2Fwikipedia%2Fcommons%2Fthumb%2F7%2F71%2FZaadpluizen_van_een_Clematis_texensis_%2527Princess_Diana%2527._18-07-2023_%2528actm.%2529_02.jpg%2F1024px-Zaadpluizen_van_een_Clematis_texensis_%2527Princess_Diana%2527._18-07-2023_%2528actm.%2529_02.jpg"; + + let response = client.get_image_by_url(url).await; + + assert_eq!(response.status, 200); + } + #[tokio::test] + async fn it_should_allow_registered_users_to_get_a_proxied_image_by_url() { + let mut env = TestEnv::new(); + + env.start(api::Version::V1).await; + + if !env.provides_a_tracker() { + println!("test skipped. It requires a tracker to be running."); + return; + } + + let mut test_torrent = random_torrent(); + + test_torrent.index_info.description = String::from( + "![image info](https://upload.wikimedia.org/wikipedia/commons/thumb/7/71/Zaadpluizen_van_een_Clematis_texensis_%27Princess_Diana%27._18-07-2023_%28actm.%29_02.jpg/1024px-Zaadpluizen_van_een_Clematis_texensis_%27Princess_Diana%27._18-07-2023_%28actm.%29_02.jpg)", + ); + + let torrent_uploader = new_logged_in_user(&env).await; + + upload_torrent(&torrent_uploader, &test_torrent.index_info, &env).await; + + let logged_non_admin = new_logged_in_user(&env).await; + + let client = Client::authenticated(&env.server_socket_addr().unwrap(), &logged_non_admin.token); + + let url = "https%3A%2F%2Fupload.wikimedia.org%2Fwikipedia%2Fcommons%2Fthumb%2F7%2F71%2FZaadpluizen_van_een_Clematis_texensis_%2527Princess_Diana%2527._18-07-2023_%2528actm.%2529_02.jpg%2F1024px-Zaadpluizen_van_een_Clematis_texensis_%2527Princess_Diana%2527._18-07-2023_%2528actm.%2529_02.jpg"; + + let response = client.get_image_by_url(url).await; + + assert_eq!(response.status, 200); + } + #[tokio::test] + async fn it_should_allow_admin_users_to_get_a_proxied_image_by_url() { + let mut env = TestEnv::new(); + + env.start(api::Version::V1).await; + + if !env.provides_a_tracker() { + println!("test skipped. It requires a tracker to be running."); + return; + } + + let mut test_torrent = random_torrent(); + + test_torrent.index_info.description = String::from( + "![image info](https://upload.wikimedia.org/wikipedia/commons/thumb/7/71/Zaadpluizen_van_een_Clematis_texensis_%27Princess_Diana%27._18-07-2023_%28actm.%29_02.jpg/1024px-Zaadpluizen_van_een_Clematis_texensis_%27Princess_Diana%27._18-07-2023_%28actm.%29_02.jpg)", + ); + + let torrent_uploader = new_logged_in_user(&env).await; + + upload_torrent(&torrent_uploader, &test_torrent.index_info, &env).await; + + let logged_in_admin = new_logged_in_admin(&env).await; + + let client = Client::authenticated(&env.server_socket_addr().unwrap(), &logged_in_admin.token); + + let url = "https%3A%2F%2Fupload.wikimedia.org%2Fwikipedia%2Fcommons%2Fthumb%2F7%2F71%2FZaadpluizen_van_een_Clematis_texensis_%2527Princess_Diana%2527._18-07-2023_%2528actm.%2529_02.jpg%2F1024px-Zaadpluizen_van_een_Clematis_texensis_%2527Princess_Diana%2527._18-07-2023_%2528actm.%2529_02.jpg"; + + let response = client.get_image_by_url(url).await; + + assert_eq!(response.status, 200); + } +}