Abduqodiri Qurbonzoda
74 lines
2.3 KiB
Kotlin
74 lines
2.3 KiB
Kotlin
/*
|
|
* Copyright 2010-2020 JetBrains s.r.o. and Kotlin Programming Language contributors.
|
|
* Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
|
|
*/
|
|
|
|
package kotlin.jdk7.test
|
|
|
|
import java.nio.file.Path
|
|
import kotlin.io.path.createSymbolicLinkPointingTo
|
|
import kotlin.io.path.deleteIfExists
|
|
import kotlin.io.path.deleteRecursively
|
|
import kotlin.test.AfterTest
|
|
|
|
abstract class AbstractPathTest {
|
|
private val cleanUpActions = mutableListOf<Pair<Path, (Path) -> Unit>>()
|
|
|
|
fun Path.cleanup(): Path {
|
|
cleanUpActions.add(this to { it.deleteIfExists() })
|
|
return this
|
|
}
|
|
|
|
fun Path.cleanupRecursively(): Path {
|
|
cleanUpActions.add(this to { it.deleteRecursively() })
|
|
return this
|
|
}
|
|
|
|
@AfterTest
|
|
fun cleanUp() {
|
|
for ((path, action) in cleanUpActions) {
|
|
try {
|
|
action(path)
|
|
} catch (e: Throwable) {
|
|
println("Failed to execute cleanup action for $path")
|
|
}
|
|
}
|
|
}
|
|
|
|
fun Path.tryCreateSymbolicLinkTo(original: Path): Path? {
|
|
return try {
|
|
this.createSymbolicLinkPointingTo(original)
|
|
} catch (e: Exception) {
|
|
// the underlying OS may not support symbolic links or may require a privilege
|
|
println("Creating a symbolic link failed with $e")
|
|
null
|
|
}
|
|
}
|
|
|
|
fun withRestrictedRead(vararg paths: Path, alsoReset: List<Path> = emptyList(), block: () -> Unit) {
|
|
try {
|
|
if (paths.all { it.toFile().setReadable(false) }) {
|
|
block()
|
|
} else {
|
|
System.err.println("Couldn't restrict read access")
|
|
}
|
|
} finally {
|
|
paths.forEach { it.toFile().setReadable(true) }
|
|
alsoReset.forEach { it.toFile().setReadable(true) }
|
|
}
|
|
}
|
|
|
|
fun withRestrictedWrite(vararg paths: Path, alsoReset: List<Path> = emptyList(), block: () -> Unit) {
|
|
try {
|
|
if (paths.all { it.toFile().setWritable(false) }) {
|
|
block()
|
|
} else {
|
|
System.err.println("Couldn't restrict write access")
|
|
}
|
|
} finally {
|
|
paths.forEach { it.toFile().setWritable(true) }
|
|
alsoReset.forEach { it.toFile().setWritable(true) }
|
|
}
|
|
}
|
|
}
|