added arg to show columns

Author: jeffreyaven

.gitignore

@@ -3,3 +3,4 @@ stackql_linux_amd64.zip
 nohup.out
 /.stackql
 /pystackql
+test.log

README.md

@@ -5,12 +5,13 @@ This repository contains a set of tests for StackQL providers. It is intended to
 ## Usage
 
 ```bash
-test-provider.sh <provider_name> [<signed> <provider_path> <test_for_anonymous_cols>]
+test-provider.sh <provider_name> [<signed> <provider_path> <show_columns> <test_for_anonymous_cols>]
 ```
 
 - `provider_name` - the name of the provider to test
 - `signed` - whether the provider is signed or not (default: false)
 - `provider_path` - the path to the provider (default: .)
+- `show_columns` - whether to show columns in output (default: false)
 - `test_for_anonymous_cols` - whether to test for anonymous columns (default: false)
 
 ## Example
@@ -31,16 +32,16 @@ true \
 ```
 
 ```bash
-# test a local aws unsigned provider from another directory
+# test a local aws unsigned provider from another directory showing columns
 sh test-provider.sh \
 aws \
 false \
-/mnt/c/Users/javen/Documents/LocalGitRepos/stackql/local-registry
+/mnt/c/Users/javen/Documents/LocalGitRepos/stackql/local-registry \
+true
 ```
 
-
 ```bash
-# test a local azure unsigned provider from the current directory with anonymous column checks
+# test a local azure unsigned provider from the current directory showing columns
 sh test-provider.sh \
 azure \
 false \

test-provider.py

@@ -2,6 +2,15 @@
 
 provider = sys.argv[1]
 
+showcols = None
+if len(sys.argv) > 2:
+      showcols = sys.argv[2]
+
+if showcols is not None:
+      showcols = True
+else:
+      showcols = False
+
 start_time = datetime.datetime.now()
 
 import psycopg
@@ -72,6 +81,8 @@ def run_query(query):
                   total_selectable_resources = total_selectable_resources + 1
                   iql_desc_query = "DESCRIBE EXTENDED %s.%s.%s" % (provider, service, resource)
                   desc = run_query(iql_desc_query)
+                  if showcols:
+                        print(desc)
                   print("%s columns in %s.%s.%s" % (len(desc), provider, service, resource))
 
             if len(methods.query("SQLVerb == 'INSERT'")) > 0:

test-provider.sh

@@ -1,7 +1,8 @@
 provider=$1
 signed=$2
 regpath=$3
-anoncolcheck=$4
+showcols=$4
+anoncolcheck=$5
 
 if [ -z "$provider" ]; then
     echo "provider (arg 1) must be set"
@@ -56,9 +57,9 @@ else
     echo "server is already running"
 fi
 
-if [ -z "$anoncolcheck" ]; then
+if [ -z "$showcols" ]; then
     python3 test-provider.py $provider
 else
-    # TODO implement anoncolcheck for server tests
-    python3 test-provider.py $provider $anoncolcheck
+    python3 test-provider.py $provider $showcols
+    # TODO implement anoncolcheck for server tests    
 fi