# test_songs.py


from songs import Song, SongCollection
import unittest



class SongCollectionTest(unittest.TestCase):
    def setUp(self):
        self._collection = SongCollection()


    def test_new_collections_have_size_zero(self):
        self.assertEqual(self._collection.size(), 0)


    def test_after_adding_one_song_to_a_collection__size_is_1(self):
        self._collection.add(self._create_test_song())
        self.assertEqual(self._collection.size(), 1)


    def _create_test_song(self) -> Song:
        return Song()



if __name__ == '__main__':
    unittest.main()