Merge pull request #39 from muhammad7865/feature/testing-implementation

muhammad7865 · web-flow · commit fc4d9dbd2a30 · 2026-02-18T00:10:00.000+05:00
Malware_Testing
diff --git a/app/build.gradle.kts b/app/build.gradle.kts
@@ -58,6 +58,12 @@ android {
         buildConfig = true  // For API keys
     }
 
+    testOptions {
+        unitTests {
+            isReturnDefaultValues = true
+        }
+    }
+
 }
 
 dependencies {
@@ -110,6 +116,10 @@ dependencies {
     implementation(libs.androidx.ui.tooling.preview)
     implementation(libs.androidx.material3)
     testImplementation(libs.junit)
+    testImplementation(libs.androidx.test.core)
+    testImplementation(libs.mockk)
+    testImplementation(libs.kotlinx.coroutines.test)
+    testImplementation(libs.robolectric)
     androidTestImplementation(libs.androidx.junit)
     androidTestImplementation(libs.androidx.espresso.core)
     androidTestImplementation(platform(libs.androidx.compose.bom))
diff --git a/app/src/main/java/com/droid/cybershield/core/features/OnnxScanner.kt b/app/src/main/java/com/droid/cybershield/core/features/OnnxScanner.kt
@@ -13,7 +13,7 @@ class OnnxScanner private constructor(
     private val env: OrtEnvironment,
     private val session: OrtSession,
     private val order: List<String>
-) {
+) : MalwareScanner {
     companion object {
         private const val TAG = "OnnxScanner"
         @Volatile
@@ -55,7 +55,7 @@ class OnnxScanner private constructor(
         }
     }
 
-    suspend fun predict(payload: Map<String, Any>): Float {
+    override suspend fun predict(payload: Map<String, Any>): Float {
 
         return try {
             val n = order.size
diff --git a/app/src/main/java/com/droid/cybershield/core/inference/MalwareScanner.kt b/app/src/main/java/com/droid/cybershield/core/inference/MalwareScanner.kt
@@ -0,0 +1,5 @@
+package com.droid.cybershield.core.inference
+
+interface MalwareScanner {
+    suspend fun predict(payload: Map<String, Any>): Float
+}
diff --git a/app/src/main/java/com/droid/cybershield/data/repository/MalwareRepositoryImpl.kt b/app/src/main/java/com/droid/cybershield/data/repository/MalwareRepositoryImpl.kt
@@ -20,7 +20,9 @@ import com.droid.cybershield.core.features.SafeAppWhitelist
 import javax.inject.Inject
 
 class MalwareRepositoryImpl @Inject constructor(
-    private val malwareDao: MalwareDao
+    private val malwareDao: MalwareDao,
+    private val scanner: com.droid.cybershield.core.inference.MalwareScanner,
+    private val scanHistoryManager: ScanHistoryManager
 ) : MalwareRepository {
     override suspend fun scanInstalledApps(
         context: Context,
@@ -36,7 +38,7 @@ class MalwareRepositoryImpl @Inject constructor(
 
         onProgress?.invoke(MalwareRepository.Progress(MalwareRepository.Progress.Phase.Extracting, 100))
 
-        val scanner = OnnxScanner.get(context)
+        // Scanner is injected
 
         val total = apps.size
         var processed = 0
@@ -204,7 +206,7 @@ class MalwareRepositoryImpl @Inject constructor(
             appResults = appResults
         )
 
-        ScanHistoryManager(context).saveScanReport(
+        scanHistoryManager.saveScanReport(
             com.droid.cybershield.data.model.ScanReport(
                 id = report.id,
                 timestamp = report.timestamp,
@@ -232,7 +234,7 @@ class MalwareRepositoryImpl @Inject constructor(
     }
 
     override fun getHistory(context: Context): List<ScanReport> {
-        val history = ScanHistoryManager(context).getScanHistory()
+        val history = scanHistoryManager.getScanHistory()
         return history.map { report ->
             ScanReport(
                 id = report.id,
diff --git a/app/src/main/java/com/droid/cybershield/di/AppModule.kt b/app/src/main/java/com/droid/cybershield/di/AppModule.kt
@@ -8,6 +8,7 @@ import com.droid.cybershield.data.local.dao.PhishingDao
 import com.droid.cybershield.data.remote.api.SafeBrowsingApi
 import com.droid.cybershield.data.remote.client.SafeBrowsingClient
 import com.droid.cybershield.data.repository.MalwareRepositoryImpl
+import com.droid.cybershield.data.model.ScanHistoryManager
 import com.droid.cybershield.data.repository.PhishingRepositoryImpl
 import com.droid.cybershield.domain.repository.MalwareRepository
 import com.droid.cybershield.domain.repository.PhishingRepository
@@ -31,12 +32,26 @@ import javax.inject.Singleton
 object AppModule {
     @Provides
     @Singleton
-    fun provideMalwareRepository(malwareDao: MalwareDao): MalwareRepository = MalwareRepositoryImpl(malwareDao)
+    fun provideScanHistoryManager(@ApplicationContext context: Context) = ScanHistoryManager(context)
+
+    @Provides
+    @Singleton
+    fun provideMalwareRepository(
+        malwareDao: MalwareDao,
+        scanner: com.droid.cybershield.core.inference.MalwareScanner,
+        scanHistoryManager: ScanHistoryManager
+    ): MalwareRepository = MalwareRepositoryImpl(malwareDao, scanner, scanHistoryManager)
 
     @Provides
     @Singleton
     fun provideScanInstalledApps(repo: MalwareRepository) = ScanInstalledApps(repo)
 
+    @Provides
+    @Singleton
+    fun provideMalwareScanner(@ApplicationContext context: Context): com.droid.cybershield.core.inference.MalwareScanner {
+        return com.droid.cybershield.core.inference.OnnxScanner.get(context)
+    }
+
     @Provides
     @Singleton
     fun provideGetHistory(repo: MalwareRepository) = GetHistory(repo)
diff --git a/app/src/test/java/com/droid/cybershield/ExampleUnitTest.kt b/app/src/test/java/com/droid/cybershield/ExampleUnitTest.kt
diff --git a/app/src/test/java/com/droid/cybershield/data/local/dao/MalwareDaoTest.kt b/app/src/test/java/com/droid/cybershield/data/local/dao/MalwareDaoTest.kt
@@ -0,0 +1,61 @@
+package com.droid.cybershield.data.local.dao
+
+import android.content.Context
+import androidx.room.Room
+import androidx.test.core.app.ApplicationProvider
+import com.droid.cybershield.data.local.CyberShieldDatabase
+import com.droid.cybershield.data.local.entity.MalwareCacheEntity
+import kotlinx.coroutines.test.runTest
+import org.junit.After
+import org.junit.Assert.assertEquals
+import org.junit.Assert.assertNotNull
+import org.junit.Before
+import org.junit.Test
+import org.junit.runner.RunWith
+import org.robolectric.RobolectricTestRunner
+import org.robolectric.annotation.Config
+
+@RunWith(RobolectricTestRunner::class)
+@Config(sdk = [33]) // Use a recent SDK
+class MalwareDaoTest {
+
+    private lateinit var database: CyberShieldDatabase
+    private lateinit var dao: MalwareDao
+
+    @Before
+    fun setup() {
+        val context = ApplicationProvider.getApplicationContext<Context>()
+        // In-memory database for testing
+        database = Room.inMemoryDatabaseBuilder(context, CyberShieldDatabase::class.java)
+            .allowMainThreadQueries() // Allowed for testing
+            .build()
+        dao = database.malwareDao()
+    }
+
+    @After
+    fun tearDown() {
+        database.close()
+    }
+
+    @Test
+    fun insertAndGetCache() = runTest {
+        // Arrange
+        val entity = MalwareCacheEntity(
+            packageName = "com.test.app",
+            isSafe = true,
+            timestamp = 123456789L,
+            versionCode = 1,
+            predictionScore = 0.1f
+        )
+
+        // Act
+        dao.insertCache(entity)
+        val retrieved = dao.getCache("com.test.app")
+
+        // Assert
+        assertNotNull(retrieved)
+        assertEquals(entity.packageName, retrieved?.packageName)
+        assertEquals(entity.isSafe, retrieved?.isSafe)
+        assertEquals(entity.predictionScore, retrieved?.predictionScore)
+    }
+}
diff --git a/app/src/test/java/com/droid/cybershield/data/repository/MalwareRepositoryStressTest.kt b/app/src/test/java/com/droid/cybershield/data/repository/MalwareRepositoryStressTest.kt
@@ -0,0 +1,61 @@
+package com.droid.cybershield.data.repository
+
+import android.content.Context
+import android.content.pm.ApplicationInfo
+import android.content.pm.PackageManager
+import com.droid.cybershield.core.inference.MalwareScanner
+import com.droid.cybershield.data.local.dao.MalwareDao
+import com.droid.cybershield.data.model.ScanHistoryManager
+import io.mockk.coEvery
+import io.mockk.every
+import io.mockk.mockk
+import kotlinx.coroutines.test.runTest
+import org.junit.Assert.assertTrue
+import org.junit.Test
+
+class MalwareRepositoryStressTest {
+
+    private val malwareDao: MalwareDao = mockk(relaxed = true)
+    private val scanner: MalwareScanner = mockk()
+    private val scanHistoryManager: ScanHistoryManager = mockk(relaxed = true)
+    private val context: Context = mockk()
+    private val packageManager: PackageManager = mockk()
+
+    @Test
+    fun `stress test 1000 apps performance`() = runTest {
+        // Arrange
+        val repository = MalwareRepositoryImpl(malwareDao, scanner, scanHistoryManager)
+        every { context.packageManager } returns packageManager
+        every { context.packageName } returns "com.test.app" // Exclude self
+
+        // Generate 1000 fake apps
+        val apps = (1..1000).map { i ->
+            ApplicationInfo().apply {
+                packageName = "com.fake.app.$i"
+                flags = 0 // Not system app
+                sourceDir = "/data/app/com.fake.app.$i.apk"
+            }
+        }
+        every { packageManager.getInstalledApplications(0) } returns apps
+        every { packageManager.getApplicationLabel(any()) } returns "Fake App"
+        // Mock getPackageInfo to throw exception (simulating app not found or simple path) 
+        // OR mock it properly to return permissions. 
+        // Let's mock it to return null to hit the catch block and use default logic (faster setup)
+        // or return a basic PackageInfo.
+        every { packageManager.getPackageInfo(any<String>(), any<Int>()) } throws RuntimeException("Skip detail")
+
+        // Mock Scanner to be fast
+        coEvery { scanner.predict(any()) } returns 0.1f
+
+        // Act
+        val startTime = System.currentTimeMillis()
+        val result = repository.scanInstalledApps(context)
+        val endTime = System.currentTimeMillis()
+        val duration = endTime - startTime
+
+        // Assert
+        println("Scanned 1000 apps in ${duration}ms")
+        assertTrue("Scanning 1000 apps took too long: ${duration}ms", duration < 2000)
+        assertTrue(result.totalApps >= 1000)
+    }
+}
diff --git a/app/src/test/java/com/droid/cybershield/data/repository/MalwareRepositoryTest.kt b/app/src/test/java/com/droid/cybershield/data/repository/MalwareRepositoryTest.kt
@@ -0,0 +1,91 @@
+package com.droid.cybershield.data.repository
+
+import com.droid.cybershield.core.features.FeatureExtractor
+import com.droid.cybershield.core.inference.MalwareScanner
+import com.droid.cybershield.data.local.dao.MalwareDao
+import com.droid.cybershield.data.local.entity.MalwareCacheEntity
+import com.droid.cybershield.data.model.ScanHistoryManager
+import io.mockk.coEvery
+import io.mockk.mockk
+import kotlinx.coroutines.test.runTest
+import org.junit.Assert.assertEquals
+import org.junit.Before
+import org.junit.Test
+import android.content.Context
+import android.content.pm.PackageManager
+import android.content.pm.ApplicationInfo
+import android.content.pm.PackageInfo
+import io.mockk.every
+import io.mockk.mockkStatic
+import io.mockk.unmockkAll
+
+class MalwareRepositoryTest {
+
+    private lateinit var repository: MalwareRepositoryImpl
+    private val malwareDao: MalwareDao = mockk()
+    private val context: Context = mockk()
+    private val packageManager: PackageManager = mockk()
+    private val scanner: MalwareScanner = mockk()
+    private val scanHistoryManager: ScanHistoryManager = mockk(relaxed = true)
+
+    @Before
+    fun setup() {
+        repository = MalwareRepositoryImpl(malwareDao, scanner, scanHistoryManager)
+        every { context.packageManager } returns packageManager
+        every { context.packageName } returns "com.test.app"
+    }
+
+    @Test
+    fun `scanInstalledApps should skip Whitelisted apps`() = runTest {
+        // Arrange
+        val app = ApplicationInfo().apply {
+            packageName = "com.whatsapp"
+            flags = 0
+            sourceDir = "/data/app/com.whatsapp.apk"
+        }
+        every { packageManager.getInstalledApplications(0) } returns listOf(app)
+        every { packageManager.getApplicationLabel(app) } returns "WhatsApp"
+
+        // Act
+        val result = repository.scanInstalledApps(context)
+
+        // Assert
+        assertEquals(0, result.maliciousApps)
+        assertEquals(1, result.benignApps) // WhatsApp is benign
+    }
+    
+    @Test
+    fun `scanInstalledApps should use Cache when version matches`() = runTest {
+         // Arrange
+        val app = ApplicationInfo().apply {
+            packageName = "com.unknown.app"
+            flags = 0
+            sourceDir = "/data/app/com.unknown.app.apk"
+        }
+        val packageInfo = PackageInfo().apply {
+            versionCode = 100
+        }
+        
+        every { packageManager.getInstalledApplications(0) } returns listOf(app)
+        every { packageManager.getApplicationLabel(app) } returns "Unknown App"
+        every { packageManager.getPackageInfo("com.unknown.app", 0) } returns packageInfo
+        
+        // Cache says it is SAFE
+        coEvery { malwareDao.getCache("com.unknown.app") } returns MalwareCacheEntity(
+            packageName = "com.unknown.app",
+            isSafe = true,
+            timestamp = System.currentTimeMillis(),
+            versionCode = 100,
+            predictionScore = 0.1f
+        )
+
+        // Act
+        val result = repository.scanInstalledApps(context)
+
+        // Assert
+        assertEquals(0, result.maliciousApps)
+        assertEquals(1, result.benignApps)
+        // Verify we NEVER called the AI model
+        coEvery { scanner.predict(any()) } answers { throw IllegalStateException("AI should not be called!") }
+    }
+}
diff --git a/app/src/test/java/com/droid/cybershield/presentation/malware/MalwareViewModelTest.kt b/app/src/test/java/com/droid/cybershield/presentation/malware/MalwareViewModelTest.kt
diff --git a/app/src/test/java/com/droid/cybershield/rules/MainDispatcherRule.kt b/app/src/test/java/com/droid/cybershield/rules/MainDispatcherRule.kt
diff --git a/gradle/libs.versions.toml b/gradle/libs.versions.toml
