mirror of
https://github.com/pkkid/python-plexapi
synced 2024-11-23 12:13:18 +00:00
72 lines
3 KiB
Python
72 lines
3 KiB
Python
|
#!/usr/bin/env python3
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
"""
|
||
|
|
You can run this test suite with the following command:
|
||
|
|
>> python tests.py -u <USERNAME> -p <PASSWORD> -s <SERVERNAME>
|
||
|
|
"""
|
||
|
|
import argparse, pkgutil, sys, time, traceback
|
||
|
|
from os.path import dirname, abspath
|
||
|
|
sys.path.append(dirname(dirname(abspath(__file__))))
|
||
|
|
|
||
|
|
from plexapi.server import PlexServer
|
||
|
|
from plexapi.myplex import MyPlexAccount
|
||
|
|
from utils import log, itertests
|
||
|
|
|
||
|
|
|
||
|
|
def runtests(args):
|
||
|
|
# Register known tests
|
||
|
|
for loader, name, ispkg in pkgutil.iter_modules([dirname(abspath(__file__))]):
|
||
|
|
if name.startswith('test_'):
|
||
|
|
log(0, 'Registering tests from %s.py' % name)
|
||
|
|
loader.find_module(name).load_module(name)
|
||
|
|
# Create Account and Plex objects
|
||
|
|
account = MyPlexAccount.signin(args.username, args.password)
|
||
|
|
log(0, 'Signed into MyPlex as %s (%s)' % (account.username, account.email))
|
||
|
|
if args.resource:
|
||
|
|
plex = account.resource(args.resource).connect()
|
||
|
|
log(0, 'Connected to PlexServer resource %s' % plex.friendlyName)
|
||
|
|
else:
|
||
|
|
plex = PlexServer(args.baseurl, args.token)
|
||
|
|
log(0, 'Connected to PlexServer %s' % plex.friendlyName)
|
||
|
|
log(0, '')
|
||
|
|
# Run all specified tests
|
||
|
|
tests = {'passed':0, 'failed':0}
|
||
|
|
for test in itertests(args.query):
|
||
|
|
starttime = time.time()
|
||
|
|
log(0, test['name'], 'green')
|
||
|
|
try:
|
||
|
|
test['func'](account, plex)
|
||
|
|
runtime = time.time() - starttime
|
||
|
|
log(2, 'PASS! (runtime: %.3fs)' % runtime, 'blue')
|
||
|
|
tests['passed'] += 1
|
||
|
|
except Exception as err:
|
||
|
|
errstr = str(err)
|
||
|
|
errstr += '\n%s' % traceback.format_exc() if args.verbose else ''
|
||
|
|
log(2, 'FAIL: %s' % errstr, 'red')
|
||
|
|
tests['failed'] += 1
|
||
|
|
log(0, '')
|
||
|
|
# Log a final report
|
||
|
|
log(0, 'Tests Run: %s' % sum(tests.values()))
|
||
|
|
log(0, 'Tests Passed: %s' % tests['passed'])
|
||
|
|
if tests['failed']:
|
||
|
|
log(0, 'Tests Failed: %s' % tests['failed'], 'red')
|
||
|
|
if not tests['failed']:
|
||
|
|
log(0, '')
|
||
|
|
log(0, 'EVERYTHING OK!! :)')
|
||
|
|
raise SystemExit(tests['failed'])
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
parser = argparse.ArgumentParser(description='Run PlexAPI tests.')
|
||
|
|
# Auth Method 1: Pass --username, --password, and optionally --resource.
|
||
|
|
parser.add_argument('-u', '--username', help='Username for your MyPlex account.')
|
||
|
|
parser.add_argument('-p', '--password', help='Password for your MyPlex account.')
|
||
|
|
parser.add_argument('-r', '--resource', help='Name of the Plex resource (requires user/pass).')
|
||
|
|
# Auth Method 2: Pass --baseurl and --token.
|
||
|
|
parser.add_argument('-b', '--baseurl', help='Baseurl needed for auth token authentication')
|
||
|
|
parser.add_argument('-t', '--token', help='Auth token (instead of user/pass)')
|
||
|
|
# Misc options
|
||
|
|
parser.add_argument('-q', '--query', help='Only run the specified tests.')
|
||
|
|
parser.add_argument('-v', '--verbose', default=False, action='store_true', help='Print verbose logging.')
|
||
|
|
runtests(parser.parse_args())
|