database handling works perfectly nowHEADmaster

1 files changed, 30 insertions, 0 deletions

diff --git a/MobiReader/test/tests/SqlConnectionJUnitTest.java b/MobiReader/test/tests/SqlConnectionJUnitTest.java
index 4176cc8..89b6365 100644
--- a/MobiReader/test/tests/SqlConnectionJUnitTest.java
+++ b/MobiReader/test/tests/SqlConnectionJUnitTest.java
@@ -67,6 +67,7 @@ public class SqlConnectionJUnitTest {
         assertEquals(path, bookFromDB.getPathToContent());
     }
     
+    
     @Test
     public void testAddingFewBooksOfSameAuthorToDb() throws ClassNotFoundException
     {
@@ -92,6 +93,34 @@ public class SqlConnectionJUnitTest {
     }       
     
     @Test
+    public void testGettingAllBooksFromDb() throws ClassNotFoundException
+    {
+        String title1 = "Title1";
+        String title2 = "Title2";
+        String title3 = "Title3";
+        String authorName1 = "Gepetto";
+        String authorName2 = "Geronimo";
+        Author gepetto = new Author(authorName1);
+        Author geronimo = new Author(authorName2);
+        String path = "/somepath/book.txt";
+        ArrayList<Book> books = new ArrayList<>();
+        books.add(new Book(title1, gepetto, path));
+        books.add(new Book(title2, gepetto, path));
+        books.add(new Book(title3, gepetto, path));
+        books.add(new Book(title1, geronimo, path));
+        books.add(new Book(title2, geronimo, path));
+        books.add(new Book(title3, geronimo, path));
+        handler.addBooks(books);
+        ArrayList<Book> booksFromDB = handler.getAllBooks();
+        assertEquals(books.size(), booksFromDB.size());
+        for(int i = 0; i < books.size(); i++ )
+        {
+            assertEquals(books.get(i).getTitle(), booksFromDB.get(i).getTitle());
+            assertEquals(books.get(i).getAuthor().getName(), booksFromDB.get(i).getAuthor().getName());
+        }
+    }
+     
+    @Test
     public void testAddingAuthorToDb() throws Exception
     {
         String authorName = "Geronimo";
@@ -103,6 +132,7 @@ public class SqlConnectionJUnitTest {
         assertEquals(additionalInfo, authorFromDB.getAdditionalInfo());
     }
     
+    
     @Test
     public void testAddingTheSameAuthorToDbFewTimes() throws Exception
     {