From e26814e2e7d1eb4d41035dafb42f9e1dc09e20fd Mon Sep 17 00:00:00 2001
From: mykola2312 <49044616+mykola2312@users.noreply.github.com>
Date: Wed, 21 Feb 2024 15:05:18 +0200
Subject: [PATCH] add tests to check best format finding functions

---
 .gitignore       |  2 +-
 src/bot.rs       |  2 +-
 src/dl/yt_dlp.rs | 26 ++++++++++++++++++++++++++
 src/main.rs      |  2 +-
 4 files changed, 29 insertions(+), 3 deletions(-)

diff --git a/.gitignore b/.gitignore
index 4dc466c..95963b3 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,5 +1,5 @@
 /target
-.env
+.env*
 *.json
 *.txt
 *.m4a
diff --git a/src/bot.rs b/src/bot.rs
index 96f1e66..1cbfd96 100644
--- a/src/bot.rs
+++ b/src/bot.rs
@@ -1 +1 @@
-pub mod bot;
\ No newline at end of file
+pub mod bot;
diff --git a/src/dl/yt_dlp.rs b/src/dl/yt_dlp.rs
index d6c8ba3..c075d0a 100644
--- a/src/dl/yt_dlp.rs
+++ b/src/dl/yt_dlp.rs
@@ -170,3 +170,29 @@ impl YtDlp {
         Ok(YtDlpInfo::parse(&output.stdout)?)
     }
 }
+
+#[cfg(test)]
+mod tests {
+    use super::YtDlp;
+    use std::env;
+
+    #[tokio::test]
+    async fn best_av_format() {
+        dotenv::from_filename(".env.test").unwrap();
+        let info = YtDlp::load_info(env::var("TEST_URL").unwrap().as_str())
+            .await
+            .unwrap();
+        let video = info.best_av_format().unwrap();
+        assert_eq!(video.format_id, "22");
+    }
+
+    #[tokio::test]
+    async fn best_audio_format() {
+        dotenv::from_filename(".env.test").unwrap();
+        let info = YtDlp::load_info(env::var("TEST_URL").unwrap().as_str())
+            .await
+            .unwrap();
+        let video = info.best_audio_format().unwrap();
+        assert_eq!(video.format_id, "140");
+    }
+}
diff --git a/src/main.rs b/src/main.rs
index e0f64bd..264b208 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -1,5 +1,5 @@
-use std::env;
 use dotenv::dotenv;
+use std::env;
 
 mod bot;
 use bot::bot::bot_main;