tests - Support running as `root` (for CI)

[tpg/opendispense2.git] / tests / TEST_basic.sh

diff --git a/tests/TEST_basic.sh b/tests/TEST_basic.sh
index f2f1301..76c6490 100755 (executable)
--- a/tests/TEST_basic.sh
+++ b/tests/TEST_basic.sh
@@ -1,20 +1,33 @@
 #!/bin/bash
-set -eux
+set -eu
 TESTNAME=basic
 
 . _common.sh
 
-if $DISPENSE acct tpg; then
-       FAIL "Database contains '$USER'"
+# (CI runs as root, and root is auto-added by databse creation)
+if [[ "$USER" -ne "root" ]]; then
+       # Ensure that the databse doesn't already contain the current user
+       # - 1. Protects against running the test against the production database
+       # - 2. Allows adding the current user as an admin
+       if $DISPENSE acct "$USER"; then
+               FAIL "Database contains '$USER'"
+       fi
+       # Add the current user as an admin
+       sqlite3 "${BASEDIR}cokebank.db" "INSERT INTO accounts (acct_name,acct_is_admin,acct_uid) VALUES ('${USER}',1,1);"
+else
+       if $DISPENSE acct accmurph; then
+               FAIL "Database contains 'accmurph' (running as root, might be using prodution)"
+       fi
 fi
 
-sqlite3 "${BASEDIR}cokebank.db" "INSERT INTO accounts (acct_name,acct_is_admin,acct_uid) VALUES ('${USER}',1,1);"
-
+# Try to add a new user
 TRY_COMMAND "$DISPENSE user add unittest_user0"
 
+# Ensure that the add worked
 LOG "Checking for test user"
 TRY_COMMAND $DISPENSE acct unittest_user0 | grep ': $    0.00'
 
+# Manipulate user's balance
 TRY_COMMAND $DISPENSE acct unittest_user0 +100 Unit_test
 TRY_COMMAND $DISPENSE acct unittest_user0 | grep ': $    1.00'
 TRY_COMMAND $DISPENSE acct unittest_user0 -100 Unit_test