From 7a403c7f93ebaddcfd9f38c7d3ca1ac1014b23cf Mon Sep 17 00:00:00 2001
From: Matthias Beyer <mail@beyermatthias.de>
Date: Sun, 31 Jan 2016 18:03:49 +0100
Subject: [PATCH] tests: Add imag-store tests

---
 imag-store/tests/001-create_test.sh | 49 +++++++++++++++++++++++
 imag-store/tests/utils.sh           | 61 +++++++++++++++++++++++++++++
 2 files changed, 110 insertions(+)
 create mode 100644 imag-store/tests/001-create_test.sh
 create mode 100644 imag-store/tests/utils.sh

diff --git a/imag-store/tests/001-create_test.sh b/imag-store/tests/001-create_test.sh
new file mode 100644
index 00000000..b7d641b6
--- /dev/null
+++ b/imag-store/tests/001-create_test.sh
@@ -0,0 +1,49 @@
+#!/usr/bin/env bash
+
+source $(dirname ${BASH_SOURCE[0]})/utils.sh
+
+test_call() {
+    imag-store create -p /test-call
+    if [[ ! $? -eq 0 ]]; then
+        err "Return value should be zero, was non-zero"
+        return 1;
+    fi
+}
+
+test_mkstore() {
+    imag-store create -p /test-mkstore || { err "Calling imag failed"; return 1; }
+    if [[ -d ${STORE} ]]; then
+        out "Store exists."
+    else
+        err "No store created"
+        return 1
+    fi
+}
+
+test_std_header() {
+    local expected=$(cat <<EOS
+---
+[imag]
+links = []
+version = "0.1.0"
+---
+
+EOS
+)
+
+    imag-store create -p /test-std-header
+    local result=$(cat ${STORE}/test-std-header)
+    if [[ "$expected" == "$result" ]]; then
+        out "Expected store entry == result"
+    else
+        err "${STORE}/test differs from expected"
+        return 1
+    fi
+}
+
+invoke_tests                    \
+    test_call                   \
+    test_mkstore                \
+    test_std_header             \
+    test_std_header_plus_custom
+
diff --git a/imag-store/tests/utils.sh b/imag-store/tests/utils.sh
new file mode 100644
index 00000000..7f12cb55
--- /dev/null
+++ b/imag-store/tests/utils.sh
@@ -0,0 +1,61 @@
+#!/usr/bin/env bash
+
+COLOR_OFF='\e[0m'       # Text Reset
+RED='\e[0;31m'          # Red
+YELLOW='\e[0;33m'       # Yellow
+GREEN='\e[0;32m'        # Green
+
+RUNTIME="/tmp"
+STORE="${RUNTIME}/store"
+
+out() {
+    echo -e "${YELLOW}:: $*${COLOR_OFF}"
+}
+
+success() {
+    echo -e "${GREEN}>> $*${COLOR_OFF}"
+}
+
+err() {
+    echo -e "${RED}!! $*${COLOR_OFF}"
+}
+
+imag-store() {
+    local searchdir=$(dirname ${BASH_SOURCE[0]})/../target/debug/
+    [[ -d $searchdir ]] || { err "FATAL: No directory $searchdir"; exit 1; }
+    local bin=$(find $searchdir -iname imag-store -type f -executable)
+    local flags="--debug --rtp $RUNTIME"
+    out "Calling '$bin $flags $*'"
+    $bin $flags $*
+}
+
+reset_store() {
+    rm -r "${STORE}"
+}
+
+call_test() {
+    out "-- TESTING: '$1' --"
+    $1
+    result=$?
+    if [[ -z "$DONT_RESET_STORE" ]]; then
+        out "Reseting store"
+        reset_store
+        out "Store reset done"
+    fi
+    [[ $result -eq 0 ]] || { err "-- FAILED: '$1'. Exiting."; exit 1; }
+    success "-- SUCCESS: '$1' --"
+}
+
+invoke_tests() {
+    out "Invoking tests."
+    if [[ ! -z "$INVOKE_TEST" ]]; then
+        out "Invoking only $INVOKE_TEST"
+        call_test "$INVOKE_TEST"
+    else
+        out "Invoking $*"
+        for t in $*; do
+            call_test "$t"
+        done
+    fi
+}
+