thoughts/thoughts-backend/tests/api/user.rs

use axum::http::StatusCode;
use http_body_util::BodyExt;
use models::domains::top_friends;
use sea_orm::{ColumnTrait, EntityTrait, QueryFilter};
use serde_json::{json, Value};

use utils::testing::{make_get_request, make_jwt_request, make_post_request};

use crate::api::main::{create_user_with_password, login_user, setup};

#[tokio::test]
async fn test_post_users() {
    let app = setup().await;

    let body = r#"{"username": "test", "password": "password123"}"#.to_owned();
    let response = make_post_request(app.router, "/auth/register", body, None).await;

    assert_eq!(response.status(), StatusCode::CREATED);

    let body = response.into_body().collect().await.unwrap().to_bytes();
    let v: Value = serde_json::from_slice(&body).unwrap();

    assert_eq!(v["username"], "test");
    assert!(v["display_name"].is_null());
}

#[tokio::test]
pub(super) async fn test_post_users_error() {
    let app = setup().await;

    let body = r#"{"username": "1", "password": "password123"}"#.to_owned();
    let response = make_post_request(app.router, "/auth/register", body, None).await;

    assert_eq!(response.status(), StatusCode::UNPROCESSABLE_ENTITY);

    let body = response.into_body().collect().await.unwrap().to_bytes();
    let result: Value = serde_json::from_slice(&body).unwrap();
    assert_eq!(result["message"], "Validation error");
    assert_eq!(result["details"]["username"][0]["code"], "length");
}

#[tokio::test]
pub async fn test_get_users() {
    let app = setup().await;

    let body = r#"{"username": "test", "password": "password123"}"#.to_owned();
    make_post_request(app.router.clone(), "/auth/register", body, None).await;

    let response = make_get_request(app.router, "/users", None).await;
    assert_eq!(response.status(), StatusCode::OK);

    let body = response.into_body().collect().await.unwrap().to_bytes();
    let v: Value = serde_json::from_slice(&body).unwrap();

    assert!(v["users"].is_array());
    let users_array = v["users"].as_array().unwrap();
    assert_eq!(users_array.len(), 1);
    assert_eq!(users_array[0]["username"], "test");
}

#[tokio::test]
async fn test_me_endpoints() {
    let app = setup().await;

    // 1. Register a new user
    let register_body = json!({
        "username": "me_user",
        "password": "password123"
    })
    .to_string();
    let response =
        make_post_request(app.router.clone(), "/auth/register", register_body, None).await;
    assert_eq!(response.status(), StatusCode::CREATED);

    // 2. Log in to get a token
    let token = login_user(app.router.clone(), "me_user", "password123").await;

    // 3. GET /users/me to fetch initial profile
    let response = make_jwt_request(app.router.clone(), "/users/me", "GET", None, &token).await;
    assert_eq!(response.status(), StatusCode::OK);
    let body = response.into_body().collect().await.unwrap().to_bytes();
    let v: Value = serde_json::from_slice(&body).unwrap();
    assert_eq!(v["username"], "me_user");
    assert!(v["bio"].is_null());
    assert!(v["display_name"].is_null());

    // 4. PUT /users/me to update the profile
    let update_body = json!({
        "display_name": "Me User",
        "bio": "This is my updated bio.",
        "avatar_url": "https://example.com/avatar.png"
    })
    .to_string();
    let response = make_jwt_request(
        app.router.clone(),
        "/users/me",
        "PUT",
        Some(update_body),
        &token,
    )
    .await;
    assert_eq!(response.status(), StatusCode::OK);
    let body = response.into_body().collect().await.unwrap().to_bytes();
    let v_updated: Value = serde_json::from_slice(&body).unwrap();
    assert_eq!(v_updated["display_name"], "Me User");
    assert_eq!(v_updated["bio"], "This is my updated bio.");

    // 5. GET /users/me again to verify the update was persisted
    let response = make_jwt_request(app.router.clone(), "/users/me", "GET", None, &token).await;
    assert_eq!(response.status(), StatusCode::OK);
    let body = response.into_body().collect().await.unwrap().to_bytes();
    let v_verify: Value = serde_json::from_slice(&body).unwrap();
    assert_eq!(v_verify["display_name"], "Me User");
    assert_eq!(v_verify["bio"], "This is my updated bio.");
}

#[tokio::test]
async fn test_update_me_top_friends() {
    let app = setup().await;

    // 1. Create users for the test
    let user_me = create_user_with_password(&app.db, "me_user", "password123").await;
    let friend1 = create_user_with_password(&app.db, "friend1", "password123").await;
    let friend2 = create_user_with_password(&app.db, "friend2", "password123").await;
    let _friend3 = create_user_with_password(&app.db, "friend3", "password123").await;

    // 2. Log in as "me_user"
    let token = login_user(app.router.clone(), "me_user", "password123").await;

    // 3. Update profile to set top friends
    let update_body = json!({
        "top_friends": ["friend1", "friend2"]
    })
    .to_string();

    let response = make_jwt_request(
        app.router.clone(),
        "/users/me",
        "PUT",
        Some(update_body),
        &token,
    )
    .await;
    assert_eq!(response.status(), StatusCode::OK);

    // 4. Verify the database state directly
    let top_friends_list = top_friends::Entity::find()
        .filter(top_friends::Column::UserId.eq(user_me.id))
        .all(&app.db)
        .await
        .unwrap();

    assert_eq!(top_friends_list.len(), 2);
    assert_eq!(top_friends_list[0].friend_id, friend1.id);
    assert_eq!(top_friends_list[0].position, 1);
    assert_eq!(top_friends_list[1].friend_id, friend2.id);
    assert_eq!(top_friends_list[1].position, 2);

    // 5. Update again with a different list to test replacement
    let update_body_2 = json!({
        "top_friends": ["friend2"]
    })
    .to_string();

    let response = make_jwt_request(
        app.router.clone(),
        "/users/me",
        "PUT",
        Some(update_body_2),
        &token,
    )
    .await;
    assert_eq!(response.status(), StatusCode::OK);

    // 6. Verify the new state
    let top_friends_list_2 = top_friends::Entity::find()
        .filter(top_friends::Column::UserId.eq(user_me.id))
        .all(&app.db)
        .await
        .unwrap();

    assert_eq!(top_friends_list_2.len(), 1);
    assert_eq!(top_friends_list_2[0].friend_id, friend2.id);
    assert_eq!(top_friends_list_2[0].position, 1);
}