Test openai default tts

Author: rikhuijzer

README.md

@@ -2,15 +2,9 @@
 
 <h3 align="center">Use AI in the terminal</h3>
 
-## Philosophy
-
-The philosophy of this project is mainly to not handle state.
-Like curl or ffmpeg, this should make it easier to use in scripts and to share examples online.
-Settings are done via command line arguments and environment variables.
-
 ## Examples
 
-### Text to Speech and Bash
+### Text to Speech in Bash
 
 We can read a file out loud from the command line.
 For example, with the OpenAI API:
@@ -42,7 +36,7 @@ After adding `spk.sh` to your PATH, you can use it like this:
 $ cat myfile.txt | spk
 ```
 
-### Other Text to Speech
+### Other Text to Speech Commands
 
 ```sh
 $ DEEPINFRA_KEY="$(cat /path/to/key)"; cat myfile.txt | ata tts | vlc -
@@ -52,3 +46,8 @@ $ DEEPINFRA_KEY="$(cat /path/to/key)"; cat myfile.txt | ata tts | vlc -
 $ DEEPINFRA_KEY="$(cat /path/to/key)"; cat myfile.txt | ata tts --output myfile.mp3
 ```
 
+## Philosophy
+
+The philosophy of this project is mainly to not handle state.
+Like curl or ffmpeg, this should make it easier to use in scripts and to share examples online.
+Settings are done via command line arguments and environment variables.

tests/cli.rs

@@ -50,15 +50,32 @@ fn load_key(provider: &Provider) -> String {
 }
 
 #[test]
-fn tts_no_args_output() -> Result<(), Box<dyn std::error::Error>> {
+fn tts_no_args() -> Result<(), Box<dyn std::error::Error>> {
     let dir = tempfile::tempdir().unwrap();
     let mut cmd = ata();
     let key = load_key(&Provider::DeepInfra);
+    let cmd = cmd
+        .arg("tts")
+        .env("DEEPINFRA_KEY", key)
+        .write_stdin("Hello world")
+        .current_dir(&dir);
+    let output = cmd.assert().success().get_output().stdout.clone();
+
+    assert!(output.len() > 0);
+
+    Ok(())
+}
+
+fn tts_default_settings_helper(provider: &Provider) -> Result<(), Box<dyn std::error::Error>> {
+    let dir = tempfile::tempdir().unwrap();
+    let mut cmd = ata();
+    let key = load_key(provider);
+    let name = provider.key_name();
     cmd.arg("tts")
         .arg("--output")
         .arg("output.mp3")
-        .env("DEEPINFRA_KEY", key)
-        .write_stdin("Hello world")
+        .env(name, key)
+        .write_stdin("Hi")
         .current_dir(&dir)
         .assert()
         .success();
@@ -70,18 +87,11 @@ fn tts_no_args_output() -> Result<(), Box<dyn std::error::Error>> {
 }
 
 #[test]
-fn tts_no_args() -> Result<(), Box<dyn std::error::Error>> {
-    let dir = tempfile::tempdir().unwrap();
-    let mut cmd = ata();
-    let key = load_key(&Provider::DeepInfra);
-    let cmd = cmd
-        .arg("tts")
-        .env("DEEPINFRA_KEY", key)
-        .write_stdin("Hello world")
-        .current_dir(&dir);
-    let output = cmd.assert().success().get_output().stdout.clone();
-
-    assert!(output.len() > 0);
+fn tts_no_args_deepinfra() -> Result<(), Box<dyn std::error::Error>> {
+    tts_default_settings_helper(&Provider::DeepInfra)
+}
 
-    Ok(())
+#[test]
+fn tts_no_args_openai() -> Result<(), Box<dyn std::error::Error>> {
+    tts_default_settings_helper(&Provider::OpenAI)
 }