feat: Show AWS region in aws module (#482)

tests/testsuite/aws.rs

@@ -1,10 +1,12 @@
+use std::fs::File;
+use std::io::{self, Write};
+
 use ansi_term::Color;
-use std::io;
 
 use crate::common;
 
 #[test]
-fn no_profile_set() -> io::Result<()> {
+fn no_region_set() -> io::Result<()> {
     let output = common::render_module("aws").env_clear().output()?;
     let expected = "";
     let actual = String::from_utf8(output.stdout).unwrap();
@@ -12,6 +14,31 @@ fn no_profile_set() -> io::Result<()> {
     Ok(())
 }
 
+#[test]
+fn region_set() -> io::Result<()> {
+    let output = common::render_module("aws")
+        .env_clear()
+        .env("AWS_REGION", "ap-northeast-2")
+        .output()?;
+    let expected = format!("on {} ", Color::Yellow.bold().paint("☁️  ap-northeast-2"));
+    let actual = String::from_utf8(output.stdout).unwrap();
+    assert_eq!(expected, actual);
+    Ok(())
+}
+
+#[test]
+fn default_region_set() -> io::Result<()> {
+    let output = common::render_module("aws")
+        .env_clear()
+        .env("AWS_REGION", "ap-northeast-2")
+        .env("AWS_DEFAULT_REGION", "ap-northeast-1")
+        .output()?;
+    let expected = format!("on {} ", Color::Yellow.bold().paint("☁️  ap-northeast-1"));
+    let actual = String::from_utf8(output.stdout).unwrap();
+    assert_eq!(expected, actual);
+    Ok(())
+}
+
 #[test]
 fn profile_set() -> io::Result<()> {
     let output = common::render_module("aws")
@@ -23,3 +50,59 @@ fn profile_set() -> io::Result<()> {
     assert_eq!(expected, actual);
     Ok(())
 }
+
+#[test]
+fn default_profile_set() -> io::Result<()> {
+    let dir = common::new_tempdir()?;
+    let config_path = dir.path().join("config");
+    let mut file = File::create(&config_path)?;
+
+    file.write_all(
+        "[default]
+region = us-east-1
+
+[profile astronauts]
+region = us-east-2
+"
+        .as_bytes(),
+    )?;
+
+    let output = common::render_module("aws")
+        .env_clear()
+        .env("AWS_CONFIG_FILE", config_path.to_string_lossy().as_ref())
+        .output()?;
+    let expected = format!("on {} ", Color::Yellow.bold().paint("☁️  us-east-1"));
+    let actual = String::from_utf8(output.stdout).unwrap();
+    assert_eq!(expected, actual);
+    Ok(())
+}
+
+#[test]
+fn profile_and_config_set() -> io::Result<()> {
+    let dir = common::new_tempdir()?;
+    let config_path = dir.path().join("config");
+    let mut file = File::create(&config_path)?;
+
+    file.write_all(
+        "[default]
+region = us-east-1
+
+[profile astronauts]
+region = us-east-2
+"
+        .as_bytes(),
+    )?;
+
+    let output = common::render_module("aws")
+        .env_clear()
+        .env("AWS_CONFIG_FILE", config_path.to_string_lossy().as_ref())
+        .env("AWS_PROFILE", "astronauts")
+        .output()?;
+    let expected = format!(
+        "on {} ",
+        Color::Yellow.bold().paint("☁️  astronauts(us-east-2)")
+    );
+    let actual = String::from_utf8(output.stdout).unwrap();
+    assert_eq!(expected, actual);
+    Ok(())
+}