use axum::Json;
use axum::extract::State;
use entertainarr_domain::language::Language;
use entertainarr_domain::tvshow::prelude::TvShowService;
use crate::entity::ApiResource;
use crate::entity::tvshow_subscription::TvShowSubscriptionCreateDocument;
use crate::server::extractor::user::CurrentUser;
use crate::server::handler::error::ApiErrorResponse;
pub async fn handle<S>(
State(state): State<S>,
CurrentUser(user_id): CurrentUser,
Json(payload): Json<ApiResource<TvShowSubscriptionCreateDocument>>,
) -> Result<axum::http::StatusCode, ApiErrorResponse>
where
S: crate::server::prelude::ServerState,
{
state
.tvshow_service()
.subscribe(user_id, payload.data.attributes.source.into(), Language::En)
.await
.map(|_| axum::http::StatusCode::CREATED)
.map_err(|err| {
tracing::error!(error = ?err, "unable to insert subscription");
ApiErrorResponse::internal()
})
}
#[cfg(test)]
mod tests {
use axum::Json;
use axum::extract::State;
use axum::http::StatusCode;
use entertainarr_domain::tvshow::prelude::{MockTvShowService, SubscribeError};
use crate::entity::ApiResource;
use crate::entity::tvshow::TvShowSource;
use crate::entity::tvshow_subscription::TvShowSubscriptionCreateDocument;
use crate::server::extractor::user::CurrentUser;
use crate::server::prelude::tests::MockServerState;
#[tokio::test]
async fn should_fail_if_service_fails() {
let mut tvshow_service = MockTvShowService::new();
tvshow_service
.expect_subscribe()
.return_once(|user_id, _src, _lang| {
assert_eq!(user_id, 1);
Box::pin(async move { Err(SubscribeError::NotFound) })
});
let state = MockServerState::builder().tvshow(tvshow_service).build();
let err = super::handle(
State(state),
CurrentUser(1),
Json(ApiResource::new(TvShowSubscriptionCreateDocument::new(
TvShowSource::Tmdb { tmdb_id: 1234 },
))),
)
.await
.unwrap_err();
assert_eq!(err.status_code, StatusCode::INTERNAL_SERVER_ERROR);
}
#[tokio::test]
async fn should_succeed() {
let mut tvshow_service = MockTvShowService::new();
tvshow_service
.expect_subscribe()
.return_once(|user_id, _src, _lang| {
assert_eq!(user_id, 1);
Box::pin(async move { Ok(()) })
});
let state = MockServerState::builder().tvshow(tvshow_service).build();
let res = super::handle(
State(state),
CurrentUser(1),
Json(ApiResource::new(TvShowSubscriptionCreateDocument::new(
TvShowSource::Tmdb { tmdb_id: 1234 },
))),
)
.await
.unwrap();
assert_eq!(res, StatusCode::CREATED);
}
}