xos/tests/rest_usertest.py

# 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"]))