Fill in remaining hooks

hooks/environment

@@ -0,0 +1,11 @@
+#!/bin/bash
+
+set -euo pipefail
+
+echo "--- :datadog: Recording step start time"
+
+BUILDKITE_PLUGIN_DATADOG_STATS_STEP_START_TIME_MS=$(date +%s%3N)
+
+export BUILDKITE_PLUGIN_DATADOG_STATS_STEP_START_TIME_MS
+
+echo "Step started at ${BUILDKITE_PLUGIN_DATADOG_STATS_STEP_START_TIME_MS}"

hooks/post-checkout

@@ -15,10 +15,10 @@ basedir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && cd .. && pwd )"
 
 NOW=$(date +%s%3N)
 
-echo "--- :datadog: Recording command run time"
+echo "--- :datadog: Recording checkout run time"
 
 METRIC_NAME=$(getMetricName "checkout.duration")
-METRIC_VALUE=$(timeDiff "$BUILDKITE_PLUGIN_DATADOG_STATS_COMMAND_START_TIME_MS")
+METRIC_VALUE=$(timeDiff "$BUILDKITE_PLUGIN_DATADOG_STATS_CHECKOUT_START_TIME_MS")
 TAGS=$(getTags)
 
 reportToDatadog "${METRIC_NAME}" "${METRIC_VALUE}" d "${TAGS}"

hooks/pre-checkout

@@ -8,4 +8,4 @@ BUILDKITE_PLUGIN_DATADOG_STATS_CHECKOUT_START_TIME_MS=$(date +%s%3N)
 
 export BUILDKITE_PLUGIN_DATADOG_STATS_CHECKOUT_START_TIME_MS
 
-echo "Command started at ${BUILDKITE_PLUGIN_DATADOG_STATS_CHECKOUT_START_TIME_MS}"
+echo "Checkout started at ${BUILDKITE_PLUGIN_DATADOG_STATS_CHECKOUT_START_TIME_MS}"

hooks/pre-exit

@@ -0,0 +1,24 @@
+#!/bin/bash
+
+set -euo pipefail
+
+basedir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && cd .. && pwd )"
+
+# shellcheck source=lib/get-tags.sh
+. "$basedir/lib/get-tags.sh"
+# shellcheck source=lib/report-to-datadog.sh
+. "$basedir/lib/report-to-datadog.sh"
+# shellcheck source=lib/time-diff.sh
+. "$basedir/lib/time-diff.sh"
+# shellcheck source=lib/get-metric-name.sh
+. "$basedir/lib/get-metric-name.sh"
+
+NOW=$(date +%s%3N)
+
+echo "--- :datadog: Recording step run time"
+
+METRIC_NAME=$(getMetricName "step.duration")
+METRIC_VALUE=$(timeDiff "$BUILDKITE_PLUGIN_DATADOG_STATS_STEP_START_TIME_MS")
+TAGS=$(getTags)
+
+reportToDatadog "${METRIC_NAME}" "${METRIC_VALUE}" d "${TAGS}"

tests/environment.bats

@@ -0,0 +1,10 @@
+#!/usr/bin/env bats
+
+load "$BATS_PATH/load.bash"
+
+@test "Records the start time of the step" {
+  run "$PWD/hooks/environment"
+
+  assert_success
+  assert_output --partial "Step started at $(date +%s)"
+}

tests/post-checkout.bats

@@ -0,0 +1,24 @@
+#!/usr/bin/env bats
+
+load "$BATS_PATH/load.bash"
+
+@test "Sets the correct tags and runtime for the master branch" {
+  NOW=$(date +%s%3N)
+  export BUILDKITE_PLUGIN_DATADOG_STATS_CHECKOUT_START_TIME_MS=$((NOW-900))
+  export BUILDKITE_BRANCH=master
+  export BUILDKITE_PIPELINE_SLUG=monorepo
+  export BUILDKITE_COMMAND="cd somewhere && make do-something"
+  export BUILDKITE_LABEL=":shipit: deploy-prod"
+
+  run "$PWD/hooks/post-checkout"
+
+  assert_success
+  assert_output --partial "Reporting buildkite.steps.checkout.duration with value=90"
+  assert_output --partial "tags=is_master:true,pipeline_slug:monorepo,step_command:cd somewhere && make do-something,step_label::shipit: deploy-prod,retry_count:0"
+
+  unset BUILDKITE_PLUGIN_DATADOG_STATS_COMMAND_START_TIME_MS
+  unset BUILDKITE_BRANCH
+  unset BUILDKITE_PIPELINE_SLUG
+  unset BUILDKITE_COMMAND
+  unset BUILDKITE_LABEL
+}

tests/post-commands.bats → tests/post-command.bats

@@ -4,7 +4,7 @@ load "$BATS_PATH/load.bash"
 
 @test "Sets the correct tags and runtime for the master branch" {
   NOW=$(date +%s%3N)
-  export BUILDKITE_PLUGIN_DATADOG_STATS_COMMAND_START_TIME_MS=$(($NOW-900))
+  export BUILDKITE_PLUGIN_DATADOG_STATS_COMMAND_START_TIME_MS=$((NOW-900))
   export BUILDKITE_BRANCH=master
   export BUILDKITE_PIPELINE_SLUG=monorepo
   export BUILDKITE_COMMAND="cd somewhere && make do-something"
@@ -25,7 +25,7 @@ load "$BATS_PATH/load.bash"
 
 @test "Sets the correct tags and runtime for a non-master branch" {
   NOW=$(date +%s%3N)
-  export BUILDKITE_PLUGIN_DATADOG_STATS_COMMAND_START_TIME_MS=$(($NOW-900))
+  export BUILDKITE_PLUGIN_DATADOG_STATS_COMMAND_START_TIME_MS=$((NOW-900))
   export BUILDKITE_BRANCH=some-branch
   export BUILDKITE_PIPELINE_SLUG=monorepo
   export BUILDKITE_COMMAND="cd somewhere && make do-something"
@@ -46,7 +46,7 @@ load "$BATS_PATH/load.bash"
 
 @test "It supports specifying additional tags by value and env var" {
   NOW=$(date +%s%3N)
-  export BUILDKITE_PLUGIN_DATADOG_STATS_COMMAND_START_TIME_MS=$(($NOW-900))
+  export BUILDKITE_PLUGIN_DATADOG_STATS_COMMAND_START_TIME_MS=$((NOW-900))
   export BUILDKITE_BRANCH=some-branch
   export BUILDKITE_PIPELINE_SLUG=monorepo
   export BUILDKITE_COMMAND="cd somewhere && make do-something"

tests/pre-checkout.bats

@@ -0,0 +1,10 @@
+#!/usr/bin/env bats
+
+load "$BATS_PATH/load.bash"
+
+@test "Records the start time of the checkout" {
+  run "$PWD/hooks/pre-checkout"
+
+  assert_success
+  assert_output --partial "Checkout started at $(date +%s)"
+}

tests/pre-commands.bats → tests/pre-command.bats

@@ -6,6 +6,5 @@ load "$BATS_PATH/load.bash"
   run "$PWD/hooks/pre-command"
 
   assert_success
-  # Check that we produced the right output at least to the second
   assert_output --partial "Command started at $(date +%s)"
 }

tests/pre-exit.bats

@@ -0,0 +1,24 @@
+#!/usr/bin/env bats
+
+load "$BATS_PATH/load.bash"
+
+@test "Sets the correct tags and runtime for the master branch" {
+  NOW=$(date +%s%3N)
+  export BUILDKITE_PLUGIN_DATADOG_STATS_STEP_START_TIME_MS=$((NOW-900))
+  export BUILDKITE_BRANCH=master
+  export BUILDKITE_PIPELINE_SLUG=monorepo
+  export BUILDKITE_COMMAND="cd somewhere && make do-something"
+  export BUILDKITE_LABEL=":shipit: deploy-prod"
+
+  run "$PWD/hooks/pre-exit"
+
+  assert_success
+  assert_output --partial "Reporting buildkite.steps.step.duration with value=90"
+  assert_output --partial "tags=is_master:true,pipeline_slug:monorepo,step_command:cd somewhere && make do-something,step_label::shipit: deploy-prod,retry_count:0"
+
+  unset BUILDKITE_PLUGIN_DATADOG_STATS_COMMAND_START_TIME_MS
+  unset BUILDKITE_BRANCH
+  unset BUILDKITE_PIPELINE_SLUG
+  unset BUILDKITE_COMMAND
+  unset BUILDKITE_LABEL
+}