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!")