client/pre-commit.sh

#!/bin/sh
#
# Git pre-commit script that runs our unit tests as needed.
#
# To install:
#   ln -s ../../pre-commit.sh .git/hooks/pre-commit

# Redirect output to stderr.
exec 1>&2

function log() {
    echo 'pre-commit:' $@
}

function dirModified() {
    # sets is_modified if files in directory $1 have been modified.
    changed_files="$(git diff --cached --name-only --diff-filter=ACM $1 | wc -l | tr -d '[:space:]')"
    if [ "$changed_files" != "0" ]; then
        is_modified=true
    else
        is_modified=false
    fi
    if [ "$is_modified" = true ]; then
        log Modifications detected in "$1"
    fi
}

# cd to root (may not need this)
ROOT_DIR="$(git rev-parse --show-toplevel)"
cd "$ROOT_DIR"

log 'Running pre-commit checks, pass --no-verify to "git commit" to'
log 'disable.'
echo

dirModified wandb
wandb_modified="$is_modified"

# Unset all the env variables that are set during git commit hooks,
# which break the tests currently.
unset GIT_DIR
unset GIT_INDEX_FILE
unset GIT_AUTHOR_DATE
unset GIT_AUTHOR_NAME
unset GIT_PREFIX
unset GIT_AUTHOR_EMAIL


if [ "$wandb_modified" = true ]; then
    log "Running wandb tests."
    pytest
    if [ $? -ne 0 ]; then
        wandb_failed=true
    fi
fi

failed=false
if [ "$wandb_failed" = true ]; then
    log "wandb tests failed."
    log "\"cd $ROOT_DIR; pytest\" to run the tests."
    failed=true
fi

if [ "$failed" = true ]; then
    exit 1
else
    log "pre-commit check success"
fi