Browse Source

verify Trying to make better defaults

Mike Smith 11 years ago
parent
commit
f454e0efcf
1 changed files with 24 additions and 9 deletions
  1. 24 9
      toolset/run-tests.py

+ 24 - 9
toolset/run-tests.py

@@ -44,26 +44,41 @@ def main(argv=None):
     ##########################################################
     # Set up default values
     ##########################################################        
-    serverHost = os.environ.get('TFB_SERVER_HOST', 'localhost')
-    clientHost = os.environ.get('TFB_CLIENT_HOST', 'localhost')
+    serverHost = os.environ.get('TFB_SERVER_HOST')
+    clientHost = os.environ.get('TFB_CLIENT_HOST')
     clientUser = os.environ.get('TFB_CLIENT_USER')
     clientIden = os.environ.get('TFB_CLIENT_IDENTITY_FILE')
     databaHost = os.getenv('TFB_DATABASE_HOST', clientHost)
     databaUser = os.getenv('TFB_DATABASE_USER', clientUser)
     dbIdenFile = os.getenv('TFB_DATABASE_IDENTITY_FILE', clientIden)
 
+    if serverHost != None:
+        defaults["server_host"] = serverHost
+    if clientHost != None:
+        defaults["client_host"] = clientHost
+    if clientUser != None:
+        defaults["client_user"] = clientUser
+    if clientIden != None:
+        defaults["client_identity_file"] = clientIden
+    if databaHost != None:
+        defaults["database_host"] = databaHost
+    if databaUser != None:
+        defaults["database_user"] = databaUser
+    if dbIdenFile != None:
+        defaults["database_identity_file"] = dbIdenFile
+
     ##########################################################
     # Set up argument parser
     ##########################################################
     parser = argparse.ArgumentParser(description='Run the Framework Benchmarking test suite.',
         parents=[conf_parser])
-    parser.add_argument('-s', '--server-host', default=serverHost, required=true, help='The application server.')
-    parser.add_argument('-c', '--client-host', default=clientHost, required=true, help='The client / load generation server.')
-    parser.add_argument('-u', '--client-user', default=clientUser, required=true, help='The username to use for SSH to the client instance.')
-    parser.add_argument('-i', '--client-identity-file', default=clientIden, required=true, dest='client_identity_file', help='The key to use for SSH to the client instance.')
-    parser.add_argument('-d', '--database-host', default=databaHost, help='The database server.  If not provided, defaults to the value of --client-host.')
-    parser.add_argument('--database-user', default=databaUser, help='The username to use for SSH to the database instance.  If not provided, defaults to the value of --client-user.')
-    parser.add_argument('--database-identity-file', default=dbIdenFile, dest='database_identity_file', help='The key to use for SSH to the database instance.  If not provided, defaults to the value of --client-identity-file.')
+    parser.add_argument('-s', '--server-host', required=true, help='The application server.')
+    parser.add_argument('-c', '--client-host', required=true, help='The client / load generation server.')
+    parser.add_argument('-u', '--client-user', required=true, help='The username to use for SSH to the client instance.')
+    parser.add_argument('-i', '--client-identity-file', required=true, dest='client_identity_file', help='The key to use for SSH to the client instance.')
+    parser.add_argument('-d', '--database-host', help='The database server.  If not provided, defaults to the value of --client-host.')
+    parser.add_argument('--database-user', help='The username to use for SSH to the database instance.  If not provided, defaults to the value of --client-user.')
+    parser.add_argument('--database-identity-file', dest='database_identity_file', help='The key to use for SSH to the database instance.  If not provided, defaults to the value of --client-identity-file.')
     parser.add_argument('-p', dest='password_prompt', action='store_true')
     parser.add_argument('--install-software', action='store_true', help='runs the installation script before running the rest of the commands')
     parser.add_argument('--install', choices=['client', 'database', 'server', 'all'], default='all', help='Allows you to only install the server, client, or database software')