Skip to content

Commit 09772d5

Browse files
authored
Minor fix to cache to ensure schema matches input (#1948)
1 parent 08e878b commit 09772d5

File tree

1 file changed

+10
-1
lines changed

1 file changed

+10
-1
lines changed

src/confluent_kafka/schema_registry/schema_registry_client.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -853,6 +853,7 @@ def register_schema_full_response(
853853

854854
registered_schema = RegisteredSchema.from_dict(response)
855855

856+
# The registered schema may not be fully populated
856857
self._cache.set_schema(subject_name, registered_schema.schema_id, schema)
857858

858859
return registered_schema
@@ -931,7 +932,15 @@ def lookup_schema(
931932
.format(_urlencode(subject_name), normalize_schemas, deleted),
932933
body=request)
933934

934-
registered_schema = RegisteredSchema.from_dict(response)
935+
result = RegisteredSchema.from_dict(response)
936+
937+
# Ensure the schema matches the input
938+
registered_schema = RegisteredSchema(
939+
schema_id=result.schema_id,
940+
subject=result.subject,
941+
version=result.version,
942+
schema=schema,
943+
)
935944

936945
self._cache.set_registered_schema(schema, registered_schema)
937946

0 commit comments

Comments
 (0)