def setUp(self):
self.client = SchemaRegistryClient('test_domain.gc.com')
class TestSchemaRegistryClient(TestCase):
def setUp(self):
self.client = SchemaRegistryClient('test_domain.gc.com')
def test_get_schema(self):
with Mocker() as m:
m.get(url('/schemas/ids/abc123'), json={
'schema': STRING_SCHEMA
})
schema = self.client.get_schema('abc123')
self.assertEquals({"type": "string"}, schema)
def test_get_non_existent_schema(self):
with Mocker() as m:
with self.assertRaises(SchemaRegistryException) as cm:
m.get(
url('/schemas/ids/abc123'),
json={
"error_code": 40403,
"message": "Schema not found"
},
status_code=404)
self.client.get_schema('abc123')
self.assertEquals(cm.exception.code, 40403)
self.assertEquals(cm.exception.message, 'Schema not found')
def test_get_subjects(self):
with Mocker() as m:
m.get(
url('/subjects'),
json=["beans", "frank"])
self.assertEquals(
["beans", "frank"],
self.client.get_subjects())
def test_get_subjects_internal_server_error(self):
with Mocker() as m:
with self.assertRaises(SchemaRegistryException):
m.get(
url('/subjects'),
json={
"error_code": 50001,
"message": "Error in the backend datastore"
},
status_code=500)
self.client.get_subjects()
def test_get_subject_version_ids(self):
with Mocker() as m:
m.get(
url('/subjects/test/versions'),
json=[1, 2, 3])
self.assertEquals(
[1, 2, 3],
self.client.get_subject_version_ids('test'))
def test_get_subject_version_ids_non_existent_subject(self):
with Mocker() as m:
with self.assertRaises(SchemaRegistryException):
m.get(
url('/subjects/test/versions'),
json={
"error_code": 40401,
"message": "Subject not found"
},
status_code=404)
self.client.get_subject_version_ids('test')
def test_get_subject_version(self):
with Mocker() as m:
m.get(
url('/subjects/test/versions/34'),
json={
'name': 'test',
'version': 34,
'schema': STRING_SCHEMA
})
self.assertEquals(
SCHEMA,
self.client.get_subject_version('test', 34))
def test_get_subject_version_non_existent_version(self):
with Mocker() as m:
with self.assertRaises(SchemaRegistryException):
m.get(
url('/subjects/test/versions/34'),
json={
"error_code": 42202,
"message": "Invalid version"
},
status_code=422)
self.client.get_subject_version('test', 34)
def test_get_latest_subject_version(self):
with Mocker() as m:
m.get(
url('/subjects/test/versions/latest'),
json={
'name': 'test',
'version': 34,
'schema': STRING_SCHEMA
})
self.assertEquals(
SCHEMA,
self.client.get_subject_latest_version('test'))
def test_register_subject_version(self):
with Mocker() as m:
m.post(
url('/subjects/test/versions'),
json={'id':34})
self.assertEquals(
34,
self.client.register_subject_version('test', SCHEMA))
self.assertEquals(
{'schema': STRING_SCHEMA},
m.last_request.json())
def test_register_subject_version_incompatible(self):
with Mocker() as m:
with self.assertRaises(SchemaRegistryException):
m.post(
url('/subjects/test/versions'), status_code=409)
self.client.register_subject_version('test', SCHEMA)
def test_schema_registration_for_subject(self):
with Mocker() as m:
m.post(
url('/subjects/test'),
json={
'subject': 'test',
'id': 34,
'version': 3,
'schema': STRING_SCHEMA
})
schema_id, version = self.client.schema_registration_for_subject('test', SCHEMA)
self.assertEquals(34, schema_id)
self.assertEquals(3, version)
self.assertEquals({'schema': STRING_SCHEMA}, m.last_request.json())
def test_schema_registration_for_subject_unknown_schema(self):
with Mocker() as m:
with self.assertRaises(SchemaRegistryException):
m.post(
url('/subjects/test'),
json={
'error_code': 40403,
'message': 'Schema not found'
}, status_code=404)
self.client.schema_registration_for_subject('test', SCHEMA)
def test_schema_is_registered_for_subject(self):
with Mocker() as m:
m.post(
url('/subjects/test'),
json={
'subject': 'test',
'id': 34,
'version': 3,
'schema': STRING_SCHEMA
})
self.assertTrue(self.client.schema_is_registered_for_subject('test', SCHEMA))
self.assertEquals({'schema': STRING_SCHEMA}, m.last_request.json())
def test_schema_is_registered_for_subject_unknown_schema(self):
with Mocker() as m:
m.post(
url('/subjects/test'),
json={
'error_code': 40403,
'message': 'Schema not found'
}, status_code=404)
self.assertFalse(self.client.schema_is_registered_for_subject('test', SCHEMA))
def test_schema_is_compatible_with_subject_version(self):
with Mocker() as m:
m.post(
url('/compatibility/subjects/test/versions/3'),
json={'is_compatible': True})
self.assertTrue(self.client.schema_is_compatible_with_subject_version('test', 3, SCHEMA))
self.assertEquals({'schema': STRING_SCHEMA}, m.last_request.json())
def test_schema_is_not_compatible_with_subject_version(self):
with Mocker() as m:
m.post(
url('/compatibility/subjects/test/versions/3'),
json={'is_compatible': False})
self.assertFalse(self.client.schema_is_compatible_with_subject_version('test', 3, SCHEMA))
self.assertEquals({'schema': STRING_SCHEMA}, m.last_request.json())
def test_schema_is_compatible_with_subject_version_bad_subject(self):
with Mocker() as m:
with self.assertRaises(SchemaRegistryException):
m.post(
url('/compatibility/subjects/test/versions/3'),
json={
"error_code": 42202,
"message": "Invalid version"
},
status_code=422)
self.client.schema_is_compatible_with_subject_version('test', 3, SCHEMA)
def test_set_global_compatibility_level(self):
with Mocker() as m:
m.put(url('/config'), json={'compatibility': 'FULL'})
self.client.set_global_compatibility_level(CompatibilityLevel.full)
self.assertEqual({'compatibility': 'FULL'}, m.last_request.json())
def test_get_global_compatibility_level(self):
with Mocker() as m:
m.get(url('/config'), json={'compatibility': 'FULL'})
level = self.client.get_global_compatibility_level()
self.assertEquals(CompatibilityLevel.full, level)
def test_set_subject_compatibility_level(self):
with Mocker() as m:
m.put(url('/config/test'), json={'compatibility': 'FULL'})
self.client.set_subject_compatibility_level('test', CompatibilityLevel.full)
self.assertEqual({'compatibility': 'FULL'}, m.last_request.json())
def test_get_subject_compatibility_level(self):
with Mocker() as m:
m.get(url('/config/test'), json={'compatibility': 'FULL'})
level = self.client.get_subject_compatibility_level('test')
self.assertEquals(CompatibilityLevel.full, level)
