Examples with InsecureShiftCodec org.apache.gobblin.test.crypto.InsecureShiftCodec used on opensource projects

Example 1 with InsecureShiftCodec

use of org.apache.gobblin.test.crypto.InsecureShiftCodec in project incubator-gobblin by apache.

the class AvroStringFieldEncryptorConverterTest method testNestedConversion.

@Test
public void testNestedConversion() throws DataConversionException, IOException, SchemaConversionException {
    AvroStringFieldEncryptorConverter converter = new AvroStringFieldEncryptorConverter();
    WorkUnitState wuState = new WorkUnitState();
    wuState.getJobState().setProp("converter.fieldsToEncrypt", "nestedRecords.*.fieldToEncrypt");
    wuState.getJobState().setProp("converter.encrypt.algorithm", "insecure_shift");
    converter.init(wuState);
    GenericRecord inputRecord = getRecordFromFile(getClass().getClassLoader().getResource("record_with_arrays.avro").getPath());
    Schema inputSchema = inputRecord.getSchema();
    Schema outputSchema = converter.convertSchema(inputSchema, wuState);
    List<String> origValues = new ArrayList<>();
    for (Object o : (List) inputRecord.get("nestedRecords")) {
        GenericRecord r = (GenericRecord) o;
        origValues.add(r.get("fieldToEncrypt").toString());
    }
    Iterable<GenericRecord> recordIt = converter.convertRecord(outputSchema, inputRecord, wuState);
    GenericRecord record = recordIt.iterator().next();
    Assert.assertEquals(outputSchema, inputSchema);
    List<String> decryptedValues = new ArrayList<>();
    for (Object o : (List) record.get("nestedRecords")) {
        GenericRecord r = (GenericRecord) o;
        String encryptedValue = r.get("fieldToEncrypt").toString();
        InsecureShiftCodec codec = new InsecureShiftCodec(Maps.<String, Object>newHashMap());
        InputStream in = codec.decodeInputStream(new ByteArrayInputStream(encryptedValue.getBytes(StandardCharsets.UTF_8)));
        byte[] decryptedValue = new byte[in.available()];
        in.read(decryptedValue);
        decryptedValues.add(new String(decryptedValue, StandardCharsets.UTF_8));
    }
    Assert.assertEquals(decryptedValues, origValues);
}

Example 2 with InsecureShiftCodec

use of org.apache.gobblin.test.crypto.InsecureShiftCodec in project incubator-gobblin by apache.

the class AvroStringFieldEncryptorConverterTest method testConversion.

@Test
public void testConversion() throws DataConversionException, IOException, SchemaConversionException {
    AvroStringFieldEncryptorConverter converter = new AvroStringFieldEncryptorConverter();
    WorkUnitState wuState = new WorkUnitState();
    wuState.getJobState().setProp("converter.fieldsToEncrypt", "field1");
    wuState.getJobState().setProp("converter.encrypt.algorithm", "insecure_shift");
    converter.init(wuState);
    GenericRecord inputRecord = TestUtils.generateRandomAvroRecord();
    Schema inputSchema = inputRecord.getSchema();
    Schema outputSchema = converter.convertSchema(inputSchema, wuState);
    String fieldValue = (String) inputRecord.get("field1");
    Iterable<GenericRecord> recordIt = converter.convertRecord(outputSchema, inputRecord, wuState);
    GenericRecord record = recordIt.iterator().next();
    Assert.assertEquals(outputSchema, inputSchema);
    String encryptedValue = (String) record.get("field1");
    InsecureShiftCodec codec = new InsecureShiftCodec(Maps.<String, Object>newHashMap());
    InputStream in = codec.decodeInputStream(new ByteArrayInputStream(encryptedValue.getBytes(StandardCharsets.UTF_8)));
    byte[] decryptedValue = new byte[in.available()];
    in.read(decryptedValue);
    Assert.assertEquals(new String(decryptedValue, StandardCharsets.UTF_8), fieldValue);
}