Files

117 lines
3.9 KiB
Kotlin

/*
* Copyright 2010-2015 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jetbrains.kotlin.integration
import com.intellij.openapi.util.io.FileUtil
import org.jetbrains.kotlin.test.util.KtTestUtil
import org.junit.Assert
import java.io.File
class CompilerFileLimitTest : CompilerSmokeTestBase() {
fun testLargeKotlinFile() {
val size = 300
val dir = tempDir("largeKotlinFileSrc")
val file = File(dir, "largeKotlinFile.kt")
file.writeText(generateLargeKotlinFile(size))
assertIsLargeButNotTooLarge(file)
runCompiler("largeKotlinFile.compile", file.absolutePath, "-d", tempDir("largeKotlinFileOut").absolutePath)
}
private fun generateLargeKotlinFile(size: Int): String {
return buildString {
append("package large\n\n")
(0..size).forEach {
appendLine("class Class$it")
appendLine("{")
appendLine("\tfun foo(): Long = $it")
appendLine("}")
appendLine("\n")
repeat(2000) {
appendLine("// kotlin rules ... and stuff")
}
}
appendLine("fun main()")
appendLine("{")
appendLine("\tval result = Class5().foo() + Class$size().foo()")
appendLine("\tprintln(result)")
appendLine("}")
}
}
fun testLargeJavaFile() {
val size = 300
val dir = tempDir("largeJavaFileSrc")
val javaDir = File(dir, "large")
javaDir.mkdir()
val javaFile = File(javaDir, "Large.java")
javaFile.writeText(generateLargeJavaFile(size))
val ktFile = File(dir, "useLargerJava.kt")
ktFile.writeText(generateKotlinFileThatUsesLargeJavaFile(size))
assertIsLargeButNotTooLarge(javaFile)
runCompiler("largeJavaFile.compile", dir.absolutePath, "-d", tempDir("largeJavaFileOut").absolutePath)
}
private fun assertIsLargeButNotTooLarge(file: File) {
Assert.assertTrue(file.length() > 15 * FileUtil.MEGABYTE)
Assert.assertTrue(file.length() < 20 * FileUtil.MEGABYTE)
}
private fun generateKotlinFileThatUsesLargeJavaFile(size: Int): String {
return buildString {
append("package usesLarge\n\n")
append("import large.Large\n\n")
appendLine("fun main()")
appendLine("{")
appendLine("\tval result = Large.Class0().foo() + Large.Class$size().foo()")
appendLine("\tprintln(result)")
appendLine("}")
}
}
private fun generateLargeJavaFile(size: Int): String {
return buildString {
append("package large;\n\n")
appendLine("public class Large")
appendLine("{")
(0..size).forEach {
appendLine("\tpublic static class Class$it")
appendLine("\t{")
appendLine("\t\tpublic long foo()")
appendLine("\t\t{")
appendLine("\t\t\t return $it;")
appendLine("\t\t}")
appendLine("\t}")
appendLine("\n")
repeat(2000) {
appendLine("// kotlin rules ... and stuff")
}
}
appendLine("}")
}
}
private fun tempDir(markerName: String) = KtTestUtil.tmpDir("${CompilerFileLimitTest::class.simpleName}$markerName")
}