diff options
Diffstat (limited to 'test/cpp/qps/run_auth_test.py')
-rwxr-xr-x | test/cpp/qps/run_auth_test.py | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/test/cpp/qps/run_auth_test.py b/test/cpp/qps/run_auth_test.py index 75bdab22a4..06b064cad0 100755 --- a/test/cpp/qps/run_auth_test.py +++ b/test/cpp/qps/run_auth_test.py @@ -115,7 +115,11 @@ def main(): if not os.path.exists(ACCESS_TOKENS_DIR): os.makedirs(ACCESS_TOKENS_DIR) - email = sys.argv[2] + if len(sys.argv) > 2: + email = sys.argv[2] + else: + email = raw_input('Enter your e-mail id: ') + email = email.split('@')[0].lower() userId = re.sub('[.]', '', email) |