from django.contrib.auth.models import User import debttest import json import requests # First of all, it initializes debt information. for i in range(1, 10): username = "******".format(i) try: user = User.objects.get(username=username) user.delete() print("\tDeleted user {0}".format(username)) except User.DoesNotExist: # no problem. continue print("---------------") for (username, pwd) in debttest.create_users(10): user = User.objects.create_user(username, password=pwd) user.save() print("\tCreated user {0}".format(username)) print("Initialization Successful!")
try: res = requests.get(link).json() return res except Exception: print("ERROR: Cannot get {0}".format(link)) exit(1) def check_key(debt_json, key): if key not in debt_json: print("{0} not in {1}.".format((key, debt_json))) exit(1) userN = 10 user_pairs = debttest.create_users(userN) # get id of each user print("1. Getting users list.") users_json = get_json_or_error("http://localhost:8000/users/") users = [(uname, upwd, get_id(users_json, uname)) for (uname, upwd) in user_pairs] # remove existing debts print("2. Checking GET http://localhost:8000/debts/") debts_old = get_json_or_error("http://localhost:8000/debts/") print("3. Checking DELETE http://localhost:8000/debts/") for debt in debts_old: print("\tDeleting debt {0}".format(debt["id"])) try:
import json import requests # First of all, it initializes debt information. def remove_user(username): try: user = User.objects.get(username=username) user.delete() print("\tDeleted user {0}".format(username)) except User.DoesNotExist: # no problem. pass return remove_user("debt_admin") for i in range(1, 10): username = "******".format(i) remove_user(username) print("---------------") newusers = [("debt_admin", "debt_adminpasswd")] newusers = newusers + debttest.create_users(10) for (username, pwd) in newusers: user = User.objects.create_user(username, password=pwd) user.save() print("\tCreated user {0}".format(username)) print("Initialization Successful!")