From 69982e9db84572c290259fa51fb37a20ae97d253 Mon Sep 17 00:00:00 2001
From: MohitMali <futuretechnoworld.mohit@gmail.com>
Date: Fri, 28 Jul 2023 15:39:21 +0530
Subject: [PATCH] Fixed ZimMangeViewModelTest

---
 .../kiwixmobile/zimManager/ZimManageViewModelTest.kt | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/app/src/test/java/org/kiwix/kiwixmobile/zimManager/ZimManageViewModelTest.kt b/app/src/test/java/org/kiwix/kiwixmobile/zimManager/ZimManageViewModelTest.kt
index aa812e9f6..68f3f3a6f 100644
--- a/app/src/test/java/org/kiwix/kiwixmobile/zimManager/ZimManageViewModelTest.kt
+++ b/app/src/test/java/org/kiwix/kiwixmobile/zimManager/ZimManageViewModelTest.kt
@@ -20,8 +20,8 @@ package org.kiwix.kiwixmobile.zimManager
 
 import android.app.Application
 import android.net.ConnectivityManager
-import android.net.ConnectivityManager.TYPE_WIFI
-import android.net.NetworkInfo
+import android.net.NetworkCapabilities
+import android.net.NetworkCapabilities.TRANSPORT_WIFI
 import com.jraska.livedata.test
 import io.mockk.clearAllMocks
 import io.mockk.every
@@ -97,7 +97,7 @@ class ZimManageViewModelTest {
   private val connectivityManager: ConnectivityManager = mockk()
 
   @Suppress("DEPRECATION")
-  private val networkInfo: NetworkInfo = mockk()
+  private val networkCapabilities: NetworkCapabilities = mockk()
   private val sharedPreferenceUtil: SharedPreferenceUtil = mockk()
   lateinit var viewModel: ZimManageViewModel
 
@@ -134,8 +134,10 @@ class ZimManageViewModelTest {
     every { connectivityBroadcastReceiver.networkStates } returns networkStates
     every { application.registerReceiver(any(), any()) } returns mockk()
     every { dataSource.booksOnDiskAsListItems() } returns booksOnDiskListItems
-    every { connectivityManager.getNetworkInfo(TYPE_WIFI) } returns networkInfo
-    every { networkInfo.isConnected } returns true
+    every {
+      connectivityManager.getNetworkCapabilities(connectivityManager.activeNetwork)
+    } returns networkCapabilities
+    every { networkCapabilities.hasTransport(TRANSPORT_WIFI) } returns true
     viewModel = ZimManageViewModel(
       downloadDao,
       newBookDao,