- Testsuite expansion

app/src/main/java/com/appttude/h_mal/atlas_weather/viewmodel/MainViewModel.kt

@@ -48,7 +48,6 @@ class MainViewModel(
                 // Save data if not null
                 repository.saveLastSavedAt(CURRENT_LOCATION)
                 repository.saveCurrentWeatherToRoom(entityItem)
-                onSuccess(Unit)
             } catch (e: Exception) {
                 onError(e.message!!)
             }

app/src/main/java/com/appttude/h_mal/atlas_weather/viewmodel/WorldViewModel.kt

@@ -52,6 +52,7 @@ class WorldViewModel(
                 } else {
                     repository.getSingleWeather(locationName)
                 }
+                onSuccess(Unit)
                 repository.saveCurrentWeatherToRoom(weatherEntity)
                 repository.saveLastSavedAt(weatherEntity.id)
             } catch (e: IOException) {