mirror of
https://github.com/spl0k/supysonic.git
synced 2024-11-14 14:12:17 +00:00
99 lines
3.5 KiB
Python
99 lines
3.5 KiB
Python
# coding: utf-8
|
|
#
|
|
# This file is part of Supysonic.
|
|
# Supysonic is a Python implementation of the Subsonic server API.
|
|
#
|
|
# Copyright (C) 2017-2018 Alban 'spl0k' Féron
|
|
#
|
|
# Distributed under terms of the GNU AGPLv3 license.
|
|
|
|
import re
|
|
|
|
from lxml import etree
|
|
from supysonic.managers.user import UserManager
|
|
from supysonic.py23 import strtype
|
|
|
|
from ..testbase import TestBase
|
|
|
|
path_replace_regexp = re.compile(r'/(\w+)')
|
|
|
|
NS = 'http://subsonic.org/restapi'
|
|
NSMAP = { 'sub': NS }
|
|
|
|
class ApiTestBase(TestBase):
|
|
__with_api__ = True
|
|
|
|
def setUp(self):
|
|
super(ApiTestBase, self).setUp()
|
|
|
|
xsd = etree.parse('tests/assets/subsonic-rest-api-1.8.0.xsd')
|
|
self.schema = etree.XMLSchema(xsd)
|
|
|
|
def _find(self, xml, path):
|
|
"""
|
|
Helper method that insert the namespace in ElementPath 'path'
|
|
"""
|
|
|
|
path = path_replace_regexp.sub(r'/{{{}}}\1'.format(NS), path)
|
|
return xml.find(path)
|
|
|
|
def _xpath(self, elem, path):
|
|
"""
|
|
Helper method that insert a prefix and map the namespace in XPath 'path'
|
|
"""
|
|
|
|
path = path_replace_regexp.sub(r'/sub:\1', path)
|
|
return elem.xpath(path, namespaces = NSMAP)
|
|
|
|
def _make_request(self, endpoint, args = {}, tag = None, error = None, skip_post = False, skip_xsd = False):
|
|
"""
|
|
Makes both a GET and POST requests against the API, assert both get the same response.
|
|
If the user isn't provided with the 'u' and 'p' in 'args', the default 'alice' is used.
|
|
Validate the response against the Subsonic API XSD and assert it matches the expected tag or error.
|
|
|
|
:param endpoint: request endpoint, with the '/rest/'prefix nor the '.view' suffix
|
|
:param args: dict of parameters. 'u', 'p', 'c' and 'v' are automatically added
|
|
:param tag: topmost expected element, right beneath 'subsonic-response'
|
|
:param error: if the given 'args' should produce an error at 'endpoint', this is the expected error code
|
|
:param skip_post: don't do the POST request
|
|
:param skip_xsd: skip XSD validation
|
|
|
|
:return: a 2-tuple (resp, child) if no error, where 'resp' is the full response object, 'child' a
|
|
'lxml.etree.Element' mathching 'tag' (if any). If there's an error (when expected), only returns
|
|
the response object
|
|
"""
|
|
|
|
if not isinstance(args, dict):
|
|
raise TypeError("'args', expecting a dict, got " + type(args).__name__)
|
|
if tag and not isinstance(tag, strtype):
|
|
raise TypeError("'tag', expecting a str, got " + type(tag).__name__)
|
|
|
|
args.update({ 'c': 'tests', 'v': '1.8.0' })
|
|
if 'u' not in args:
|
|
args.update({ 'u': 'alice', 'p': 'Alic3' })
|
|
|
|
uri = '/rest/{}.view'.format(endpoint)
|
|
rg = self.client.get(uri, query_string = args)
|
|
if not skip_post:
|
|
rp = self.client.post(uri, data = args)
|
|
self.assertEqual(rg.data, rp.data)
|
|
|
|
xml = etree.fromstring(rg.data)
|
|
if not skip_xsd:
|
|
self.schema.assert_(xml)
|
|
|
|
if xml.get('status') == 'ok':
|
|
self.assertIsNone(error)
|
|
if tag:
|
|
self.assertEqual(xml[0].tag, '{{{}}}{}'.format(NS, tag))
|
|
return rg, xml[0]
|
|
else:
|
|
self.assertEqual(len(xml), 0)
|
|
return rg, None
|
|
else:
|
|
self.assertIsNone(tag)
|
|
self.assertEqual(xml[0].tag, '{{{}}}error'.format(NS))
|
|
self.assertEqual(xml[0].get('code'), str(error))
|
|
return rg
|
|
|