Catch exception related to search functionality

2 files changed, 11 insertions, 3 deletions

diff --git a/src/main/java/musichub/business/MusicHub.java b/src/main/java/musichub/business/MusicHub.java
index bbac505..660f0f8 100644
--- a/src/main/java/musichub/business/MusicHub.java
+++ b/src/main/java/musichub/business/MusicHub.java
@@ -340,7 +340,7 @@ public class MusicHub {
 
     }
 
-    public void searchAudioElement() throws UnsupportedAudioFileException, NoAlbumFoundException, LineUnavailableException, IOException {
+    public void searchAudioElement() throws UnsupportedAudioFileException, NoAlbumFoundException, LineUnavailableException, IOException, NoElementFoundException {
         Scanner scanner = new Scanner(System.in);
         System.out.println("Entrez le titre ou l'artiste de la musique que vous souhaitez chercher dans la base de données");
         String word = scanner.next().toLowerCase(Locale.ROOT);
@@ -353,7 +353,7 @@ public class MusicHub {
         }
 
         if(searchResult.isEmpty()){
-            System.err.println("Aucun résultat pour votre recherche");
+            throw new NoElementFoundException("Any result for your search");
         }
         if (searchResult.size()==1){
             this.getAudioElement(searchResult, searchResult.get(0).getTitle());
diff --git a/src/main/java/musichub/main/Main.java b/src/main/java/musichub/main/Main.java
index cd1fd5f..b0f98f4 100644
--- a/src/main/java/musichub/main/Main.java
+++ b/src/main/java/musichub/main/Main.java
@@ -246,7 +246,15 @@ public class Main {
                     break;
                 case 'r':
                     //search a music
-                    theHub.searchAudioElement();
+                    try {
+                        theHub.searchAudioElement();
+                    }
+                    catch (NoElementFoundException e){
+                        System.err.println(e.getMessage());
+                    }
+                    catch (java.io.FileNotFoundException e){
+                        System.err.println(e.getMessage()+" Please create a file with the extension .wav inside the song folder at root of the app");
+                    }
                     printAvailableCommands();
                 case 'o':
                     //consult the app logs