From 68ab7ceebd901cccbc40240bc5981feb33f5612d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Karol=20G=C3=B3rzy=C5=84ski?= <kargor8@st.amu.edu.pl>
Date: Fri, 8 Jan 2021 17:38:05 +0100
Subject: [PATCH] SES-128 Fix Error response

---
 .../Controllers/WeaponController.cs                 | 13 +++++++++----
 1 file changed, 9 insertions(+), 4 deletions(-)

diff --git a/SessionCompanion/SessionCompanion/Controllers/WeaponController.cs b/SessionCompanion/SessionCompanion/Controllers/WeaponController.cs
index 69bbd60..1dada82 100644
--- a/SessionCompanion/SessionCompanion/Controllers/WeaponController.cs
+++ b/SessionCompanion/SessionCompanion/Controllers/WeaponController.cs
@@ -25,14 +25,19 @@ namespace SessionCompanion.Controllers
         [HttpGet("getAllWeapons")]
         public async Task<Either<List<WeaponViewModel>, ErrorResponse>> GetWeapons() 
         {
-            var weapons = _service.Get().ToList();
-            if (!weapons.Any())
+            try
+            {
+                var weapons = _service.Get().ToList();
+                return weapons;
+            }
+            catch (Exception e)
+            {
                 return new ErrorResponse()
                 {
                     StatusCode = 204,
-                    Message = "No weapons in database"
+                    Message = e.Message
                 };
-            return weapons;
+            }
         }
     }
 }