Making build.py argument handling robust.

dbcli · Feb 23, 2018 · 4eb4555 · 4eb4555
1 parent 5950f60
commit 4eb4555
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/build.py b/build.py
@@ -141,8 +141,9 @@ def test():
     actions = sys.argv[1:] or default_actions
 
     for action in actions:
-        if action in targets:
-            targets[action]()
+        current_action = action.lower()
+        if current_action in targets:
+            targets[current_action]()
         else:
-            print('{0} is not a supported action'.format(action[1]))
+            print('{0} is not a supported action'.format(current_action[1]))
             print('Supported actions are {}'.format(list(targets.keys())))
diff --git a/tests/mssqlutils.py b/tests/mssqlutils.py
@@ -63,7 +63,7 @@ def run_and_return_string_from_formatter(client, sql, join=False, expanded=False
     """
 
     for rows, col, message, query, is_error in client.execute_single_statement(sql):
-        settings = OutputSettings(table_format='mssql', dcmlfmt='d', floatfmt='g',
+        settings = OutputSettings(table_format='psql', dcmlfmt='d', floatfmt='g',
                                   expanded=expanded)
         formatted = format_output(None, rows, col, message, settings)
         if join: