.ci/ -> test/

5 files changed, 193 insertions, 0 deletions

diff --git a/test/bin/main.sh b/test/bin/main.sh
new file mode 100755
index 0000000..6207677
--- /dev/null
+++ b/test/bin/main.sh
@@ -0,0 +1,19 @@
+#!/usr/bin/env bash
+
+# Copyright (c) 2019 Egor Tensin <Egor.Tensin@gmail.com>
+# This file is part of the "VK scripts" project.
+# For details, see https://github.com/egor-tensin/vk-scripts.
+# Distributed under the MIT License.
+
+set -o errexit -o nounset -o pipefail
+
+script_dir="$( dirname -- "${BASH_SOURCE[0]}" )"
+script_dir="$( cd -- "$script_dir" && pwd )"
+readonly script_dir
+
+"$script_dir/mutuals.sh"
+
+"$script_dir/sessions.sh"
+
+"$script_dir/status_once.sh"
+"$script_dir/status.sh"
diff --git a/test/bin/mutuals.sh b/test/bin/mutuals.sh
new file mode 100755
index 0000000..383dbdd
--- /dev/null
+++ b/test/bin/mutuals.sh
@@ -0,0 +1,23 @@
+#!/usr/bin/env bash
+
+# Copyright (c) 2019 Egor Tensin <Egor.Tensin@gmail.com>
+# This file is part of the "VK scripts" project.
+# For details, see https://github.com/egor-tensin/vk-scripts.
+# Distributed under the MIT License.
+
+set -o errexit -o nounset -o pipefail
+
+script_dir="$( dirname -- "${BASH_SOURCE[0]}" )"
+script_dir="$( cd -- "$script_dir" && pwd )"
+readonly script_dir
+
+test_users() {
+    "$script_dir/../lib/test.sh" vk.mutuals --format csv  "$@"
+    "$script_dir/../lib/test.sh" vk.mutuals --format json "$@"
+}
+
+main() {
+    test_users kreed58 maxkorzh_official
+}
+
+main
diff --git a/test/bin/sessions.sh b/test/bin/sessions.sh
new file mode 100755
index 0000000..7a01dc3
--- /dev/null
+++ b/test/bin/sessions.sh
@@ -0,0 +1,69 @@
+#!/usr/bin/env bash
+
+# Copyright (c) 2019 Egor Tensin <Egor.Tensin@gmail.com>
+# This file is part of the "VK scripts" project.
+# For details, see https://github.com/egor-tensin/vk-scripts.
+# Distributed under the MIT License.
+
+set -o errexit -o nounset -o pipefail
+
+script_dir="$( dirname -- "${BASH_SOURCE[0]}" )"
+script_dir="$( cd -- "$script_dir" && pwd )"
+readonly script_dir
+script_name="$( basename -- "${BASH_SOURCE[0]}" )"
+readonly script_name
+
+readonly db_path="$script_dir/../share/test_db.csv"
+
+dump() {
+    local msg
+    for msg; do
+        echo "$script_name: $msg"
+    done
+}
+
+test_output() {
+    local output_path
+    output_path="$( mktemp --dry-run )"
+
+    local rm_aux_files
+    rm_aux_files="$( printf -- 'rm -f -- %q' "$output_path" )"
+
+    trap "$rm_aux_files" RETURN
+
+    "$script_dir/../lib/test.sh" vk.tracking.sessions "$@" "$db_path" "$output_path"
+
+    if file --brief --dereference --mime -- "$output_path" | grep --quiet -- 'charset=binary$'; then
+        dump 'Output is a binary file, not going to show that'
+        return 0
+    fi
+
+    cat "$output_path"
+}
+
+group_by() {
+    test_output --output-format csv  --group-by "$@"
+    test_output --output-format json --group-by "$@"
+    test_output --output-format plot --group-by "$@"
+}
+
+online_sessions() {
+    group_by user
+    group_by hour
+    group_by date
+    group_by weekday
+}
+
+fix_matplotlib() {
+    # Get rid of:
+    # tkinter.TclError: no display name and no $DISPLAY environment variable
+    mkdir -p -- ~/.config/matplotlib
+    echo 'backend: Agg' > ~/.config/matplotlib/matplotlibrc
+}
+
+main() {
+    fix_matplotlib
+    online_sessions
+}
+
+main
diff --git a/test/bin/status.sh b/test/bin/status.sh
new file mode 100755
index 0000000..9ff5d6a
--- /dev/null
+++ b/test/bin/status.sh
@@ -0,0 +1,60 @@
+#!/usr/bin/env bash
+
+# Copyright (c) 2019 Egor Tensin <Egor.Tensin@gmail.com>
+# This file is part of the "VK scripts" project.
+# For details, see https://github.com/egor-tensin/vk-scripts.
+# Distributed under the MIT License.
+
+set -o errexit -o nounset -o pipefail
+
+script_dir="$( dirname -- "${BASH_SOURCE[0]}" )"
+script_dir="$( cd -- "$script_dir" && pwd )"
+readonly script_dir
+script_name="$( basename -- "${BASH_SOURCE[0]}" )"
+readonly script_name
+
+dump() {
+    local msg
+    for msg; do
+        echo "$script_name: $msg"
+    done
+}
+
+test_users() {
+    local log_path
+    log_path="$( mktemp )"
+    local db_path
+    db_path="$( mktemp --dry-run )"
+
+    local rm_aux_files
+    rm_aux_files="$( printf -- 'rm -f -- %q %q' "$log_path" "$db_path" )"
+    trap "$rm_aux_files" RETURN
+
+    "$script_dir/../lib/test.sh" vk.tracking.status "$@" --log "$log_path" --format csv --output "$db_path" &
+    local pid="$!"
+
+    sleep 3
+    dump "Log file path: $log_path"
+    dump "DB file path: $db_path"
+    dump "PID: $pid"
+
+    local timeout=10
+    dump "Sleeping for $timeout seconds..."
+    sleep "$timeout"
+
+    dump 'Terminating track_status.py...'
+    kill "$pid"
+    dump 'Waiting for track_status.py to terminate...'
+    wait "$pid" || true
+
+    dump "Log file:"
+    cat "$log_path"
+    dump "DB:"
+    cat "$db_path"
+}
+
+main() {
+    test_users egor.tensin
+}
+
+main "$@"
diff --git a/test/bin/status_once.sh b/test/bin/status_once.sh
new file mode 100755
index 0000000..890cc3d
--- /dev/null
+++ b/test/bin/status_once.sh
@@ -0,0 +1,22 @@
+#!/usr/bin/env bash
+
+# Copyright (c) 2019 Egor Tensin <Egor.Tensin@gmail.com>
+# This file is part of the "VK scripts" project.
+# For details, see https://github.com/egor-tensin/vk-scripts.
+# Distributed under the MIT License.
+
+set -o errexit -o nounset -o pipefail
+
+script_dir="$( dirname -- "${BASH_SOURCE[0]}" )"
+script_dir="$( cd -- "$script_dir" && pwd )"
+readonly script_dir
+
+test_users() {
+    "$script_dir/../lib/test.sh" vk.tracking.status --only-once "$@"
+}
+
+main() {
+    test_users egor.tensin
+}
+
+main