Merge pull request confluentinc#952 from choogeboom/hoogamaphone_fix_get_compatibility

Matt Howlett · web-flow · commit c3dd07c6d6f5 · 2021-03-10T13:09:55.000-06:00
Make get_compatibility return the actual compatibility level
diff --git a/src/confluent_kafka/schema_registry/schema_registry_client.py b/src/confluent_kafka/schema_registry/schema_registry_client.py
@@ -625,7 +625,8 @@ def get_compatibility(self, subject_name=None):
         else:
             url = 'config'
 
-        return self._rest_client.get(url)
+        result = self._rest_client.get(url)
+        return result['compatibilityLevel']
 
 
 class Schema(object):
diff --git a/tests/integration/schema_registry/test_api_client.py b/tests/integration/schema_registry/test_api_client.py
@@ -383,8 +383,7 @@ def test_api_subject_config_update(kafka_cluster, load_file):
     sr.set_compatibility(subject_name=subject,
                          level="FULL_TRANSITIVE")
 
-    assert sr.get_compatibility(
-        subject_name=subject)['compatibilityLevel'] == "FULL_TRANSITIVE"
+    assert sr.get_compatibility(subject_name=subject) == "FULL_TRANSITIVE"
 
 
 def test_api_config_invalid(kafka_cluster):
@@ -415,7 +414,7 @@ def test_api_config_update(kafka_cluster):
 
     for level in ["BACKWARD", "BACKWARD_TRANSITIVE", "FORWARD", "FORWARD_TRANSITIVE"]:
         sr.set_compatibility(level=level)
-        assert sr.get_compatibility()['compatibilityLevel'] == level
+        assert sr.get_compatibility() == level
 
 
 def test_api_register_logical_schema(kafka_cluster, load_file):
