Travis: add online_sessions.sh

1 files changed, 46 insertions, 0 deletions

diff --git a/.travis/online_sessions.sh b/.travis/online_sessions.sh
new file mode 100755
index 0000000..79ce878
--- /dev/null
+++ b/.travis/online_sessions.sh
@@ -0,0 +1,46 @@
+#!/usr/bin/env bash
+
+set -o errexit -o nounset -o pipefail
+
+readonly db_path='.travis/test_db.csv'
+
+_online_sessions() {
+    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
+
+    echo "Running online_sessions.py..."
+    python3 -m bin.online_sessions "$@" "$db_path" "$output_path"
+
+    if file --brief --dereference --mime -- "$output_path" | grep --quiet -- '^binary'; then
+        return 0
+    fi
+
+    echo "Output:"
+    cat "$output_path"
+}
+
+online_sessions() {
+    # Get rid of:
+    # tkinter.TclError: no display name and no $DISPLAY environment variable
+    mkdir -p -- ~/.config/matplotlib
+    echo 'backend: Agg' > ~/.config/matplotlib/matplotlibrc
+
+    _online_sessions --group-by user --output-format csv
+    _online_sessions --group-by user --output-format json
+
+    _online_sessions --group-by user --output-format plot
+    _online_sessions --group-by hour --output-format plot
+    _online_sessions --group-by date --output-format plot
+    _online_sessions --group-by weekday --output-format plot
+}
+
+main() {
+    online_sessions
+}
+
+main "$@"