use derive_builder::Builder;
use mockito::{Mock, Server};
use serde::{Deserialize, Serialize};
use serde_json::json;
#[derive(Serialize, Deserialize, Debug, Clone, Builder, Default)]
#[builder(setter(into))]
#[builder(default)]
#[builder(field(public))]
pub struct MockParams {
pub access_token: String,
pub status_code: usize,
pub sub: String,
pub name: Option<String>,
pub picture: Option<String>,
pub email: Option<String>,
pub error_message: Option<String>,
}
pub async fn make_mock(server: &mut Server, builder: Option<MockParamsBuilder>) -> Mock {
let mut builder = builder.unwrap_or_default();
if builder.access_token.is_none() {
builder.access_token("test_access_token".to_string());
}
if builder.status_code.is_none() {
builder.status_code(200usize);
}
if builder.sub.is_none() {
builder.sub("U123456".to_string());
}
if builder.error_message.is_none() {
builder.error_message("error occurred".to_string());
}
let params = builder.build().unwrap();
let body_json = if params.status_code == 200 {
let mut json = json!({
"sub": params.sub,
});
if params.name.is_some() {
json["name"] = params.name.clone().into();
}
if params.picture.is_some() {
json["picture"] = params.picture.clone().into();
}
if params.email.is_some() {
json["email"] = params.email.clone().into();
}
json
} else {
json!({
"message": params.error_message
})
};
server
.mock("GET", "/oauth2/v2.1/userinfo")
.match_header(
"authorization",
format!("Bearer {}", params.access_token).as_str(),
)
.with_status(params.status_code)
.with_header("content-type", "application/json")
.with_body(body_json.to_string())
.create_async()
.await
}
#[cfg(test)]
mod tests {
use crate::{LineOptions, error::Error, line_login::get_oauth2_v2_1_userinfo};
use super::*;
#[tokio::test]
async fn test_make_mock_get_oauth2_v2_1_userinfo_success() {
let mut server = Server::new_async().await;
let mut builder = MockParamsBuilder::default();
builder.name(Some("Test User".into()));
builder.picture(Some("https://example.com/picture.jpg".into()));
builder.email(Some("test@example.com".into()));
let mock = make_mock(&mut server, Some(builder)).await;
let res = get_oauth2_v2_1_userinfo::execute_get(
"test_access_token",
&LineOptions {
prefix_url: Some(server.url()),
..Default::default()
},
)
.await
.unwrap();
assert_eq!(res.0.sub, "U123456");
assert_eq!(res.0.name, Some("Test User".into()));
assert_eq!(
res.0.picture,
Some("https://example.com/picture.jpg".into())
);
assert_eq!(
res.0.extra.get("email"),
Some(&serde_json::json!("test@example.com"))
);
mock.assert_async().await;
}
#[tokio::test]
async fn test_make_mock_get_oauth2_v2_1_userinfo_failure() {
let mut server = Server::new_async().await;
let mut builder = MockParamsBuilder::default();
builder.status_code(400usize);
let mock = make_mock(&mut server, Some(builder)).await;
let res = get_oauth2_v2_1_userinfo::execute_get(
"test_access_token",
&LineOptions {
prefix_url: Some(server.url()),
..Default::default()
},
)
.await;
match res {
Err(e) => match *e {
Error::Line(response, status_code, _header) => {
assert_eq!(status_code, 400);
assert_eq!(response.message, "error occurred");
}
_ => panic!("Unexpected error"),
},
_ => panic!("Unexpected response"),
}
mock.assert_async().await;
}
}