Expose creation of CompatibilityChecker (fd4s#988)

* Expose creation of compat checker * trigger build
aartigao · Oct 16, 2023 · e8c6339 · e8c6339
1 parent e8aac1e
commit e8c6339
Showing 1 changed file with 41 additions and 32 deletions.
diff --git a/modules/vulcan-testkit-munit/src/main/scala/fs2/kafka/vulcan/testkit/SchemaSuite.scala b/modules/vulcan-testkit-munit/src/main/scala/fs2/kafka/vulcan/testkit/SchemaSuite.scala
@@ -39,42 +39,51 @@ trait SchemaSuite extends FunSuite {
     override def apply(): CompatibilityChecker[IO] = checker
 
     override def beforeAll(): Unit =
-      checker = clientSettings.createSchemaRegistryClient
-        .map { client =>
-          new CompatibilityChecker[IO] {
-            private def registrySchema(subject: String): IO[Schema] =
-              for {
-                metadata <- IO.delay(client.getLatestSchemaMetadata(subject))
-                schema <- IO.delay(
-                  client.getSchemaById(metadata.getId).asInstanceOf[AvroSchema]
-                )
-              } yield schema.rawSchema()
+      checker = newCompatibilityChecker(clientSettings)
+        .unsafeRunSync()
+  }
+
+  def newCompatibilityChecker(
+    clientSettings: SchemaRegistryClientSettings[IO]
+  ): IO[CompatibilityChecker[IO]] =
+    clientSettings.createSchemaRegistryClient
+      .map { client =>
+        new CompatibilityChecker[IO] {
+
+          private def registrySchema(subject: String): IO[Schema] =
+            for {
+              metadata <- IO.delay(client.getLatestSchemaMetadata(subject))
+              schema <- IO.delay(
+                client.getSchemaById(metadata.getId).asInstanceOf[AvroSchema]
+              )
+            } yield schema.rawSchema()
 
-            def checkReaderCompatibility[A](
-              reader: Codec[A],
-              writerSubject: String
-            ): IO[SchemaCompatibility.SchemaPairCompatibility] = {
-              val vulcanSchema = codecAsSchema(reader)
-              registrySchema(writerSubject).map { regSchema =>
-                SchemaCompatibility.checkReaderWriterCompatibility(
-                  vulcanSchema,
-                  regSchema
-                )
-              }
+          def checkReaderCompatibility[A](
+            reader: Codec[A],
+            writerSubject: String
+          ): IO[SchemaCompatibility.SchemaPairCompatibility] = {
+            val vulcanSchema = codecAsSchema(reader)
+            registrySchema(writerSubject).map { regSchema =>
+              SchemaCompatibility.checkReaderWriterCompatibility(
+                vulcanSchema,
+                regSchema
+              )
             }
+          }
 
-            def checkWriterCompatibility[A](writer: Codec[A], readerSubject: String)
-              : IO[SchemaCompatibility.SchemaPairCompatibility] = {
-              val vulcanSchema = codecAsSchema(writer)
-              registrySchema(readerSubject).map { regSchema =>
-                SchemaCompatibility.checkReaderWriterCompatibility(
-                  regSchema,
-                  vulcanSchema
-                )
-              }
+          def checkWriterCompatibility[A](
+            writer: Codec[A],
+            readerSubject: String
+          ): IO[SchemaCompatibility.SchemaPairCompatibility] = {
+            val vulcanSchema = codecAsSchema(writer)
+            registrySchema(readerSubject).map { regSchema =>
+              SchemaCompatibility.checkReaderWriterCompatibility(
+                regSchema,
+                vulcanSchema
+              )
             }
           }
+
         }
-        .unsafeRunSync()
-  }
+      }
 }