More tests

tests/db.py

@@ -47,6 +47,50 @@ class DbTestCase(unittest.TestCase):
 
         return root_folder, child_folder
 
+    def create_some_tracks(self, artist = None, album = None):
+        root, child = self.create_some_folders()
+
+        if not artist:
+                artist = db.Artist()
+                artist.name = u'Test Artist'
+
+        if not album:
+                album = db.Album()
+                album.artist = artist
+                album.name = u'Test Album'
+
+        track1 = db.Track()
+        track1.title = 'Track Title'
+        track1.album = album
+        track1.artist = artist
+        track1.disc = 1
+        track1.number = 1
+        track1.duration = 3
+        track1.bitrate = 320
+        track1.path = '/child/01'
+        track1.content_type = u'audio/mpeg'
+        track1.last_modification = 1234
+        track1.root_folder = root
+        track1.folder = child
+        self.store.add(track1)
+
+        track2 = db.Track()
+        track2.title = 'One Awesome Song'
+        track2.album = album
+        track2.artist = artist
+        track2.disc = 1
+        track2.number = 2
+        track2.duration = 5
+        track2.bitrate = 96
+        track2.path = '/child/02'
+        track2.content_type = u'audio/mpeg'
+        track2.last_modification = 1234
+        track2.root_folder = root
+        track2.folder = child
+        self.store.add(track2)
+
+        return track1, track2
+
     def test_folder_base(self):
         root_folder, child_folder = self.create_some_folders()
 
@@ -63,7 +107,7 @@ class DbTestCase(unittest.TestCase):
         self.assertTrue(root[u'isDir'])
         self.assertEqual(root[u'title'], u'Root folder')
         self.assertEqual(root[u'album'], u'Root folder')
-        self.assertTrue(date_regex.match(root['created']))
+        self.assertRegexpMatches(root['created'], date_regex)
 
         child = child_folder.as_subsonic_child(user)
         self.assertIn(u'parent', child)
@@ -102,7 +146,7 @@ class DbTestCase(unittest.TestCase):
         self.assertIn(u'starred', root)
         self.assertIn(u'userRating', root)
         self.assertIn(u'averageRating', root)
-        self.assertTrue(date_regex.match(root[u'starred']))
+        self.assertRegexpMatches(root[u'starred'], date_regex)
         self.assertEqual(root[u'userRating'], 2)
         self.assertEqual(root[u'averageRating'], 3.5)
 
@@ -110,3 +154,87 @@ class DbTestCase(unittest.TestCase):
         self.assertNotIn(u'starred', child)
         self.assertNotIn(u'userRating', child)
 
+    def test_artist(self):
+        artist = db.Artist()
+        artist.name = u'Test Artist'
+        self.store.add(artist)
+
+        # Assuming SQLite doesn't enforce foreign key constraints
+        MockUser = namedtuple(u'User', [ u'id' ])
+        user = MockUser(uuid.uuid4())
+
+        star = db.StarredArtist()
+        star.user_id = user.id
+        star.starred_id = artist.id
+        self.store.add(star)
+
+        artist_dict = artist.as_subsonic_artist(user)
+        self.assertIsInstance(artist_dict, dict)
+        self.assertIn(u'id', artist_dict)
+        self.assertIn(u'name', artist_dict)
+        self.assertIn(u'albumCount', artist_dict)
+        self.assertIn(u'starred', artist_dict)
+        self.assertEqual(artist_dict[u'name'], u'Test Artist')
+        self.assertEqual(artist_dict[u'albumCount'], 0)
+        self.assertRegexpMatches(artist_dict[u'starred'], date_regex)
+
+        album = db.Album()
+        album.name = u'Test Artist' # self-titled
+        artist.albums.add(album)
+
+        album = db.Album()
+        album.name = u'The Album After The Frist One'
+        artist.albums.add(album)
+
+        artist_dict = artist.as_subsonic_artist(user)
+        self.assertEqual(artist_dict[u'albumCount'], 2)
+
+    def test_album(self):
+        artist = db.Artist()
+        artist.name = u'Test Artist'
+
+        album = db.Album()
+        album.artist = artist
+        album.name = u'Test Album'
+
+        # Assuming SQLite doesn't enforce foreign key constraints
+        MockUser = namedtuple(u'User', [ u'id' ])
+        user = MockUser(uuid.uuid4())
+
+        star = db.StarredAlbum()
+        star.user_id = user.id
+        star.starred = album
+
+        self.store.add(album)
+        self.store.add(star)
+
+        # No tracks, shouldn't be stored under normal circumstances
+        self.assertRaises(ValueError, album.as_subsonic_album, user)
+
+        self.create_some_tracks(artist, album)
+
+        album_dict = album.as_subsonic_album(user)
+        self.assertIsInstance(album_dict, dict)
+        self.assertIn(u'id', album_dict)
+        self.assertIn(u'name', album_dict)
+        self.assertIn(u'artist', album_dict)
+        self.assertIn(u'artistId', album_dict)
+        self.assertIn(u'songCount', album_dict)
+        self.assertIn(u'duration', album_dict)
+        self.assertIn(u'created', album_dict)
+        self.assertIn(u'starred', album_dict)
+        self.assertEqual(album_dict[u'name'], album.name)
+        self.assertEqual(album_dict[u'artist'], artist.name)
+        self.assertEqual(album_dict[u'artistId'], str(artist.id))
+        self.assertEqual(album_dict[u'songCount'], 2)
+        self.assertEqual(album_dict[u'duration'], 8)
+        self.assertRegexpMatches(album_dict[u'created'], date_regex)
+        self.assertRegexpMatches(album_dict[u'starred'], date_regex)
+
+    def test_track(self):
+        track1, track2 = self.create_some_tracks()
+        # TODO
+
+if __name__ == '__main__':
+    unittest.main()
+