diff --git a/src/com/twu/biblioteca/Biblioteca.java b/src/com/twu/biblioteca/Biblioteca.java index f6e35f5..521c3f3 100644 --- a/src/com/twu/biblioteca/Biblioteca.java +++ b/src/com/twu/biblioteca/Biblioteca.java @@ -11,11 +11,12 @@ public class Biblioteca { List books = new ArrayList(); public Biblioteca() { + initialiazeBookList(); } public Biblioteca(String welcomeMessage) { - this.welcomeMessage = welcomeMessage; + initialiazeBookList(); } public String welcomeUser() { @@ -23,9 +24,11 @@ public String welcomeUser() { } public List listBooks() { - List bookList = new ArrayList(); + return books; + } + + private void initialiazeBookList() { Book book = new Book("Harry Potter"); - bookList.add(book); - return bookList; + books.add(book); } } diff --git a/src/com/twu/biblioteca/BibliotecaApp.java b/src/com/twu/biblioteca/BibliotecaApp.java index 43df94a..a87fe87 100755 --- a/src/com/twu/biblioteca/BibliotecaApp.java +++ b/src/com/twu/biblioteca/BibliotecaApp.java @@ -5,5 +5,9 @@ public class BibliotecaApp { public static void main(String[] args) { Biblioteca biblioteca = new Biblioteca(); System.out.println(biblioteca.welcomeUser()); + + for(Book book : biblioteca.listBooks()) { + System.out.println(book.name + "\n"); + } } }