Attempt to canonicalize enum constants (#70)

* Attempt to canonicalize enum constants * Just use the owner name --------- Co-authored-by: Jesse Wilson <[email protected]>
cashapp · Nov 20, 2024 · 4bb758d · 4bb758d
1 parent 6480b4f
commit 4bb758d
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/burst-kotlin-plugin/src/main/kotlin/app/cash/burst/kotlin/Argument.kt b/burst-kotlin-plugin/src/main/kotlin/app/cash/burst/kotlin/Argument.kt
@@ -210,10 +210,10 @@ private fun enumValueArguments(
 ): List<Argument> {
   val enumEntries = referenceClass.declarations.filterIsInstance<IrEnumEntry>()
   val hasDefaultValue = parameter.defaultValue != null
-  val defaultEnumSymbol = parameter.defaultValue?.let { defaultValue ->
+  val defaultEnumSymbolName = parameter.defaultValue?.let { defaultValue ->
     val expression = defaultValue.expression
     when {
-      expression is IrGetEnumValue -> expression.symbol
+      expression is IrGetEnumValue -> expression.symbol.owner.name
       expression is IrConst<*> && expression.value == null -> null
       else -> unexpectedDefaultValue(parameter)
     }
@@ -225,7 +225,7 @@ private fun enumValueArguments(
         EnumValueArgument(
           original = parameter,
           type = parameter.type,
-          isDefault = hasDefaultValue && enumEntry.symbol == defaultEnumSymbol,
+          isDefault = hasDefaultValue && enumEntry.symbol.owner.name == defaultEnumSymbolName,
           value = enumEntry,
         ),
       )
@@ -235,7 +235,7 @@ private fun enumValueArguments(
         NullArgument(
           original = parameter,
           type = parameter.type,
-          isDefault = hasDefaultValue && defaultEnumSymbol == null,
+          isDefault = hasDefaultValue && defaultEnumSymbolName == null,
         ),
       )
     }