airbytehq / json-avro-converter Goto Github PK

  @Test
  public void testJsonToAvroConverterBinary() throws JsonProcessingException {
    String base64String = "4E/QIOo6aRCi2AgAKzAwnQ=="; // Corresponds to Hex: e04fd020ea3a6910a2d808002b30309d
    // The following works because all bytes have a mapping when using UTF-8 encoding.
    // echo "r@nd0mT3xt" | base64 --> ckBuZDBtVDN4dAo=
    // String base64String = "ckBuZDBtVDN4dAo=";
    final JsonAvroConverter converter = JsonAvroConverter.builder()
            .build();
    final String avroSchema = """
      {
        "type": "record",
        "name": "test_schema",
        "fields": [
          {
            "name": "binary_field_base64",
            "type": "bytes"
          }
        ]
      }""";
    final Schema schema =  new Schema.Parser().parse(avroSchema);
    final JsonNode jsonObject = JsonHelper.deserialize(String.format("{\"binary_field_base64\":\"%s\"}", base64String));
    final GenericData.Record actualAvroObject = converter.convertToGenericDataRecord(WRITER.writeValueAsBytes(jsonObject), schema);

    java.nio.ByteBuffer retrievedByteBuffer = (java.nio.ByteBuffer) actualAvroObject.get("binary_field_base64");
    Assertions.assertArrayEquals(java.util.Base64.getDecoder().decode(base64String), retrievedByteBuffer.array());
  }