View Models cleanup

ShivikaGL · May 21, 2020 · a80e1c1 · a80e1c1
1 parent f0f636a
commit a80e1c1
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/MockingProject/Modules/DependencyInjection/Controllers/ServiceLocatorController.swift b/MockingProject/Modules/DependencyInjection/Controllers/ServiceLocatorController.swift
@@ -23,7 +23,10 @@ class ServiceLocatorController: UIViewController {
 
         let serviceLocator = DIServiceLocator.shared
 
-        guard let model: HomeViewModel = serviceLocator.resolve() else { return }
+        guard let model: HomeViewModel = serviceLocator.resolve() else {
+            print("Couldnt resolve model")
+
+            return }
         self.viewModel = model
 
 		showLoader()

diff --git a/MockingProject/Modules/DependencyInjection/ViewModel/SwinjectViewModel.swift b/MockingProject/Modules/DependencyInjection/ViewModel/SwinjectViewModel.swift
@@ -20,7 +20,6 @@ class SwinjectViewModel: HomeViewModelProtocol {
 
     func fetchEmployees(completion: @escaping ([Employee]?, String?) -> Void) {
         self.apiManager!.getEmployees { (result: DataResponse<EmployeesResponse, AFError>) in
-            print(result)
             switch result.result {
             case .success(let response):
                 if response.status == "success" {

diff --git a/MockingProject/SupportingFiles/AppDelegate.swift b/MockingProject/SupportingFiles/AppDelegate.swift
@@ -26,7 +26,7 @@ class AppDelegate: UIResponder, UIApplicationDelegate {
         serviceLocator.register(APIManager() as APIManager)
 
         guard let manager: APIManager = serviceLocator.resolve() else { return true }
-        serviceLocator.register(SwinjectViewModel(manager: manager))
+        serviceLocator.register(HomeViewModel(manager: manager))
 
         return true
     }