From 7f713cad12682fcd5206d30868c2df8a4a554b87 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C3=89douard=20Lopez?= <contact@edouard-lopez.com>
Date: Fri, 16 Feb 2024 19:09:57 +0100
Subject: [PATCH] test: add feature tests

---
 tests/_pure_prompt.test.fish             |  1 +
 tests/_pure_prompt_aws_profile.test.fish | 56 ++++++++++++++++++++++++
 2 files changed, 57 insertions(+)
 create mode 100644 tests/_pure_prompt_aws_profile.test.fish

diff --git a/tests/_pure_prompt.test.fish b/tests/_pure_prompt.test.fish
index d10213ff..e66ae39f 100644
--- a/tests/_pure_prompt.test.fish
+++ b/tests/_pure_prompt.test.fish
@@ -4,6 +4,7 @@ source (status dirname)/../functions/_pure_prompt.fish
 source (status dirname)/../functions/_pure_prompt_jobs.fish
 source (status dirname)/../functions/_pure_prompt_nixdevshell.fish
 source (status dirname)/../functions/_pure_prompt_virtualenv.fish
+source (status dirname)/../functions/_pure_prompt_aws_profile.fish
 source (status dirname)/../functions/_pure_prompt_vimode.fish
 source (status dirname)/../functions/_pure_prompt_symbol.fish
 source (status dirname)/../functions/_pure_get_prompt_symbol.fish
diff --git a/tests/_pure_prompt_aws_profile.test.fish b/tests/_pure_prompt_aws_profile.test.fish
new file mode 100644
index 00000000..633e9db4
--- /dev/null
+++ b/tests/_pure_prompt_aws_profile.test.fish
@@ -0,0 +1,56 @@
+source (status dirname)/fixtures/constants.fish
+source (status dirname)/../functions/_pure_set_default.fish
+source (status dirname)/../functions/_pure_prompt_aws_profile.fish
+@echo (_print_filename (status filename))
+
+function before_each
+    _purge_configs
+    _disable_colors # we use mocks so cleaning them must happen before
+
+    set --erase AWS_PROFILE
+    set --erase AWS_VAULT
+end
+
+before_each
+@test "_pure_prompt_aws_profile: ensure default behaviour has no error" (
+    source (status dirname)/../conf.d/pure.fish
+
+    _pure_prompt_aws_profile
+) $status -eq $SUCCESS
+
+
+before_each
+@test "_pure_prompt_aws_profile: ensure default behaviour print nothing when no AWS variables" (
+    source (status dirname)/../conf.d/pure.fish
+
+    echo (_pure_prompt_aws_profile)
+) = $EMPTY
+
+before_each
+@test "_pure_prompt_aws_profile: print AWS_VAULT when present" (
+    set --universal pure_enable_aws_profile true
+    set --universal pure_symbol_aws_profile_prefix "🅰"
+    set --global AWS_VAULT my-vault
+
+    _pure_prompt_aws_profile
+) = '🅰my-vault'
+
+before_each
+@test "_pure_prompt_aws_profile: print AWS_PROFILE when present" (
+    set --universal pure_enable_aws_profile true
+    set --universal pure_symbol_aws_profile_prefix "🅰"
+    set --global AWS_PROFILE my-profile
+
+    _pure_prompt_aws_profile
+) = '🅰my-profile'
+
+
+before_each
+@test "_pure_prompt_aws_profile: print only AWS_VAULT when vault and profile are present" (
+    set --universal pure_enable_aws_profile true
+    set --universal pure_symbol_aws_profile_prefix "🅰"
+    set --global AWS_VAULT my-vault
+    set --global AWS_PROFILE my-profile
+
+    _pure_prompt_aws_profile
+) = '🅰my-vault'