MDS/openide
1
0
Fork 0
mirror of https://gitflic.ru/project/openide/openide.git synced 2026-09-27 10:03:11 +07:00
Code Issues Packages Projects Releases Wiki Activity

testFramework [python]: Fix mockPathAndAdd.

Browse Source 
mokk doesn't work in devbuild tests, so we mock env map explicitly


(cherry picked from commit 01d8ec764fff1ae8d92abe86e455ebbbb2a329f8)

IJ-CR-220875

GitOrigin-RevId: e955846cb747a1fde6321cc5e9a6b17d34aa310b
This commit is contained in:
Ilya.Kazakevich
2026-08-29 15:34:58 +00:00
committed by intellij-monorepo-bot
parent 817aded11a
commit fa13e71a30
1 changed files with 18 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/junit5Tests-framework/src/com/intellij/python/junit5Tests/framework/EnvironmentVariablesPathMock.kt
+18 -12
View File
@@ -1,13 +1,11 @@
// Copyright 2000-2025 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
// Copyright 2000-2026 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.
package com.intellij.python.junit5Tests.framework
import com.intellij.util.EnvironmentUtil
import io.mockk.every
import io.mockk.mockkStatic
import io.mockk.unmockkStatic
import uk.org.webcompere.systemstubs.environment.EnvironmentVariables
import java.io.File
import java.nio.file.Path
import java.util.function.Supplier
import kotlin.io.path.pathString
/**
@@ -19,22 +17,30 @@ import kotlin.io.path.pathString
*/
fun EnvironmentVariables.mockPathAndAdd(vararg pathsToAdd: Path) {
// TODO: Use native calls to `SetEnvironmentVariable` and `setenv(3)` to change env for children processes (they inherit parent envs).
val (pathKey, paths) = variables.entries.find { it.key.isPath }?.toPair()
?: Pair(PATH, "")
val oldVars = HashMap(variables)
val pathKey = variables.keys.firstOrNull { it.isPath } ?: PATH
val paths = variables[pathKey] ?: ""
val newPathVal = (paths.split(File.pathSeparator) + pathsToAdd.map { it.pathString }).joinToString(File.pathSeparator)
set(pathKey, newPathVal) // Mock System.env
mockkStatic(EnvironmentUtil::class) // Mock EnvironmentUtil
every { EnvironmentUtil.getValue(any()) }.coAnswers {
if ((it.invocation.args[0] as String).isPath) newPathVal else it.invocation.originalCall.invoke() as String
}
val currentEnvs = HashMap(variables)
currentEnvs[pathKey] = newPathVal
currentEnvs[PATH] = newPathVal
EnvironmentUtil.setEnvironmentLoader(MyMap(new = currentEnvs, old = oldVars))
}
/**
* See [mockPathAndAdd]
*/
fun EnvironmentVariables.unMockPath() {
unmockkStatic(EnvironmentUtil::class)
val original = HashMap((EnvironmentUtil.getEnvironmentMap() as MyMap).old)
EnvironmentUtil.setEnvironmentLoader { original }
}
private const val PATH = "PATH"
private val String.isPath: Boolean get() = uppercase() == PATH
private val String.isPath: Boolean get() = uppercase() == PATH
private class MyMap(private val new: Map<String, String>, val old: Map<String, String>) : Supplier<Map<String, String>>,
Map<String, String> by new {
override fun get(): Map<String, String> = this
}