Leverage sealed hierarchy for cast that will always succeed (#2257)

cashapp · Aug 22, 2024 · bbc5455 · bbc5455
1 parent 4714b9c
commit bbc5455
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...protocol-host/src/commonMain/kotlin/app/cash/redwood/protocol/host/HostProtocolAdapter.kt b/...protocol-host/src/commonMain/kotlin/app/cash/redwood/protocol/host/HostProtocolAdapter.kt
@@ -52,8 +52,8 @@ public class HostProtocolAdapter<W : Any>(
   factory: ProtocolFactory<W>,
   private val eventSink: EventSink,
 ) : ChangesSink {
-  private val factory = requireNotNull(factory as? GeneratedProtocolFactory<W>) {
-    "Factory ${factory::class} was not generated by Redwood or is out of date"
+  private val factory = when (factory) {
+    is GeneratedProtocolFactory -> factory
   }
 
   private val nodes = mutableMapOf<Id, ProtocolNode<W>>(