From 2fd53a7a01b15a9eff64508c57ad20c9c063f661 Mon Sep 17 00:00:00 2001
From: Jose Diaz-Gonzalez <email@josediazgonzalez.com>
Date: Fri, 21 Jun 2019 10:24:33 -0400
Subject: [PATCH] fix: allow default trace function to work

Also add some simple tests for the trace plugin.
---
 dokku                    |  2 +-
 tests/unit/40_trace.bats | 62 ++++++++++++++++++++++++++++++++++++++++
 2 files changed, 63 insertions(+), 1 deletion(-)
 create mode 100644 tests/unit/40_trace.bats

diff --git a/dokku b/dokku
index 09b64ecf1..561a82774 100755
--- a/dokku
+++ b/dokku
@@ -137,7 +137,7 @@ execute_dokku_cmd() {
     deploy | delete | ls | run | cleanup)
       local PLUGIN_NAME="00_dokku-standard"
       ;;
-    trace | url | urls | report)
+    url | urls | report)
       local PLUGIN_NAME="00_dokku-standard"
       ;;
   esac
diff --git a/tests/unit/40_trace.bats b/tests/unit/40_trace.bats
new file mode 100644
index 000000000..5a16b139a
--- /dev/null
+++ b/tests/unit/40_trace.bats
@@ -0,0 +1,62 @@
+#!/usr/bin/env bats
+
+load test_helper
+
+setup() {
+  global_setup
+  rm -f "$DOKKU_ROOT/.dokkurc/DOKKU_TRACE"
+}
+
+teardown() {
+  rm -f "$DOKKU_ROOT/.dokkurc/DOKKU_TRACE"
+  global_teardown
+}
+
+@test "(trace) trace:on" {
+  run /bin/bash -c "dokku trace:on"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+  run /bin/bash -c "test -f $DOKKU_ROOT/.dokkurc/DOKKU_TRACE"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+}
+
+@test "(trace) trace on" {
+  run /bin/bash -c "dokku trace on"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+  run /bin/bash -c "test -f $DOKKU_ROOT/.dokkurc/DOKKU_TRACE"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+}
+
+@test "(trace) trace:off, trace off" {
+  touch "$DOKKU_ROOT/.dokkurc/DOKKU_TRACE"
+  run /bin/bash -c "dokku trace:off"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+  run /bin/bash -c "test -f $DOKKU_ROOT/.dokkurc/DOKKU_TRACE"
+  echo "output: $output"
+  echo "status: $status"
+  assert_failure
+
+  touch "$DOKKU_ROOT/.dokkurc/DOKKU_TRACE"
+  run /bin/bash -c "dokku trace off"
+  echo "output: $output"
+  echo "status: $status"
+  assert_success
+
+
+  run /bin/bash -c "test -f $DOKKU_ROOT/.dokkurc/DOKKU_TRACE"
+  echo "output: $output"
+  echo "status: $status"
+  assert_failure
+}