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;