| # Copyright 2017-present Open Networking Foundation |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| |
| |
| """ |
| UserTest - tests whether a user is able to fetch his own user record, |
| and modify fields. |
| |
| All users should be able to set their phone number. |
| Only admins should be able to set their is_admin bit |
| """ |
| |
| from __future__ import print_function |
| import json |
| import os |
| import requests |
| import sys |
| from urllib import urlencode |
| |
| from operator import itemgetter, attrgetter |
| |
| if len(sys.argv) != 6: |
| print( |
| "syntax: usertest <hostname> <username> <password> <admin_username> <admin_password>" |
| ) |
| sys.exit(-1) |
| |
| hostname = sys.argv[1] |
| username = sys.argv[2] |
| password = sys.argv[3] |
| |
| opencloud_auth = (username, password) |
| admin_auth = (sys.argv[4], sys.argv[5]) |
| |
| REST_API = "http://%s:8000/xos/" % hostname |
| USERS_API = REST_API + "users/" |
| |
| print("fetching user record for %s:" % username) |
| r = requests.get(USERS_API + "?" + urlencode({"email": username}), auth=opencloud_auth) |
| for user in r.json(): |
| print(" ", user["email"]) |
| |
| myself = r.json()[0] |
| |
| if myself["phone"] == "123": |
| myself["phone"] = "456" |
| else: |
| myself["phone"] = "123" |
| |
| r = requests.put(USERS_API + str(myself["id"]) + "/", data=myself, auth=opencloud_auth) |
| if r.status_code == 200: |
| print("I updated my phone to", myself["phone"]) |
| else: |
| print("I failed to update my phone") |
| |
| if myself["is_admin"]: |
| myself["is_admin"] = False |
| else: |
| myself["is_admin"] = True |
| |
| r = requests.put(USERS_API + str(myself["id"]) + "/", data=myself, auth=opencloud_auth) |
| if r.status_code == 200: |
| print("I updated my is_admin to", myself["is_admin"]) |
| else: |
| print("I failed to update my is_admin") |
| |
| r = requests.get(USERS_API + "?email=jhh@cs.arizona.edu", auth=opencloud_auth) |
| if len(r.json()) > 0: |
| print("I was able to read jhh@cs.arizona.edu") |
| else: |
| print("I was not able to read jhh@cs.arizona.edu") |
| |
| # get john's record using admin, so we can try to update it |
| r = requests.get(USERS_API + "?email=jhh@cs.arizona.edu", auth=admin_auth) |
| if len(r.json()) > 0: |
| print("Admin was able to read jhh@cs.arizona.edu") |
| jhh = r.json()[0] |
| else: |
| print("Admin was not able to read jhh@cs.arizona.edu") |
| jhh = None |
| |
| if jhh: |
| # try to update john's user record |
| r = requests.put(USERS_API + str(jhh["id"]) + "/", data=jhh, auth=opencloud_auth) |
| if r.status_code == 200: |
| print("I was able to update user", str(jhh["id"])) |
| else: |
| print("I was not able to update user", str(jhh["id"])) |