mirror of
https://github.com/pkkid/python-plexapi
synced 2024-11-26 13:40:22 +00:00
55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
|
"""
|
||
|
|
Test Library Functions
|
||
|
|
"""
|
||
|
|
import inspect, sys
|
||
|
|
import datetime, time
|
||
|
|
from plexapi import server
|
||
|
|
from plexapi.myplex import MyPlexUser
|
||
|
|
|
||
|
|
|
||
|
|
def log(indent, message):
|
||
|
|
dt = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S.%f')[:-3]
|
||
|
|
print('%s: %s%s' % (dt, ' '*indent, message))
|
||
|
|
|
||
|
|
|
||
|
|
def fetch_server(args):
|
||
|
|
if args.server:
|
||
|
|
user = MyPlexUser(args.username, args.password)
|
||
|
|
return user.getServer(args.server).connect()
|
||
|
|
return server.PlexServer()
|
||
|
|
|
||
|
|
|
||
|
|
def iter_tests(module, args):
|
||
|
|
module = sys.modules[module]
|
||
|
|
for func in sorted(module.__dict__.values()):
|
||
|
|
if inspect.isfunction(func) and inspect.getmodule(func) == module:
|
||
|
|
name = func.__name__
|
||
|
|
if name.startswith('test_') or name.startswith('example_') and (not args.name or args.name in name):
|
||
|
|
yield func
|
||
|
|
|
||
|
|
|
||
|
|
def run_tests(module, args):
|
||
|
|
plex = fetch_server(args)
|
||
|
|
tests = {'passed':0, 'failed':0}
|
||
|
|
for test in iter_tests(module, args):
|
||
|
|
startqueries = server.TOTAL_QUERIES
|
||
|
|
starttime = time.time()
|
||
|
|
log(0, test.__name__)
|
||
|
|
try:
|
||
|
|
test(plex)
|
||
|
|
tests['passed'] += 1
|
||
|
|
except Exception, err:
|
||
|
|
log(2, 'FAIL!: %s' % err)
|
||
|
|
tests['failed'] += 1
|
||
|
|
runtime = time.time() - starttime
|
||
|
|
log(2, 'Runtime: %.3fs' % runtime)
|
||
|
|
log(2, 'Queries: %s' % (server.TOTAL_QUERIES - startqueries))
|
||
|
|
log(0, '')
|
||
|
|
log(0, 'Tests Run: %s' % sum(tests.values()))
|
||
|
|
log(0, 'Tests Passed: %s' % tests['passed'])
|
||
|
|
log(0, 'Tests Failed: %s' % tests['failed'])
|
||
|
|
if not tests['failed']:
|
||
|
|
log(0, '')
|
||
|
|
log(0, 'EVERYTHING OK!! :)')
|
||
|
|
raise SystemExit(tests['failed'])
|