diff --git a/run_tests.sh b/run_tests.sh new file mode 100644 index 0000000..f007569 --- /dev/null +++ b/run_tests.sh @@ -0,0 +1,46 @@ +#!/bin/bash + +exec >> /home/azureuser/dbt_run.log 2>&1 + +# Define the Slack webhook URL +script_dir=$(dirname "$0") +webhooks_file="slack_webhook_urls.txt" +env_file="$script_dir/$webhooks_file" +if [ -f "$env_file" ]; then + export $(grep -v '^#' "$env_file" | xargs) +else + echo "Error: $webhooks_file file not found in the script directory." + exit 1 +fi + +# Messages to be sent to Slack +slack_failure_message=":rotating_light::rotating_light::rotating_light: One or more failures in dbt tests in production. :rotating_light::rotating_light::rotating_light:" +slack_success_message=":white_check_mark::white_check_mark::white_check_mark: dbt tests executed successfully in production. :white_check_mark::white_check_mark::white_check_mark:" + +# Initialize the failure flag +has_any_step_failed=0 + +cd /home/azureuser/data-dwh-dbt-project + +# Update from git +echo "Updating dbt project from git." +git checkout master +git pull + +# Activate venv +source venv/bin/activate + +# Run tests +echo "Triggering dbt test" +dbt test +if [ $? -ne 0 ]; then + has_any_step_failed=1 +fi + +# Check if any step failed and send a Slack message +if [ $has_any_step_failed -eq 1 ]; then + curl -X POST -H 'Content-type: application/json' --data "{\"text\":\"$slack_failure_message\"}" $SLACK_ALERT_WEBHOOK_URL +fi +if [ $has_any_step_failed -eq 0 ]; then + curl -X POST -H 'Content-type: application/json' --data "{\"text\":\"$slack_success_message\"}" $SLACK_RECEIPT_WEBHOOK_URL +fi \ No newline at end of file