2 files changed, 137 insertions, 0 deletions
diff --git a/src/test/java/musichub/business/PlayListTest.java b/src/test/java/musichub/business/PlayListTest.java
new file mode 100644
index 0000000..9fe63cf
--- /dev/null
+++ b/src/test/java/musichub/business/PlayListTest.java
@@ -0,0 +1,54 @@
+package musichub.business;
+
+import musichub.util.XMLHandler;
+import org.junit.jupiter.api.Test;
+import org.w3c.dom.Document;
+import org.w3c.dom.Element;
+import org.w3c.dom.NodeList;
+
+import java.util.ArrayList;
+import java.util.UUID;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+public class PlayListTest {
+    final String DIR = System.getProperty("user.dir");
+    final String PLAYLISTS_FILE_PATH = DIR + "/files/playlists.xml";
+    final XMLHandler xmlHandler = new XMLHandler();
+    String title = "Side To Side";
+    String id = "66d277ca-cbc4-471c-a07e-082363375bcc";
+    ArrayList<UUID> elementUUIDs = null;
+
+    @Test
+    void testPlaylistClass() {
+        new PlayList(title, id, elementUUIDs);
+        new PlayList(title);
+
+    }
+
+    @Test
+    void testPlaylistClassXML() {
+        NodeList playlistNodes = xmlHandler.parseXMLFile(PLAYLISTS_FILE_PATH);
+        Element playlistElement = (Element) playlistNodes.item(1);
+        new PlayList(playlistElement);
+    }
+
+    @Test
+    void testGetElements() {
+        assertNull(new PlayList(title, id, elementUUIDs).getElements());
+    }
+
+    @Test
+    void testGetTitle() {
+        assertEquals(new PlayList(title).getTitle(), "Side To Side");
+        assertNotEquals(new PlayList(title).getTitle(), "God Is A Woman");
+    }
+
+    @Test
+    void testCreateXMLElement() {
+        PlayList p = new PlayList(title);
+        Document document = xmlHandler.createXMLDocument();
+        Element root = document.createElement("playlists");
+        p.createXMLElement(document, root);
+    }
+}
diff --git a/src/test/java/musichub/business/SongTest.java b/src/test/java/musichub/business/SongTest.java
new file mode 100644
index 0000000..14f9af1
--- /dev/null
+++ b/src/test/java/musichub/business/SongTest.java
@@ -0,0 +1,83 @@
+package musichub.business;
+
+import musichub.util.XMLHandler;
+import org.junit.jupiter.api.Test;
+import org.w3c.dom.Document;
+import org.w3c.dom.Element;
+import org.w3c.dom.NodeList;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertNotEquals;
+
+public class SongTest {
+    String title = "Side To Side";
+    String artist = "Ariana Grande";
+    int length = 186;
+    String uid = "66d277ca-cbc4-471c-a07e-082363375bcc";
+    String content = "Song/Side_To_Side.wav";
+    String genre = "rock";
+
+    final String DIR = System.getProperty("user.dir");
+    final String ELEMENTS_FILE_PATH = DIR + "/files/elements.xml";
+    final XMLHandler xmlHandler = new XMLHandler();
+
+
+    @Test
+    void testSongClass() {
+        new Song(title, artist, length, uid, content, genre);
+        new Song(title, artist, length, content, genre);
+
+    }
+
+    @Test
+    void testSongClassXML() {
+        NodeList audioelementsNodes = xmlHandler.parseXMLFile(ELEMENTS_FILE_PATH);
+        Element audioElement = (Element) audioelementsNodes.item(1);
+        new Song(audioElement);
+    }
+
+    @Test
+    void testGetGenre() {
+        assertEquals(new Song(title, artist, length, content, genre)
+                        .getGenre(),
+                "rock");
+        assertNotEquals(new Song(title, artist, length, content, genre)
+                        .getGenre(),
+                "pop");
+    }
+
+    @Test
+    void testSetGenre() {
+        new Song(title, artist, length, content, "classic");
+        new Song(title, artist, length, content, "hiphop");
+        new Song(title, artist, length, content, "rock");
+        new Song(title, artist, length, content, "pop");
+        new Song(title, artist, length, content, "rap");
+
+        Song s = new Song(title, artist, length, content, "cgfdhdfhj");
+        assertEquals(s.getGenre(), "jazz");
+    }
+
+    @Test
+    void testToString() {
+        assertEquals(
+                new Song(title, artist, length, content, genre)
+                        .toString(),
+                "Title = Side To Side, Artist = Ariana Grande, Length = 186, Content = Song/Side_To_Side.wav, Genre = rock\n"
+        );
+        assertNotEquals(
+                new Song(title, artist, length, content, genre)
+                        .toString(),
+                "Title = God is a woman, Artist = Ariana Grande, Length = 186, Content = Song/Side_To_Side.wav, Genre = rock\n"
+        );
+    }
+
+    @Test
+    void testCreateXMLElement() {
+        Song s = new Song(title, artist, length, content, genre);
+        Document document = xmlHandler.createXMLDocument();
+        Element root = document.createElement("elements");
+        s.createXMLElement(document, root);
+
+    }
+}