spaceport/apiserver/misc/hammertime.py

127 lines
3.1 KiB
Python

import django, sys, os
sys.path.append("..")
os.environ['DJANGO_SETTINGS_MODULE'] = 'apiserver.settings'
django.setup()
import random
import string
import datetime
from uuid import uuid4
import requests
from apiserver.api import models
API_URL = 'http://localhost:8002'
if len(sys.argv) == 2:
token = sys.argv[1]
else:
raise('Please provide a login token in the command line')
members = models.Member.objects.all()
member_ids = list(members.values_list('id', flat=True))
courses = models.Course.objects.all()
course_ids = list(courses.values_list('id', flat=True))
randstr = lambda: str(uuid4()).split('-')[-1]
build_user = lambda: dict(
first_name='test',
last_name='tester',
username=randstr()+'.tester',
password1='protospace',
password2='protospace',
email=randstr()+'@domain.com',
existing_member='false',
)
change_member = lambda: dict(
street_address=randstr(),
city=randstr(),
postal_code=randstr(),
first_name=randstr(),
)
build_transaction = lambda member_id: dict(
member_id=member_id,
date='2020-02-02',
amount=0,
account_type='Clearing',
info_source='DB Edit',
memo='Test transaction, ignore'
)
build_card = lambda member_id: dict(
member_id=member_id,
card_number=randstr(),
active_status='card_active',
notes='qot',
)
build_search = lambda: dict(
q=random.choice(string.ascii_lowercase),
seq=123,
)
build_session = lambda: dict(
datetime=datetime.datetime.now(),
course=random.choice(course_ids),
cost=0,
)
def poster(headers, payload, route):
r = requests.post(API_URL + route, data=payload, headers=headers, timeout=5)
if r.status_code < 300:
print(r.text.strip())
else:
print(r.text)
raise Exception('Bad response code ' + str(r.status_code))
def patcher(headers, payload, route):
r = requests.patch(API_URL + route, data=payload, headers=headers, timeout=5)
if r.status_code < 300:
print(r.text.strip())
else:
print(r.text)
raise Exception('Bad response code ' + str(r.status_code))
def register_member():
payload = build_user()
poster({}, payload, '/registration/')
def edit_member():
payload = change_member()
headers = {'Authorization': 'Token ' + token}
patcher(headers, payload, '/members/'+str(random.choice(member_ids))+'/')
def create_transaction():
headers = {'Authorization': 'Token ' + token}
payload = build_transaction(random.choice(member_ids))
poster(headers, payload, '/transactions/')
def create_card():
headers = {'Authorization': 'Token ' + token}
payload = build_card(random.choice(member_ids))
poster(headers, payload, '/cards/')
def perform_search():
headers = {'Authorization': 'Token ' + token}
payload = build_search()
poster(headers, payload, '/search/')
def create_session():
headers = {'Authorization': 'Token ' + token}
payload = build_session()
poster(headers, payload, '/sessions/')
while True:
register_member()
edit_member()
perform_search()
create_transaction()
create_card()
create_session()