def test_avro_schema(self):
avro = Avro().avro_schema(
'{"type":"record",'
'"name":"Address",'
'"namespace":"org.apache.flink.formats.avro.generated",'
'"fields":['
'{"name":"num","type":"int"},'
'{"name":"street","type":"string"},'
'{"name":"city","type":"string"},'
'{"name":"state","type":"string"},'
'{"name":"zip","type":"string"}'
']}')
expected = {
'format.avro-schema':
'{"type":"record",'
'"name":"Address",'
'"namespace":"org.apache.flink.formats.avro.generated",'
'"fields":['
'{"name":"num","type":"int"},'
'{"name":"street","type":"string"},'
'{"name":"city","type":"string"},'
'{"name":"state","type":"string"},'
'{"name":"zip","type":"string"}'
']}',
'format.property-version':
'1',
'format.type':
'avro'
}
properties = avro.to_properties()
self.assertEqual(expected, properties)
def test_record_class(self):
avro = Avro().record_class("org.apache.flink.formats.avro.generated.Address")
expected = {'format.record-class': 'org.apache.flink.formats.avro.generated.Address',
'format.property-version': '1',
'format.type': 'avro'}
properties = avro.to_properties()
self.assertEqual(expected, properties)