Merge pull request #135 from elainemccloskey/SUP-2842

(SUP-2842) Add test for st0287_check_db_table_sizes

Author: MartyEwings

spec/acceptance/st0287_check_db_table_sizes_spec.rb

@@ -0,0 +1,7 @@
+require 'spec_helper_acceptance'
+describe 'tasks' do
+  it 'table sizes output sucessfully' do
+    result = run_bolt_task('support_tasks::st0287_check_db_table_sizes', 'dbname' => 'all')
+    expect(result.stdout).to contain(%r{success})
+  end
+end

tasks/st0287_check_db_table_sizes.json

@@ -7,5 +7,8 @@
       "description": "The name of the db to connect to",
       "type": "Enum['pe-puppetdb', 'pe-postgres', 'pe-classifier', 'pe-rbac', 'pe-activity', 'pe-orchestrator', 'postgres', all]"
     }
-  }
+  },
+    "implementations": [
+      {"name": "st0287_check_db_table_sizes.sh", "requirements": ["shell"], "files": ["bash_task_helper/files/task_helper.sh"], "input_method": "environment"}
+      ]
 }

tasks/st0287_check_db_table_sizes.sh

@@ -4,6 +4,8 @@
 
 declare PT_dbname
 dbname=$PT_dbname
+declare PT__installdir
+source "$PT__installdir/bash_task_helper/files/task_helper.sh"
 
 function getdbTables() {
   path=$1
@@ -36,7 +38,7 @@ elif puppet resource service postgresql-* | grep -q running; then
     dbname="puppetdb"
   fi
 else
-  echo "Node not running pe-postgresql or postgresql service, please select node which is."
+  success '{ "status": "success - Node not running pe-postgresql or postgresql service, please select node which is." }'
 fi
 
 #Run for the correct environment
@@ -51,8 +53,7 @@ fi
       getdbTables "psql" "${dbname}" "${postgresservice}" "puppetdb"
       ;;
     *)
-      echo "Cannot Determine if Puppet Enterprise or Puppet Open Source "
-      exit 1
+      fail "Cannot Determine if Puppet Enterprise or Puppet Open Source"
   esac
 
-echo " -- ST#0287 Task ended: $(date +%s) --"
+success '{ "status": "success - Task completed" }'