libraries/stdlib/jdk7/test/AbstractPathTest.kt

/*
 * 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
    }

    @IgnorableReturnValue
    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")
            }
        }
    }

    @IgnorableReturnValue
    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) }
        }
    }
}