From 84b82c24620531d6b7ce468e731957c0e89280a6 Mon Sep 17 00:00:00 2001
From: Casey Marshall <casey.marshall@snyk.io>
Date: Wed, 24 Jan 2024 17:42:21 -0600
Subject: [PATCH] chore: add controller impl test

---
 .../extensions/SnykControllerImplTest.kt      | 100 ++++++++++++++++++
 1 file changed, 100 insertions(+)
 create mode 100644 src/test/kotlin/io/snyk/plugin/extensions/SnykControllerImplTest.kt

diff --git a/src/test/kotlin/io/snyk/plugin/extensions/SnykControllerImplTest.kt b/src/test/kotlin/io/snyk/plugin/extensions/SnykControllerImplTest.kt
new file mode 100644
index 000000000..334a9a098
--- /dev/null
+++ b/src/test/kotlin/io/snyk/plugin/extensions/SnykControllerImplTest.kt
@@ -0,0 +1,100 @@
+package io.snyk.plugin.extensions
+
+import com.intellij.openapi.application.ApplicationManager
+import com.intellij.openapi.components.service
+import com.intellij.testFramework.LightPlatformTestCase
+import com.intellij.testFramework.PlatformTestUtil
+import com.intellij.testFramework.replaceService
+import io.mockk.every
+import io.mockk.mockk
+import io.mockk.mockkStatic
+import io.mockk.spyk
+import io.mockk.unmockkAll
+import io.mockk.verify
+import io.snyk.plugin.extensions.SnykControllerImpl
+import io.snyk.plugin.getCliFile
+import io.snyk.plugin.getContainerService
+import io.snyk.plugin.getIacService
+import io.snyk.plugin.getOssService
+import io.snyk.plugin.getSnykCachedResults
+import io.snyk.plugin.getSnykCliDownloaderService
+import io.snyk.plugin.isCliInstalled
+import io.snyk.plugin.isContainerEnabled
+import io.snyk.plugin.isIacEnabled
+import io.snyk.plugin.net.CliConfigSettings
+import io.snyk.plugin.net.LocalCodeEngine
+import io.snyk.plugin.pluginSettings
+import io.snyk.plugin.removeDummyCliFile
+import io.snyk.plugin.resetSettings
+import io.snyk.plugin.services.SnykApiService
+import io.snyk.plugin.services.SnykTaskQueueService
+import io.snyk.plugin.services.download.CliDownloader
+import io.snyk.plugin.services.download.LatestReleaseInfo
+import io.snyk.plugin.services.download.SnykCliDownloaderService
+import io.snyk.plugin.setupDummyCliFile
+import org.awaitility.Awaitility.await
+import snyk.container.ContainerResult
+import snyk.iac.IacResult
+import snyk.oss.OssResult
+import snyk.oss.OssService
+import snyk.trust.confirmScanningAndSetWorkspaceTrustedStateIfNeeded
+import java.util.concurrent.TimeUnit
+
+@Suppress("FunctionName")
+class SnykControllerImplTest : LightPlatformTestCase() {
+
+    private lateinit var ossServiceMock: OssService
+    private lateinit var downloaderServiceMock: SnykCliDownloaderService
+
+    override fun setUp() {
+        super.setUp()
+        unmockkAll()
+        resetSettings(project)
+        ossServiceMock = mockk(relaxed = true)
+        project.replaceService(OssService::class.java, ossServiceMock, project)
+        mockkStatic("io.snyk.plugin.UtilsKt")
+        mockkStatic("snyk.trust.TrustedProjectsKt")
+        downloaderServiceMock = spyk(SnykCliDownloaderService())
+        every { downloaderServiceMock.requestLatestReleasesInformation() } returns LatestReleaseInfo(
+            "http://testUrl",
+            "testReleaseInfo",
+            "testTag"
+        )
+        every { getSnykCliDownloaderService() } returns downloaderServiceMock
+        every { downloaderServiceMock.isFourDaysPassedSinceLastCheck() } returns false
+        every { confirmScanningAndSetWorkspaceTrustedStateIfNeeded(any(), any()) } returns true
+    }
+
+    override fun tearDown() {
+        unmockkAll()
+        resetSettings(project)
+        removeDummyCliFile()
+        super.tearDown()
+    }
+
+    fun testControllerCanTriggerScan() {
+        mockkStatic("io.snyk.plugin.UtilsKt")
+        every { isCliInstalled() } returns true
+        val fakeResult = OssResult(emptyList())
+        every { getOssService(project)?.scan() } returns fakeResult
+
+        val snykTaskQueueService = project.service<SnykTaskQueueService>()
+        val settings = pluginSettings()
+        settings.ossScanEnable = true
+        settings.snykCodeSecurityIssuesScanEnable = false
+        settings.snykCodeQualityIssuesScanEnable = false
+        settings.iacScanEnabled = false
+        settings.containerScanEnabled = false
+
+        getSnykCachedResults(project)?.currentContainerResult = null
+
+        val controller = SnykControllerImpl(project)
+        controller.scan()
+
+        PlatformTestUtil.dispatchAllInvocationEventsInIdeEventQueue()
+
+        await().atMost(2, TimeUnit.SECONDS).until {
+            getSnykCachedResults(project)?.currentOssResults != null
+        }
+    }
+}