From e4ccd8b767e810a7c3f07379d298abb83798e7fd Mon Sep 17 00:00:00 2001
From: hugovk <hugovk@users.noreply.github.com>
Date: Sat, 5 Apr 2014 00:30:33 +0300
Subject: [PATCH] More tests. Don't bother with failing Py2.6 or 3.2, they're
 unsupported.

---
 .travis.yml    |  4 ---
 test_pylast.py | 87 ++++++++++++++++++++++++++++++++++++++++++++++++--
 2 files changed, 85 insertions(+), 6 deletions(-)

diff --git a/.travis.yml b/.travis.yml
index 1b72dac..c853ea3 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -1,8 +1,6 @@
 language: python
 python:
-- "2.6"
 - "2.7"
-- "3.2"
 - "3.3"
 - "3.4"
 - "pypy"
@@ -28,7 +26,5 @@ env:
 
 matrix:
   allow_failures:
-    - python: "2.6"
-    - python: "3.2"
     - python: "3.4"
     - python: "pypy"
diff --git a/test_pylast.py b/test_pylast.py
index a2cfc24..17d5bad 100755
--- a/test_pylast.py
+++ b/test_pylast.py
@@ -237,7 +237,7 @@ class TestPyLast(unittest.TestCase):
 
     def test_unlove(self):
         # Arrange
-        artist = "Test Artist"
+        artist = pylast.Artist("Test Artist", self.network)
         title = "Test Title"
         track = pylast.Track(artist, title, self.network)
         lastfm_user = self.network.get_user(self.username)
@@ -1594,7 +1594,6 @@ class TestPyLast(unittest.TestCase):
                 break
         self.assertFalse(found)
 
-
     def test_remove_tags(self):
         # Arrange
         tags = ["testing1", "testing2"]
@@ -1613,6 +1612,90 @@ class TestPyLast(unittest.TestCase):
                 break
         self.assertFalse(found)
 
+    def test_set_tags(self):
+        # Arrange
+        tags = ["settag1", "settag2"]
+        artist = self.network.get_artist("Test Artist")
+        artist.add_tags(tags)
+        artist.add_tags("1more")
+        tags_before = artist.get_tags()
+
+        # Act
+        artist.remove_tags(tags)
+
+        # Assert
+        tags_after = artist.get_tags()
+        self.assertEqual(len(tags_after), len(tags_before) - 2)
+        found1, found2 = False, False
+        for tag in tags:
+            if tag.name == "settag1":
+                found1 = True
+            elif tag.name == "settag2":
+                found2 = True
+        self.assertTrue(found1)
+        self.assertTrue(found2)
+
+    def test_tracks_notequal(self):
+        # Arrange
+        track1 = pylast.Track("Test Artist", "Test Title", self.network)
+        track2 = pylast.Track("Test Artist", "Test Track", self.network)
+
+        # Act
+        # Assert
+        self.assertNotEqual(track1, track2)
+
+    def test_track_id(self):
+        # Arrange
+        track = pylast.Track("Test Artist", "Test Title", self.network)
+
+        # Act
+        id = track.get_id()
+
+        # Assert
+        self.assertEqual(id, "14053327")
+
+    def test_track_title_prop_caps(self):
+        # Arrange
+        track = pylast.Track("test artist", "test title", self.network)
+
+        # Act
+        title = track.get_title(properly_capitalized=True)
+
+        # Assert
+        self.assertEqual(title, "Test Title")
+
+    def test_track_listener_count(self):
+        # Arrange
+        track = pylast.Track("test artist", "test title", self.network)
+
+        # Act
+        count = track.get_listener_count()
+
+        # Assert
+        self.assertGreater(count, 21)
+
+    def test_album_rel_date(self):
+        # Arrange
+        album = pylast.Album("Test Artist", "Test Release", self.network)
+
+        # Act
+        date = album.get_release_date()
+
+        # Assert
+        self.assertIn("2011", date)
+
+    def test_album_tracks(self):
+        # Arrange
+        album = pylast.Album("Test Artist", "Test Release", self.network)
+
+        # Act
+        tracks = album.get_tracks()
+
+        # Assert
+        self.assertIsInstance(tracks, list)
+        self.assertIsInstance(tracks[0], pylast.Track)
+        self.assertEqual(len(tracks), 4)
+