Improve jdeps tracking and add test cases

src/main/kotlin/io/bazel/kotlin/plugin/jdeps/JdepsGenExtension.kt

@@ -191,6 +191,9 @@ class JdepsGenExtension(
           descriptor.typeConstructor.supertypes.forEach {
             collectTypeReferences(it)
           }
+          descriptor.annotations.forEach { annotation ->
+            collectTypeReferences(annotation.type)
+          }
         }
         is FunctionDescriptor -> {
           descriptor.returnType?.let { collectTypeReferences(it) }

src/test/kotlin/io/bazel/kotlin/builder/tasks/jvm/KotlinBuilderJvmJdepsTest.kt

@@ -221,6 +221,87 @@ class KotlinBuilderJvmJdepsTest(private val enableK2Compiler: Boolean) {
     assertIncomplete(jdeps).isEmpty()
   }
 
+  @Test
+  fun `java annotation on class is an explict dep`() {
+    val dependingTarget = runJdepsCompileTask { c: KotlinJvmTestBuilder.TaskBuilder ->
+      c.addSource(
+        "AnotherClass.kt",
+        """
+          package something
+
+          @JavaAnnotation
+          class AnotherClass {
+          }
+        """
+      )
+      c.addDirectDependencies(TEST_FIXTURES_DEP)
+    }
+    val jdeps = depsProto(dependingTarget)
+
+    assertThat(jdeps.ruleLabel).isEqualTo(dependingTarget.label())
+
+    assertExplicit(jdeps).contains(TEST_FIXTURES_DEP.singleCompileJar())
+    assertImplicit(jdeps).isEmpty()
+    assertUnused(jdeps).isEmpty()
+    assertIncomplete(jdeps).isEmpty()
+  }
+
+  @Test
+  fun `java sealed classes`() {
+    val dependingTarget = runJdepsCompileTask { c: KotlinJvmTestBuilder.TaskBuilder ->
+      c.addSource(
+        "AnotherClass.kt",
+        """
+          package something
+
+          import pkg.assertion.ExampleException
+
+          sealed class SomeClass {
+              companion object {
+                  fun mapExceptions(exception: Exception) = when (exception) {
+                      is ExampleException -> exception
+                      else                -> exception
+                  }
+              }
+          }
+        """
+      )
+      c.addDirectDependencies(TEST_FIXTURES_DEP)
+    }
+    val jdeps = depsProto(dependingTarget)
+
+    assertThat(jdeps.ruleLabel).isEqualTo(dependingTarget.label())
+
+    assertExplicit(jdeps).contains(TEST_FIXTURES_DEP.singleCompileJar())
+    assertImplicit(jdeps).isEmpty()
+    assertUnused(jdeps).isEmpty()
+    assertIncomplete(jdeps).isEmpty()
+  }
+
+  @Test
+  fun `java classes with inheritance`() {
+    val dependingTarget = runJdepsCompileTask { c: KotlinJvmTestBuilder.TaskBuilder ->
+      c.addSource(
+        "AnotherClass.kt",
+        """
+          package something
+
+          class SomeClass : SomeClassWithInheritance.BaseClassCallback {
+          }
+        """
+      )
+      c.addDirectDependencies(TEST_FIXTURES_DEP)
+    }
+    val jdeps = depsProto(dependingTarget)
+
+    assertThat(jdeps.ruleLabel).isEqualTo(dependingTarget.label())
+
+    assertExplicit(jdeps).contains(TEST_FIXTURES_DEP.singleCompileJar())
+    assertImplicit(jdeps).isEmpty()
+    assertUnused(jdeps).isEmpty()
+    assertIncomplete(jdeps).isEmpty()
+  }
+
   @Test
   fun `java annotation on property is an explict dep`() {
     val dependingTarget = runJdepsCompileTask { c: KotlinJvmTestBuilder.TaskBuilder ->

src/test/kotlin/io/bazel/kotlin/builder/tasks/testFixtures/BaseClass.java

@@ -0,0 +1,4 @@
+package something;
+
+public abstract class BaseClass {
+}

src/test/kotlin/io/bazel/kotlin/builder/tasks/testFixtures/ExampleException.java

@@ -0,0 +1,5 @@
+package pkg.assertion;
+
+public class ExampleException extends RuntimeException {
+
+}

src/test/kotlin/io/bazel/kotlin/builder/tasks/testFixtures/SomeClassWithInheritance.java

@@ -0,0 +1,9 @@
+package something;
+
+public class SomeClassWithInheritance extends BaseClass {
+
+
+    interface BaseClassCallback {
+
+    }
+}