159 lines
4.2 KiB
Python
159 lines
4.2 KiB
Python
import unittest
|
|
from datetime import date
|
|
|
|
from mopidy.scanner import Scanner, translator
|
|
from mopidy.models import Track, Artist, Album
|
|
|
|
from tests import data_folder
|
|
|
|
class FakeGstDate(object):
|
|
def __init__(self, year, month, day):
|
|
self.year = year
|
|
self.month = month
|
|
self.day = day
|
|
|
|
class TranslatorTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.data = {
|
|
'uri': 'uri',
|
|
'album': u'albumname',
|
|
'track-number': 1,
|
|
'artist': u'name',
|
|
'album-artist': 'albumartistname',
|
|
'title': u'trackname',
|
|
'track-count': 2,
|
|
'date': FakeGstDate(2006, 1, 1,),
|
|
'container-format': u'ID3 tag',
|
|
'duration': 4531,
|
|
}
|
|
|
|
self.album = {
|
|
'name': 'albumname',
|
|
'num_tracks': 2,
|
|
}
|
|
|
|
self.artist = {
|
|
'name': 'name',
|
|
}
|
|
|
|
self.albumartist = {
|
|
'name': 'albumartistname',
|
|
}
|
|
|
|
self.track = {
|
|
'uri': 'uri',
|
|
'name': 'trackname',
|
|
'date': date(2006, 1, 1),
|
|
'track_no': 1,
|
|
'length': 4531,
|
|
}
|
|
|
|
def build_track(self):
|
|
if self.albumartist:
|
|
self.album['artists'] = [Artist(**self.albumartist)]
|
|
self.track['album'] = Album(**self.album)
|
|
self.track['artists'] = [Artist(**self.artist)]
|
|
return Track(**self.track)
|
|
|
|
def check(self):
|
|
expected = self.build_track()
|
|
actual = translator(self.data)
|
|
self.assertEqual(expected, actual)
|
|
|
|
def test_basic_data(self):
|
|
self.check()
|
|
|
|
def test_missing_track_number(self):
|
|
del self.data['track-number']
|
|
del self.track['track_no']
|
|
self.check()
|
|
|
|
def test_missing_track_count(self):
|
|
del self.data['track-count']
|
|
del self.album['num_tracks']
|
|
self.check()
|
|
|
|
def test_missing_track_name(self):
|
|
del self.data['title']
|
|
del self.track['name']
|
|
self.check()
|
|
|
|
def test_missing_album_name(self):
|
|
del self.data['album']
|
|
del self.album['name']
|
|
self.check()
|
|
|
|
def test_missing_artist_name(self):
|
|
del self.data['artist']
|
|
del self.artist['name']
|
|
self.check()
|
|
|
|
def test_missing_album_artist(self):
|
|
del self.data['album-artist']
|
|
del self.albumartist['name']
|
|
self.check()
|
|
|
|
def test_missing_date(self):
|
|
del self.data['date']
|
|
del self.track['date']
|
|
self.check()
|
|
|
|
class ScannerTest(unittest.TestCase):
|
|
def setUp(self):
|
|
self.errors = {}
|
|
self.data = {}
|
|
|
|
def scan(self, path):
|
|
scanner = Scanner(data_folder(path),
|
|
self.data_callback, self.error_callback)
|
|
scanner.start()
|
|
|
|
def check(self, name, key, value):
|
|
name = data_folder(name)
|
|
self.assertEqual(self.data[name][key], value)
|
|
|
|
def data_callback(self, data):
|
|
uri = data['uri'][len('file://'):]
|
|
self.data[uri] = data
|
|
|
|
def error_callback(self, uri, errors):
|
|
uri = uri[len('file://'):]
|
|
self.errors[uri] = errors
|
|
|
|
def test_data_is_set(self):
|
|
self.scan('scanner/simple')
|
|
self.assert_(self.data)
|
|
|
|
def test_errors_is_not_set(self):
|
|
self.scan('scanner/simple')
|
|
self.assert_(not self.errors)
|
|
|
|
def test_uri_is_set(self):
|
|
self.scan('scanner/simple')
|
|
self.check('scanner/simple/song1.mp3', 'uri', 'file://'
|
|
+ data_folder('scanner/simple/song1.mp3'))
|
|
|
|
def test_duration_is_set(self):
|
|
self.scan('scanner/simple')
|
|
self.check('scanner/simple/song1.mp3', 'duration', 4680)
|
|
|
|
def test_artist_is_set(self):
|
|
self.scan('scanner/simple')
|
|
self.check('scanner/simple/song1.mp3', 'artist', 'name')
|
|
|
|
def test_album_is_set(self):
|
|
self.scan('scanner/simple')
|
|
self.check('scanner/simple/song1.mp3', 'album', 'albumname')
|
|
|
|
def test_track_is_set(self):
|
|
self.scan('scanner/simple')
|
|
self.check('scanner/simple/song1.mp3', 'title', 'trackname')
|
|
|
|
def test_nonexistant_folder_does_not_fail(self):
|
|
self.scan('scanner/does-not-exist')
|
|
self.assert_(not self.errors)
|
|
|
|
def test_other_media_is_ignored(self):
|
|
self.scan('scanner/image')
|
|
self.assert_(self.errors)
|