home/.zsh/themes/spaceship-prompt/tests/host.test.zsh

145 lines
4.7 KiB
Bash
Raw Normal View History

2023-09-06 14:03:56 +02:00
#!/usr/bin/env zsh
# Required for shunit2 to run correctly
setopt shwordsplit
SHUNIT_PARENT=$0
# ------------------------------------------------------------------------------
# SHUNIT2 HOOKS
# ------------------------------------------------------------------------------
oneTimeSetUp() {
export TERM="xterm-256color"
SPACESHIP_PROMPT_FIRST_PREFIX_SHOW=true
SPACESHIP_PROMPT_ADD_NEWLINE=false
SPACESHIP_PROMPT_ORDER=(host)
source "spaceship.zsh"
}
setUp() {
SPACESHIP_HOST_SHOW=true
SPACESHIP_HOST_SHOW_FULL=false
SPACESHIP_HOST_PREFIX="at "
SPACESHIP_HOST_SUFFIX="$SPACESHIP_PROMPT_DEFAULT_SUFFIX"
SPACESHIP_HOST_COLOR="blue"
SPACESHIP_HOST_COLOR_SSH="green"
}
oneTimeTearDown() {
unset SPACESHIP_PROMPT_FIRST_PREFIX_SHOW
unset SPACESHIP_PROMPT_ADD_NEWLINE
unset SPACESHIP_PROMPT_ORDER
}
tearDown() {
unset SPACESHIP_HOST_SHOW
unset SPACESHIP_HOST_SHOW_FULL
unset SPACESHIP_HOST_PREFIX
unset SPACESHIP_HOST_SUFFIX
unset SPACESHIP_HOST_COLOR
unset SPACESHIP_HOST_COLOR_SSH
}
# ------------------------------------------------------------------------------
# TEST CASES
# ------------------------------------------------------------------------------
test_host() {
SPACESHIP_HOST_SHOW=true
unset SSH_CONNECTION
assertEquals "do not render host by default" "" "$(spaceship::testkit::render_prompt)"
}
test_host_show() {
local expected="%{%B%}at %{%b%}%{%B%F{$SPACESHIP_HOST_COLOR}%}%m%{%b%f%}%{%B%} %{%b%}"
local expected_full="%{%B%}at %{%b%}%{%B%F{$SPACESHIP_HOST_COLOR}%}%M%{%b%f%}%{%B%} %{%b%}"
local expected_ssh="%{%B%}at %{%b%}%{%B%F{$SPACESHIP_HOST_COLOR_SSH}%}%m%{%b%f%}%{%B%} %{%b%}"
local expected_ssh_full="%{%B%}at %{%b%}%{%B%F{$SPACESHIP_HOST_COLOR_SSH}%}%M%{%b%f%}%{%B%} %{%b%}"
SPACESHIP_HOST_SHOW=true
SSH_CONNECTION=''
SPACESHIP_HOST_SHOW_FULL=false
assertEquals "SHOW=true, SSH=false, FULL=false" "" "$(spaceship::testkit::render_prompt)"
SPACESHIP_HOST_SHOW_FULL=true
assertEquals "SHOW=true, SSH=false, FULL=true" "" "$(spaceship::testkit::render_prompt)"
SSH_CONNECTION='ssh'
SPACESHIP_HOST_SHOW_FULL=false
assertEquals "SHOW=true, SSH=true, FULL=false" "$expected_ssh" "$(spaceship::testkit::render_prompt)"
SPACESHIP_HOST_SHOW_FULL=true
assertEquals "SHOW=true, SSH=true, FULL=true" "$expected_ssh_full" "$(spaceship::testkit::render_prompt)"
SPACESHIP_HOST_SHOW=always
SSH_CONNECTION='ssh'
SPACESHIP_HOST_SHOW_FULL=false
assertEquals "SHOW=always, SSH=true, FULL=false" "$expected_ssh" "$(spaceship::testkit::render_prompt)"
SPACESHIP_HOST_SHOW_FULL=true
assertEquals "SHOW=always, SSH=true, FULL=true" "$expected_ssh_full" "$(spaceship::testkit::render_prompt)"
SSH_CONNECTION=''
SPACESHIP_HOST_SHOW_FULL=false
assertEquals "SHOW=always, SSH=false, FULL=false" "$expected" "$(spaceship::testkit::render_prompt)"
SPACESHIP_HOST_SHOW_FULL=true
assertEquals "SHOW=always, SSH=false, FULL=true" "$expected_full" "$(spaceship::testkit::render_prompt)"
SPACESHIP_HOST_SHOW=false
SSH_CONNECTION='ssh'
SPACESHIP_HOST_SHOW_FULL=false
assertEquals "SHOW=false, SSH=true, FULL=false" "" "$(spaceship::testkit::render_prompt)"
SPACESHIP_HOST_SHOW_FULL=true
assertEquals "SHOW=false, SSH=true, FULL=true" "" "$(spaceship::testkit::render_prompt)"
SSH_CONNECTION=''
SPACESHIP_HOST_SHOW_FULL=false
assertEquals "SHOW=false, SSH=false, FULL=false" "" "$(spaceship::testkit::render_prompt)"
SPACESHIP_HOST_SHOW_FULL=true
assertEquals "SHOW=false, SSH=false, FULL=true" "" "$(spaceship::testkit::render_prompt)"
unset SPACESHIP_HOST_SHOW_FULL
unset SSH_CONNECTION
}
test_host_color() {
SPACESHIP_HOST_SHOW=always # force user display
SPACESHIP_HOST_COLOR=blue
local expected="%{%B%}at %{%b%}%{%B%F{$SPACESHIP_HOST_COLOR}%}%m%{%b%f%}%{%B%} %{%b%}"
local actual="$(spaceship::testkit::render_prompt)"
assertEquals "render host with custom color" "$expected" "$actual"
}
test_host_prefix() {
SPACESHIP_HOST_SHOW=always # force user display
SPACESHIP_HOST_PREFIX='prefix'
SPACESHIP_HOST_SUFFIX=''
local expected="%{%B%}$SPACESHIP_HOST_PREFIX%{%b%}%{%B%F{blue}%}%m%{%b%f%}"
local actual="$(spaceship::testkit::render_prompt)"
assertEquals "render host with prefix" "$expected" "$actual"
}
test_host_suffix() {
SPACESHIP_HOST_SHOW=always # force user display
SPACESHIP_HOST_PREFIX=''
SPACESHIP_HOST_SUFFIX='suffix'
local expected="%{%B%F{blue}%}%m%{%b%f%}%{%B%}$SPACESHIP_HOST_SUFFIX%{%b%}"
local actual="$(spaceship::testkit::render_prompt)"
assertEquals "render host with suffix" "$expected" "$actual"
}
# ------------------------------------------------------------------------------
# SHUNIT2
# Run tests with shunit2
# ------------------------------------------------------------------------------
source tests/shunit2/shunit2