# Create test users
#
# All users initially are part of default team.
#
admin = user_email_prefix + '+team_mgmt-admin' + '@' + user_email_domain
userA = user_email_prefix + '+team_mgmt-a' + '@' + user_email_domain
userB = user_email_prefix + '+team_mgmt-b' + '@' + user_email_domain
teamA = team_prefix + 'A'
teamB = team_prefix + 'B'
print('Creating test users...')
try:
ok, res = sdclient.create_user_invite(admin, first_name='TestUser', last_name='Admin', system_role='ROLE_CUSTOMER')
if not ok:
print(('-- User creation failed:', res, '. Exiting.'))
sys.exit(1)
else:
print(('-- User \'', admin, '\' created successfully.'))
ok, res = sdclient.create_user_invite(userA, first_name='TestUser', last_name='Alpha')
if not ok:
print(('-- User creation failed:', res, '. Exiting.'))
sys.exit(1)
else:
print(('-- User \'', userA, '\' created successfully.'))
ok, res = sdclient.create_user_invite(userB, first_name='TestUser', last_name='Beta')
if not ok:
print 'usage: %s <sysdig-token> team-name user-name' % sys.argv[0]
print 'You can find your token at https://app.sysdigcloud.com/#/settings/user'
sys.exit(1)
sdc_token = sys.argv[1]
#
# Instantiate the SDC client
#
sdclient = SdcClient(sdc_token, sdc_url='https://app-staging.sysdigcloud.com')
team_name = sys.argv[2]
user_name = sys.argv[3]
print 'Trying to invite a user ', user_name
res = sdclient.create_user_invite(user_name)
if res[0] == False:
print 'User creation failed: ', res[1]
else:
print 'User creation succeeded'
print 'Now trying to create a team with name ', team_name
res = sdclient.create_team(team_name)
if res[0] == False:
print 'Team creation failed: ', res[1]
else:
print 'Team creation succeeded.', res[1]
print 'Now trying to find team with name ', team_name
res = sdclient.get_team(team_name)
if res[0] == False:
print (filter)
print('Now trying to create a team with name:', 'ColonialOne')
ok, res = sdclient.create_team( 'ColonialOne', filter=nsfilter, show="container" )
if not ok:
print('Team creation failed:', res, '. Exiting.')
else:
print('Team creation succeeded.', res)
#### USER INVITES ColonialOne ####
print('Trying to invite a user:'******'*****@*****.**')
ok, res = sdclient.create_user_invite('*****@*****.**')
if not ok:
if res == 'user ' + '*****@*****.**' + ' already exists':
print('User creation failed because', '*****@*****.**', 'already exists. Continuing.')
else:
print('User creation failed:', res, '. Exiting.')
else:
print('User creation succeeded')
print('Trying to invite a user:'******'*****@*****.**')
ok, res = sdclient.create_user_invite('*****@*****.**')
if not ok:
if res == 'user ' + '*****@*****.**' + ' already exists':
print('User creation failed because', '*****@*****.**', 'already exists. Continuing.')
else:
print('User creation failed:', res, '. Exiting.')
else:
