From 61b006bbcff61984b319413b7fb3522cfc3c5ad9 Mon Sep 17 00:00:00 2001
From: phoenix <kundeng00@pm.me>
Date: Wed, 23 Apr 2025 20:27:03 -0400
Subject: [PATCH] Added test code

---
 src/callers/mod.rs |   1 +
 src/main.rs        | 101 +++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 102 insertions(+)

diff --git a/src/callers/mod.rs b/src/callers/mod.rs
index edccd1c..5abbb9f 100644
--- a/src/callers/mod.rs
+++ b/src/callers/mod.rs
@@ -3,4 +3,5 @@ pub mod song;
 
 pub mod endpoints {
     pub const QUEUESONG: &str = "/api/v2/song/queue";
+    pub const QUEUEMETADATA: &str = "/api/v2/song/metadata/queue";
 }
diff --git a/src/main.rs b/src/main.rs
index 27dd3a4..aafcc52 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -69,6 +69,10 @@ pub mod init {
                 crate::callers::endpoints::QUEUESONG,
                 post(crate::callers::song::endpoint::queue_song),
             )
+            .route(
+                crate::callers::endpoints::QUEUEMETADATA,
+                post(crate::callers::metadata::endpoint::queue_metadata),
+            )
     }
 
     pub async fn app() -> axum::Router {
@@ -245,4 +249,101 @@ mod tests {
 
         let _ = db_mgr::drop_database(&tm_pool, &db_name).await;
     }
+
+    #[tokio::test]
+    async fn test_song_metadata_queue() {
+        let tm_pool = db_mgr::get_pool().await.unwrap();
+        let db_name = db_mgr::generate_db_name().await;
+
+        match db_mgr::create_database(&tm_pool, &db_name).await {
+            Ok(_) => {
+                println!("Success");
+            }
+            Err(err) => {
+                assert!(false, "Error: {:?}", err);
+            }
+        }
+
+        let pool = db_mgr::connect_to_db(&db_name).await.unwrap();
+        db::migrations(&pool).await;
+
+        let app = crate::init::routes()
+            .await
+            .layer(axum::Extension(pool))
+            .layer(axum::extract::DefaultBodyLimit::max(1024 * 1024 * 1024))
+            .layer(TimeoutLayer::new(Duration::from_secs(300)));
+
+        // Create multipart form
+        let mut form = MultipartForm::default();
+        let _ = form.add_file("flac", "tests/Machine_gun/track01.flac");
+
+        // Create request
+        let content_type = form.content_type();
+        let body = MultipartBody::from(form);
+        let req = axum::http::Request::builder()
+            .method(axum::http::Method::POST)
+            .uri(crate::callers::endpoints::QUEUESONG)
+            .header(axum::http::header::CONTENT_TYPE, content_type)
+            .body(axum::body::Body::from_stream(body))
+            .unwrap();
+
+        // Send request
+        match app.clone().oneshot(req).await {
+            Ok(response) => {
+                let body = axum::body::to_bytes(response.into_body(), usize::MAX)
+                    .await
+                    .unwrap();
+                println!("Body: {:?}", body);
+                let resp: crate::callers::song::response::Response =
+                    serde_json::from_slice(&body).unwrap();
+                assert_eq!(false, resp.data.is_empty(), "Should not be empty");
+                assert_eq!(false, resp.data[0].is_nil(), "Should not be empty");
+                let new_payload: serde_json::Value = serde_json::json!(
+                {
+                        "id": resp.data[0],
+                        "album" : "Machine Gun: The FillMore East First Show",
+                        "album_artist" : "Jimi Hendrix",
+                        "artist" : "Jimi Hendrix",
+                        "disc" : 1,
+                        "disc_count" : 1,
+                        "duration" : 330,
+                        "genre" : "Psychadelic Rock",
+                        "title" : "Power of Soul",
+                        "track" : 1,
+                        "track_count" : 11,
+                        "year" : 2016
+                });
+
+                match app
+                    .clone()
+                    .oneshot(
+                        axum::http::Request::builder()
+                            .method(axum::http::Method::POST)
+                            .uri(crate::callers::endpoints::QUEUEMETADATA)
+                            .header(axum::http::header::CONTENT_TYPE, "application/json")
+                            .body(axum::body::Body::from(new_payload.to_string()))
+                            .unwrap(),
+                    )
+                    .await
+                {
+                    Ok(response) => {
+                        let body = axum::body::to_bytes(response.into_body(), usize::MAX)
+                            .await
+                            .unwrap();
+                        let resp: crate::callers::song::response::Response =
+                            serde_json::from_slice(&body).unwrap();
+                        assert_eq!(false, resp.data.is_empty(), "Should not be empty");
+                    }
+                    Err(err) => {
+                        assert!(false, "Error: {:?}", err);
+                    }
+                }
+            }
+            Err(err) => {
+                assert!(false, "Error: {:?}", err);
+            }
+        };
+
+        let _ = db_mgr::drop_database(&tm_pool, &db_name).await;
+    }
 }