Alexander Udalov
108 lines
4.6 KiB
Java
108 lines
4.6 KiB
Java
/*
|
|
* 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.codegen;
|
|
|
|
import org.jetbrains.kotlin.test.ConfigurationKind;
|
|
|
|
import java.lang.reflect.Array;
|
|
import java.lang.reflect.InvocationTargetException;
|
|
import java.lang.reflect.Method;
|
|
import java.util.Arrays;
|
|
|
|
public class VarArgTest extends CodegenTestCase {
|
|
@Override
|
|
protected void setUp() throws Exception {
|
|
super.setUp();
|
|
createEnvironmentWithMockJdkAndIdeaAnnotations(ConfigurationKind.JDK_ONLY);
|
|
}
|
|
|
|
public void testStringArray() throws InvocationTargetException, IllegalAccessException {
|
|
loadText("fun test(vararg ts: String) = ts");
|
|
Method main = generateFunction();
|
|
String[] args = {"mama", "papa"};
|
|
assertSame(args, main.invoke(null, new Object[] {args}));
|
|
}
|
|
|
|
public void testIntArray() throws InvocationTargetException, IllegalAccessException {
|
|
loadText("fun test(vararg ts: Int) = ts");
|
|
Method main = generateFunction();
|
|
int[] args = {3, 4};
|
|
assertSame(args, main.invoke(null, new Object[] {args}));
|
|
}
|
|
|
|
public void testIntArrayKotlinNoArgs() throws InvocationTargetException, IllegalAccessException {
|
|
loadText("fun test() = testf(); fun testf(vararg ts: Int) = ts");
|
|
Method main = generateFunction("test");
|
|
Object res = main.invoke(null);
|
|
assertEquals(0, ((int[]) res).length);
|
|
}
|
|
|
|
public void testIntArrayKotlin() throws InvocationTargetException, IllegalAccessException {
|
|
loadText("fun test() = testf(239, 7); fun testf(vararg ts: Int) = ts");
|
|
Method main = generateFunction("test");
|
|
Object res = main.invoke(null);
|
|
assertEquals(2, ((int[]) res).length);
|
|
assertEquals(239, ((int[]) res)[0]);
|
|
assertEquals(7, ((int[]) res)[1]);
|
|
}
|
|
|
|
public void testNullableIntArrayKotlin() throws InvocationTargetException, IllegalAccessException {
|
|
loadText("fun test() = testf(239.toByte(), 7.toByte()); fun testf(vararg ts: Byte?) = ts");
|
|
Method main = generateFunction("test");
|
|
Object res = main.invoke(null);
|
|
assertEquals(2, ((Byte[]) res).length);
|
|
assertEquals((byte) ((Byte[]) res)[0], (byte) 239);
|
|
assertEquals(7, (byte) ((Byte[]) res)[1]);
|
|
}
|
|
|
|
public void testIntArrayKotlinObj() throws InvocationTargetException, IllegalAccessException {
|
|
loadText("fun test() = testf(\"239\"); fun testf(vararg ts: String) = ts");
|
|
Method main = generateFunction("test");
|
|
Object res = main.invoke(null);
|
|
assertEquals(1, ((String[]) res).length);
|
|
assertEquals("239", ((String[]) res)[0]);
|
|
}
|
|
|
|
public void testArrayT() throws InvocationTargetException, IllegalAccessException {
|
|
loadText("fun test() = _array(2, 4); fun <T> _array(vararg elements : T) = elements");
|
|
Method main = generateFunction("test");
|
|
Object res = main.invoke(null);
|
|
assertEquals(2, ((Integer[]) res).length);
|
|
assertEquals(2, (int) ((Integer[]) res)[0]);
|
|
assertEquals(4, (int) ((Integer[]) res)[1]);
|
|
}
|
|
|
|
public void testArrayAsVararg() throws InvocationTargetException, IllegalAccessException {
|
|
loadText("private fun asList(vararg elems: String) = elems; fun test(ts: Array<String>) = asList(*ts); ");
|
|
Method main = generateFunction("test");
|
|
String[] args = {"mama", "papa"};
|
|
String[] result = (String []) main.invoke(null, new Object[] {args});
|
|
assertNotSame(args, result);
|
|
assertTrue(Arrays.equals(args, result));
|
|
}
|
|
|
|
public void testArrayAsVararg2() throws InvocationTargetException, IllegalAccessException {
|
|
loadText("private fun asList(vararg elems: String) = elems; fun test(ts1: Array<String>, ts2: String) = asList(*ts1, ts2); ");
|
|
Method main = generateFunction("test");
|
|
Object invoke = main.invoke(null, new String[] {"mama"}, "papa");
|
|
assertInstanceOf(invoke, String[].class);
|
|
assertEquals(2, Array.getLength(invoke));
|
|
assertEquals("mama", Array.get(invoke, 0));
|
|
assertEquals("papa", Array.get(invoke, 1));
|
|
}
|
|
}
|