mirror of
https://github.com/advplyr/audiobookshelf.git
synced 2026-07-12 12:21:34 +00:00
223e15643d
This file is provided to exercise endpoints and verify responses using `wiretap`. This is only for initial testing and will be replaced by chai or other automated testers.
259 lines
No EOL
6.7 KiB
Python
259 lines
No EOL
6.7 KiB
Python
# This Python script is designed to ping the dev server to allow
|
|
# for validation of the OpenAPI schema. Eventually this should
|
|
# be a nice thing written out in chai or something to incorporate
|
|
# to the normal ABS testing framework.
|
|
#
|
|
# Right now, this depends on `wiretap` (https://pb33f.io/wiretap/)
|
|
# being installed and set up. wiretap acts as a proxy to validate
|
|
# responses against the defined OpenAPI schema.
|
|
|
|
import requests
|
|
import json
|
|
import time
|
|
from copy import deepcopy
|
|
|
|
def write_to_json(fname, data):
|
|
json_obj = json.dumps(data, indent=4)
|
|
with open(fname,'w') as f:
|
|
f.write(json_obj)
|
|
|
|
# Server specific information
|
|
server_ip = "http://localhost"
|
|
server_port = 9090
|
|
|
|
###########################
|
|
### Setup
|
|
###########################
|
|
BASEURL = f"{server_ip}:{server_port}"
|
|
HEADERS = {}
|
|
|
|
def get_token():
|
|
url = f"{BASEURL}/login"
|
|
params = {'username': 'root', 'password': 'password'}
|
|
|
|
# Post response
|
|
r = requests.post(url = url, json = params)
|
|
data = r.json()
|
|
|
|
# Get the token out
|
|
my_token = data['user']['token']
|
|
|
|
return my_token
|
|
|
|
# Update HEADERS to include token
|
|
HEADERS['Authorization'] = f"Bearer {get_token()}"
|
|
|
|
# Get all libraries
|
|
def get_libraries():
|
|
url = f"{BASEURL}/api/libraries"
|
|
|
|
# Get response
|
|
r = requests.get(url = url, headers = HEADERS)
|
|
data = r.json()
|
|
|
|
# Get library IDs
|
|
libs = []
|
|
for lib in data['libraries']:
|
|
libs.append(lib['id'])
|
|
|
|
return libs
|
|
|
|
library_ids = get_libraries()
|
|
|
|
|
|
# Get books from library
|
|
def get_books(library_id, count=None):
|
|
url = f"{BASEURL}/api/libraries/{library_id}/items"
|
|
|
|
params = {}
|
|
if count:
|
|
params = {'limit': count}
|
|
|
|
# Get response
|
|
r = requests.get(url = url, headers = HEADERS, json = params)
|
|
data = r.json()
|
|
|
|
book_ids = []
|
|
for items in data['results']:
|
|
book_ids.append(items['id'])
|
|
|
|
return book_ids
|
|
|
|
books = get_books(library_ids[0])
|
|
|
|
# Get a library item
|
|
def get_item(id):
|
|
url = f"{BASEURL}/api/items/{id}"
|
|
# Get response
|
|
r = requests.get(url = url, headers = HEADERS)
|
|
data = r.json()
|
|
|
|
return data
|
|
|
|
print("Getting book details!")
|
|
resp = get_item(books[0])
|
|
write_to_json("book.json", resp)
|
|
|
|
###########################
|
|
### Collections tests
|
|
###########################
|
|
|
|
def create_collection(library_id, collection_name, desc=None, books=None):
|
|
url = f"{BASEURL}/api/collections"
|
|
|
|
params = {'libraryId': library_id,
|
|
'name': collection_name}
|
|
if desc:
|
|
params['description'] = desc
|
|
if books:
|
|
params['books'] = books
|
|
|
|
# Post response
|
|
r = requests.post(url = url, headers = HEADERS, json = params)
|
|
data = r.json()
|
|
|
|
print(r.request.url)
|
|
print(r.request.headers)
|
|
print(r.request.body)
|
|
|
|
return data
|
|
|
|
def get_all_collections():
|
|
url = f"{BASEURL}/api/collections"
|
|
|
|
# Post response
|
|
r = requests.get(url = url, headers = HEADERS)
|
|
data = r.json()
|
|
print(r.request.url)
|
|
print(r.request.headers)
|
|
print(r.request.body)
|
|
|
|
return data
|
|
|
|
def get_collections(library_id):
|
|
url = f"{BASEURL}/api/libraries/{library_id}/collections"
|
|
|
|
# Get response
|
|
r = requests.get(url = url, headers = HEADERS)
|
|
data = r.json()
|
|
|
|
colls = []
|
|
for item in data['results']:
|
|
colls.append(item['id'])
|
|
|
|
return colls
|
|
|
|
def update_collections(id, books):
|
|
url = f"{BASEURL}/api/collections/{id}"
|
|
print(books)
|
|
|
|
# Only update the name
|
|
data = {'name': 'New collection'}
|
|
r = requests.patch(url = url, headers = HEADERS, json = data)
|
|
|
|
# Only update the description
|
|
data = {'description': 'New description'}
|
|
r = requests.patch(url = url, headers = HEADERS, json = data)
|
|
|
|
# Only update the books
|
|
data = {'books': books}
|
|
r = requests.patch(url = url, headers = HEADERS, json = data)
|
|
|
|
# Update name and description
|
|
data = {'name': 'New collection',
|
|
'description': 'New description'}
|
|
r = requests.patch(url = url, headers = HEADERS, json = data)
|
|
|
|
# Update name and books
|
|
data = {'name': 'New collection',
|
|
'books': books}
|
|
r = requests.patch(url = url, headers = HEADERS, json = data)
|
|
|
|
# Update description and books
|
|
data = {'description': 'Second description',
|
|
'books': books}
|
|
r = requests.patch(url = url, headers = HEADERS, json = data)
|
|
|
|
# Update all things
|
|
data = {'name': 'Last collection',
|
|
'description': 'Last description',
|
|
'books': books}
|
|
r = requests.patch(url = url, headers = HEADERS, json = data)
|
|
|
|
def collection_add_book(id, book):
|
|
url = f"{BASEURL}/api/collections/{id}/book"
|
|
data = {'id': book}
|
|
r = requests.post(url = url, headers = HEADERS, json = data)
|
|
|
|
def collection_delete_book(id, book):
|
|
url = f"{BASEURL}/api/collections/{id}/book/{book}"
|
|
r = requests.delete(url = url, headers = HEADERS)
|
|
|
|
def collection_add_book_batch(id, books):
|
|
url = f"{BASEURL}/api/collections/{id}/batch/add"
|
|
data = {'books': books}
|
|
r = requests.patch(url = url, headers = HEADERS, json = data)
|
|
|
|
def delete_collections(id):
|
|
url = f"{BASEURL}/api/collections/{id}"
|
|
|
|
# Delete response
|
|
r = requests.delete(url = url, headers = HEADERS)
|
|
|
|
resp = create_collection(library_ids[0], "First None", None, None)
|
|
if True:
|
|
descriptions = [None, "First description", "Hello, here we go with some extra things."]
|
|
book_arrs = [None, [books[0]], books[1:3], books[3:7]]
|
|
|
|
#descriptions = [None]
|
|
descriptions = [None, descriptions[1]]
|
|
book_arrs = [None, book_arrs[2]]
|
|
|
|
title_idx = 0
|
|
for desc in descriptions:
|
|
for book in book_arrs:
|
|
title = f"Col {title_idx}"
|
|
print(f"Creating collection {title_idx}")
|
|
resp = create_collection(library_ids[0], title, desc, book)
|
|
print("Created Collection with following info:")
|
|
print(f" Title: {title}")
|
|
if desc:
|
|
print(f" Description: {desc}")
|
|
if book:
|
|
print(f" Books: {len(book)}")
|
|
print()
|
|
print("Response:")
|
|
write_to_json(f"collection_{title_idx}.json", resp)
|
|
|
|
#input("Press enter to continue...\n")
|
|
title_idx += 1
|
|
|
|
print("Getting collections")
|
|
# Get all collections in library
|
|
collections = get_collections(library_ids[0])
|
|
|
|
print("Updating a collection")
|
|
update_collections(collections[0], books[6:])
|
|
|
|
print("Add book to collection")
|
|
collection_add_book(collections[0], books[0])
|
|
collection_add_book(collections[0], books[0])
|
|
collection_add_book("", books[0])
|
|
collection_add_book(collections[0], "aoeu8a56oue")
|
|
|
|
print("Delete book from collection")
|
|
collection_delete_book(collections[0], books[0])
|
|
collection_delete_book(collections[0], books[0])
|
|
collection_delete_book("", books[0])
|
|
collection_delete_book(collections[0], "aoeu8a56oue")
|
|
|
|
print("Batch add to collection")
|
|
collection_add_book_batch(collections[0], books[0:3])
|
|
collection_add_book_batch(collections[0], books[0:3])
|
|
collection_add_book_batch("", books[0:3])
|
|
collection_add_book_batch(collections[0:3], [])
|
|
|
|
print("Deleting collections")
|
|
# Delete all collections
|
|
for collection in collections:
|
|
delete_collections(collection) |