github/supysonic
1
0
Fork 0
mirror of https://github.com/spl0k/supysonic.git synced 2024-09-19 10:51:04 +00:00
Code Issues Releases Wiki Activity

Prevent tests from speing some ResourceWarnings

Browse Source
This commit is contained in:
Alban Féron 2020-11-08 15:40:37 +01:00
parent 5c46c96b53
commit 8e3cd5ef4d
No known key found for this signature in database
GPG Key ID: 8CE0313646D16165
1 changed files with 67 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

119
tests/api/test_media.py
View File

@ -10,6 +10,7 @@
import os.path
import uuid
from contextlib import closing
from io import BytesIO
from PIL import Image
from pony.orm import db_session
@ -85,17 +86,19 @@ class MediaTestCase(ApiTestBase):
"stream", {"id": str(self.trackid), "size": "640x480"}, error=0
)
rv = self.client.get(
"/rest/stream.view",
query_string={
"u": "alice",
"p": "Alic3",
"c": "tests",
"id": str(self.trackid),
},
)
self.assertEqual(rv.status_code, 200)
self.assertEqual(len(rv.data), 23)
with closing(
self.client.get(
"/rest/stream.view",
query_string={
"u": "alice",
"p": "Alic3",
"c": "tests",
"id": str(self.trackid),
},
)
) as rv:
self.assertEqual(rv.status_code, 200)
self.assertEqual(len(rv.data), 23)
with db_session:
self.assertEqual(Track[self.trackid].play_count, 1)
@ -105,17 +108,19 @@ class MediaTestCase(ApiTestBase):
self._make_request("download", {"id": str(uuid.uuid4())}, error=70)
# download single file
rv = self.client.get(
"/rest/download.view",
query_string={
"u": "alice",
"p": "Alic3",
"c": "tests",
"id": str(self.trackid),
},
)
self.assertEqual(rv.status_code, 200)
self.assertEqual(len(rv.data), 23)
with closing(
self.client.get(
"/rest/download.view",
query_string={
"u": "alice",
"p": "Alic3",
"c": "tests",
"id": str(self.trackid),
},
)
) as rv:
self.assertEqual(rv.status_code, 200)
self.assertEqual(len(rv.data), 23)
with db_session:
self.assertEqual(Track[self.trackid].play_count, 0)
@ -142,47 +147,57 @@ class MediaTestCase(ApiTestBase):
)
args = {"u": "alice", "p": "Alic3", "c": "tests", "id": str(self.folderid)}
rv = self.client.get("/rest/getCoverArt.view", query_string=args)
self.assertEqual(rv.status_code, 200)
self.assertEqual(rv.mimetype, "image/jpeg")
im = Image.open(BytesIO(rv.data))
self.assertEqual(im.format, "JPEG")
self.assertEqual(im.size, (420, 420))
with closing(
self.client.get("/rest/getCoverArt.view", query_string=args)
) as rv:
self.assertEqual(rv.status_code, 200)
self.assertEqual(rv.mimetype, "image/jpeg")
im = Image.open(BytesIO(rv.data))
self.assertEqual(im.format, "JPEG")
self.assertEqual(im.size, (420, 420))
args["size"] = 600
rv = self.client.get("/rest/getCoverArt.view", query_string=args)
self.assertEqual(rv.status_code, 200)
self.assertEqual(rv.mimetype, "image/jpeg")
im = Image.open(BytesIO(rv.data))
self.assertEqual(im.format, "JPEG")
self.assertEqual(im.size, (420, 420))
with closing(
self.client.get("/rest/getCoverArt.view", query_string=args)
) as rv:
self.assertEqual(rv.status_code, 200)
self.assertEqual(rv.mimetype, "image/jpeg")
im = Image.open(BytesIO(rv.data))
self.assertEqual(im.format, "JPEG")
self.assertEqual(im.size, (420, 420))
args["size"] = 120
rv = self.client.get("/rest/getCoverArt.view", query_string=args)
self.assertEqual(rv.status_code, 200)
self.assertEqual(rv.mimetype, "image/jpeg")
im = Image.open(BytesIO(rv.data))
self.assertEqual(im.format, "JPEG")
self.assertEqual(im.size, (120, 120))
with closing(
self.client.get("/rest/getCoverArt.view", query_string=args)
) as rv:
self.assertEqual(rv.status_code, 200)
self.assertEqual(rv.mimetype, "image/jpeg")
im = Image.open(BytesIO(rv.data))
self.assertEqual(im.format, "JPEG")
self.assertEqual(im.size, (120, 120))
# rerequest, just in case
rv = self.client.get("/rest/getCoverArt.view", query_string=args)
self.assertEqual(rv.status_code, 200)
self.assertEqual(rv.mimetype, "image/jpeg")
im = Image.open(BytesIO(rv.data))
self.assertEqual(im.format, "JPEG")
self.assertEqual(im.size, (120, 120))
with closing(
self.client.get("/rest/getCoverArt.view", query_string=args)
) as rv:
self.assertEqual(rv.status_code, 200)
self.assertEqual(rv.mimetype, "image/jpeg")
im = Image.open(BytesIO(rv.data))
self.assertEqual(im.format, "JPEG")
self.assertEqual(im.size, (120, 120))
# TODO test non square covers
# Test extracting cover art from embeded media
for args["id"] in self.formats:
rv = self.client.get("/rest/getCoverArt.view", query_string=args)
self.assertEqual(rv.status_code, 200)
self.assertEqual(rv.mimetype, "image/png")
im = Image.open(BytesIO(rv.data))
self.assertEqual(im.format, "PNG")
self.assertEqual(im.size, (120, 120))
with closing(
self.client.get("/rest/getCoverArt.view", query_string=args)
) as rv:
self.assertEqual(rv.status_code, 200)
self.assertEqual(rv.mimetype, "image/png")
im = Image.open(BytesIO(rv.data))
self.assertEqual(im.format, "PNG")
self.assertEqual(im.size, (120, 120))
def test_get_avatar(self):
self._make_request("getAvatar", error=0)