From a8ae04c8e479bd75e441c7684b18e13d73dac862 Mon Sep 17 00:00:00 2001
From: Jpeedroza <pedrozajoaopedro@gmail.com>
Date: Wed, 8 Sep 2021 14:32:51 -0300
Subject: [PATCH] Created test to check availability of postgres

---
 vkpr-test/postgres.bats | 41 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 vkpr-test/postgres.bats

diff --git a/vkpr-test/postgres.bats b/vkpr-test/postgres.bats
new file mode 100644
index 00000000..99098417
--- /dev/null
+++ b/vkpr-test/postgres.bats
@@ -0,0 +1,41 @@
+
+VKPR_HOME=~/.vkpr
+
+setup_file() {
+    load 'common-setup'
+    _common_setup
+    if [ "$VKPR_TEST_SKIP_SETUP" == "true" ]; then
+        echo "setup: skipping setup due to VKPR_TEST_SKIP_SETUP=true" >&3
+    else
+        echo "setup: installing postgres...." >&3
+        rit vkpr postgres install
+        kubectl wait --for=condition=ready --timeout=1m pod --all
+        sleep 2
+    fi
+}
+
+setup() {
+    load $VKPR_HOME/bats/bats-support/load.bash
+    load $VKPR_HOME/bats/bats-assert/load.bash
+}
+
+@test "Ping in DB and must show if is accepting connections" {
+    run ping_db
+    actual="${lines[0]}"
+    expected="postgres-postgresql:5432 - accepting connections"
+    assert_equal "$actual" "$expected"
+}
+
+teardown_file() {
+    if [ "$VKPR_TEST_SKIP_TEARDOWN" == "true" ]; then
+        echo "teardown: skipping teardown due to VKPR_TEST_SKIP_TEARDOWN=true" >&3
+    else
+        echo "teardown: uninstalling postgres...." >&3
+        rit vkpr postgres remove
+        rit vkpr infra down
+    fi
+}
+
+ping_db(){
+    $VKPR_HOME/bin/kubectl run test-db --rm -it --restart='Never' --image docker.io/bitnami/postgresql:11.13.0-debian-10-r12 --env="PGUSER=postgres" --env="PGPASSWORD=123" --env="PGHOST=postgres-postgresql" --env="PGPORT=5432" --command -- pg_isready
+}
\ No newline at end of file