fix embedded root schema

php-kafka · Nov 24, 2023 · da2d170 · da2d170
1 parent 64776fd
commit da2d170
Show file tree

Hide file tree

Showing 4 changed files with 115 additions and 6 deletions.
diff --git a/.../schemaTemplates/embedded/Collection.avsc → example/schemaTemplates/Collection.avsc b/.../schemaTemplates/embedded/Collection.avsc → example/schemaTemplates/Collection.avsc
@@ -1,6 +1,7 @@
 {
     "type": "record",
     "namespace": "com.example",
+    "schema_level": "root",
     "name": "Collection",
     "fields": [
         { "name": "name", "type": "string" },

diff --git a/src/Merger/SchemaMerger.php b/src/Merger/SchemaMerger.php
@@ -118,8 +118,17 @@ private function replaceSchemaIdWithDefinition(
     ): string {
         $idString = '"' . $schemaId . '"';
         $pos = (int) strpos($rootDefinition, $idString);
+        $embeddedDefinitionWithoutLevel = $this->removeSchemaLevel($embeddedDefinition);
 
-        return substr_replace($rootDefinition, $embeddedDefinition, $pos, strlen($idString));
+        return substr_replace($rootDefinition, $embeddedDefinitionWithoutLevel, $pos, strlen($idString));
+    }
+
+    private function removeSchemaLevel(string $embeddedDefinition): string
+    {
+        $arraySchema = json_decode($embeddedDefinition, true);
+        unset($arraySchema['schema_level']);
+
+        return json_encode($arraySchema) ?? '';
     }
 
     /**

diff --git a/tests/Integration/Registry/SchemaRegistryTest.php b/tests/Integration/Registry/SchemaRegistryTest.php
@@ -61,7 +61,7 @@ public function testGetRootSchemas(): void
 
         $rootSchemas = $registry->getRootSchemas();
 
-        self::assertCount(2, $rootSchemas);
+        self::assertCount(3, $rootSchemas);
 
         foreach ($rootSchemas as $rootSchema) {
             self::assertInstanceOf(SchemaTemplateInterface::class, $rootSchema);

diff --git a/tests/Unit/Merger/SchemaMergerTest.php b/tests/Unit/Merger/SchemaMergerTest.php
@@ -95,14 +95,14 @@ public function testGetResolvedSchemaTemplate(): void
                 { "name": "items", "type": {"type": "array", "items": "com.example.Page" }, "default": [] }
             ]
         }';
-        $subschemaDefinition = '{
+        $subschemaDefinition = json_encode(json_decode('{
             "type": "record",
             "namespace": "com.example",
             "name": "Page",
             "fields": [
                 { "name": "number", "type": "int" }
             ]
-        }';
+        }'));
 
         $expectedResult = str_replace('"com.example.Page"', $subschemaDefinition, $rootDefinition);
 
@@ -134,6 +134,105 @@ public function testGetResolvedSchemaTemplate(): void
         $merger->getResolvedSchemaTemplate($rootSchemaTemplate);
     }
 
+    public function testGetResolvedSchemaTemplateWithEmbeddedRoot(): void
+    {
+        $rootDefinition = '{
+          "type": "record",
+          "namespace": "com.example",
+          "schema_level": "root",
+          "name": "Library",
+          "fields": [
+            {
+              "name": "name",
+              "type": "string"
+            },
+            {
+              "name": "foundingYear",
+              "type": [
+                "null",
+                "int"
+              ],
+              "default": null
+            },
+            {
+              "name": "type",
+              "type": [
+                "null",
+                {
+                  "name": "type",
+                  "type": "enum",
+                  "symbols": [
+                    "PUBLIC",
+                    "PRIVATE"
+                  ]
+                }
+              ],
+              "default": null
+            },
+            {
+              "name": "collection",
+              "type": {
+                "type": "array",
+                "items": "com.example.Collection"
+              },
+              "default": []
+            },
+            {
+              "name": "archive",
+              "type": {
+                "type": "array",
+                "items": "com.example.Collection"
+              },
+              "default": []
+            }
+          ]
+        }';
+        $subschemaDefinition = json_encode(json_decode('{
+            "type": "record",
+            "namespace": "com.example",
+            "schema_level": "root",
+            "name": "Collection",
+            "fields": [
+                { "name": "name", "type": "string" }
+            ]
+        }'));
+
+        $subschemaDefinitionArray = \Safe\json_decode($subschemaDefinition, true);
+        unset($subschemaDefinitionArray['schema_level']);
+        $subschemaDefinitionWithoutLevel = json_encode($subschemaDefinitionArray);
+
+        $subschemaId = '"com.example.Collection"';
+        $pos = strpos($rootDefinition, $subschemaId);
+        $expectedResult = substr_replace($rootDefinition, $subschemaDefinitionWithoutLevel, $pos, strlen($subschemaId));
+
+        $subschemaTemplate = $this->getMockForAbstractClass(SchemaTemplateInterface::class);
+        $subschemaTemplate
+            ->expects(self::once())
+            ->method('getSchemaDefinition')
+            ->willReturn($subschemaDefinition);
+        $schemaRegistry = $this->getMockForAbstractClass(SchemaRegistryInterface::class);
+        $schemaRegistry
+            ->expects(self::once())
+            ->method('getSchemaById')
+            ->with('com.example.Collection')
+            ->willReturn($subschemaTemplate);
+        $rootSchemaTemplate = $this->getMockForAbstractClass(SchemaTemplateInterface::class);
+        $rootSchemaTemplate
+            ->expects(self::once())
+            ->method('getSchemaDefinition')
+            ->willReturn($rootDefinition);
+        $rootSchemaTemplate
+            ->expects(self::once())
+            ->method('withSchemaDefinition')
+            ->with($expectedResult)
+            ->willReturn($rootSchemaTemplate);
+
+        $merger = new SchemaMerger();
+        $merger->setSchemaRegistry($schemaRegistry);
+
+        $merger->getResolvedSchemaTemplate($rootSchemaTemplate);
+    }
+
     public function testGetResolvedSchemaTemplateWithMultiEmbedd(): void
     {
         $rootDefinition = $this->reformatJsonString('{
@@ -315,14 +414,14 @@ public function testGetResolvedSchemaTemplateWithDifferentNamespaceForEmbeddedSc
                 { "name": "items", "type": {"type": "array", "items": "com.example.other.Page" }, "default": [] }
             ]
         }';
-        $subschemaDefinition = '{
+        $subschemaDefinition = json_encode(json_decode('{
             "type": "record",
             "namespace": "com.example.other",
             "name": "Page",
             "fields": [
                 { "name": "number", "type": "int" }
             ]
-        }';
+        }'));
 
         $expectedResult = str_replace('"com.example.other.Page"', $subschemaDefinition, $rootDefinition);