Add test for annotation default value instrumentation (#174)

UnitTestBot · Mar 1, 2024 · aca7b5d · aca7b5d
1 parent e8faa82
commit aca7b5d
Show file tree

Hide file tree

Showing 5 changed files with 43 additions and 13 deletions.
diff --git a/usvm-jvm-instrumentation/src/samples/java/example/AnnotatedMethodClass.java b/usvm-jvm-instrumentation/src/samples/java/example/AnnotatedMethodClass.java
diff --git a/usvm-jvm-instrumentation/src/samples/java/example/AnnotationsEx.java b/usvm-jvm-instrumentation/src/samples/java/example/AnnotationsEx.java
@@ -0,0 +1,16 @@
+package example;
+
+public class AnnotationsEx {
+    @MyAnnotation
+    public static int getSelfAnnotationCount() throws NoSuchMethodException {
+        return AnnotationsEx.class.getMethod("getSelfAnnotationCount").getAnnotations().length;
+    }
+
+    @MyAnnotation
+    public static String getAnnotationDefaultValue() throws NoSuchMethodException {
+        return AnnotationsEx.class
+                .getMethod("getAnnotationDefaultValue")
+                .getAnnotation(MyAnnotation.class)
+                .x();
+    }
+}
diff --git a/usvm-jvm-instrumentation/src/samples/java/example/MyAnnotation.java b/usvm-jvm-instrumentation/src/samples/java/example/MyAnnotation.java
@@ -5,4 +5,5 @@
 
 @Retention(RetentionPolicy.RUNTIME)
 public @interface MyAnnotation {
+    String x() default "MyAnnotation default value";
 }
diff --git a/...nstrumentation/src/test/kotlin/org/usvm/instrumentation/executor/ConcreteExecutorTests.kt b/...nstrumentation/src/test/kotlin/org/usvm/instrumentation/executor/ConcreteExecutorTests.kt
@@ -74,7 +74,7 @@ class ConcreteExecutorTests: UTestConcreteExecutorTest() {
 
     @Test
     fun `annotated method test`() = executeTest {
-        val uTest = UTestCreator.AnnotatedMethodClass.getClassAnnotationCount(jcClasspath)
+        val uTest = UTestCreator.AnnotationsEx.getSelfAnnotationCount(jcClasspath)
         val res = uTestConcreteExecutor.executeAsync(uTest)
         assertIs<UTestExecutionSuccessResult>(res)
         val result = res.result
@@ -83,6 +83,18 @@ class ConcreteExecutorTests: UTestConcreteExecutorTest() {
         assertEquals(1, result.value)
     }
 
+    @Test
+    @Disabled
+    fun `annotation default value test`() = executeTest {
+        val uTest = UTestCreator.AnnotationsEx.getAnnotationDefaultValue(jcClasspath)
+        val res = uTestConcreteExecutor.executeAsync(uTest)
+        assertIs<UTestExecutionSuccessResult>(res)
+        val result = res.result
+        assertNotNull(result)
+        assertIs<UTestConstantDescriptor.String>(result)
+        assertEquals("MyAnnotation default value", result.value)
+    }
+
     @Test
     fun `static fields test`() = executeTest {
         repeat(3) {

diff --git a/usvm-jvm-instrumentation/src/test/kotlin/org/usvm/instrumentation/util/UTestCreator.kt b/usvm-jvm-instrumentation/src/test/kotlin/org/usvm/instrumentation/util/UTestCreator.kt
@@ -365,11 +365,20 @@ object UTestCreator {
         }
     }
 
-    object AnnotatedMethodClass {
-        fun getClassAnnotationCount(jcClasspath: JcClasspath): UTest {
-            val jcClass = jcClasspath.findClass<example.AnnotatedMethodClass>()
+    object AnnotationsEx {
+        fun getSelfAnnotationCount(jcClasspath: JcClasspath): UTest {
+            val jcClass = jcClasspath.findClass<example.AnnotationsEx>()
             val jcMethod = jcClass.findDeclaredMethodOrNull("getSelfAnnotationCount")
-                ?: error("Cant find method getSelfAnnotationCount in class AnnotatedMethodClass")
+                ?: error("Cant find method getSelfAnnotationCount in class AnnotationsEx")
+            val instance = UTestNullExpression(jcClass.toType())
+            val statements = emptyList<UTestInst>()
+            return UTest(statements, UTestMethodCall(instance, jcMethod, emptyList()))
+        }
+
+        fun getAnnotationDefaultValue(jcClasspath: JcClasspath): UTest {
+            val jcClass = jcClasspath.findClass<example.AnnotationsEx>()
+            val jcMethod = jcClass.findDeclaredMethodOrNull("getAnnotationDefaultValue")
+                ?: error("Cant find method getAnnotationDefaultValue in class AnnotationsEx")
             val instance = UTestNullExpression(jcClass.toType())
             val statements = emptyList<UTestInst>()
             return UTest(statements, UTestMethodCall(instance, jcMethod, emptyList()))