Fix klib metadata (#89)

Closes: #87 Co-authored-by: Jesse Wilson <[email protected]>
cashapp · Jan 23, 2025 · 48481d4 · 48481d4
1 parent 628f6cf
commit 48481d4
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/burst-gradle-plugin/src/test/kotlin/app/cash/burst/gradle/BurstGradlePluginTest.kt b/burst-gradle-plugin/src/test/kotlin/app/cash/burst/gradle/BurstGradlePluginTest.kt
@@ -90,11 +90,12 @@ class BurstGradlePluginTest {
       )
       val klibMetadata = klib.moduleMetadata()
       val coffeeTestMetadata = klibMetadata.classes.first { it.name == "CoffeeTest" }
-      // TODO: This expected output is wrong. It should have specializations like test_Milk etc.
-      //     https://github.com/cashapp/burst/issues/87
       assertThat(coffeeTestMetadata.functions.map { it.name }).containsExactlyInAnyOrder(
         "setUp",
         "test",
+        "test_Milk",
+        "test_None",
+        "test_Oat",
       )
     }
   }

diff --git a/burst-kotlin-plugin/src/main/kotlin/app/cash/burst/kotlin/ClassSpecializer.kt b/burst-kotlin-plugin/src/main/kotlin/app/cash/burst/kotlin/ClassSpecializer.kt
@@ -163,7 +163,7 @@ internal class ClassSpecializer(
     superConstructor: IrConstructor,
     specialization: Specialization,
   ) {
-    original.addConstructor {
+    val constructor = original.addConstructor {
       initDefaults(original)
       isPrimary = false
     }.apply {
@@ -178,5 +178,6 @@ internal class ClassSpecializer(
         }
       }
     }
+    pluginContext.metadataDeclarationRegistrar.registerConstructorAsMetadataVisible(constructor)
   }
 }
diff --git a/burst-kotlin-plugin/src/main/kotlin/app/cash/burst/kotlin/FunctionSpecializer.kt b/burst-kotlin-plugin/src/main/kotlin/app/cash/burst/kotlin/FunctionSpecializer.kt
@@ -91,6 +91,7 @@ internal class FunctionSpecializer(
     // Add new declarations.
     for (function in functions) {
       originalParent.addDeclaration(function)
+      pluginContext.metadataDeclarationRegistrar.registerFunctionAsMetadataVisible(function)
     }
   }