Search in sources :

Example 1 with SetFieldOp

use of com.linkedin.data.transform.patch.request.SetFieldOp in project rest.li by linkedin.

the class TestPatchTreeRecorder method testComplexDeepSetAndRemoves.

@Test
public void testComplexDeepSetAndRemoves() {
    PatchTreeRecorder<PatchTreeTestModel> pc = makeOne();
    PatchTreeTestModel testModel = pc.getRecordingProxy();
    PatchTreeTestModel.FooUnion fooUnion = new PatchTreeTestModel.FooUnion();
    fooUnion.setInt(10);
    testModel.setFooRequired(100).setFooUnion(fooUnion).setFooOptional(null, SetMode.REMOVE_IF_NULL);
    testModel.getFooRecordTemplate().setBar(9001l);
    // GetMode should be irrelevant
    testModel.getFooRecordTemplate(GetMode.DEFAULT).setBaz(null, SetMode.REMOVE_IF_NULL);
    PatchTree ptExpect = PatchCreator.diff(new DataMap(), new PatchTreeTestModel().setFooRequired(100).setFooUnion(fooUnion).data());
    // Augment the patch request with the removes in the same order so we get the same patch request.
    ptExpect.addOperation(PatchTreeTestModel.fields().fooOptional(), new RemoveFieldOp());
    ptExpect.addOperation(PatchTreeTestModel.fields().fooRecordTemplate().bar(), new SetFieldOp(9001l));
    ptExpect.addOperation(PatchTreeTestModel.fields().fooRecordTemplate().baz(), new RemoveFieldOp());
    Assert.assertEquals(pc.generatePatchTree().getDataMap(), ptExpect.getDataMap());
}
Also used : SetFieldOp(com.linkedin.data.transform.patch.request.SetFieldOp) PatchTreeTestModel(com.linkedin.restli.client.util.test.PatchTreeTestModel) RemoveFieldOp(com.linkedin.data.transform.patch.request.RemoveFieldOp) PatchTree(com.linkedin.data.transform.patch.request.PatchTree) DataMap(com.linkedin.data.DataMap) Test(org.testng.annotations.Test)

Aggregations

DataMap (com.linkedin.data.DataMap)1 PatchTree (com.linkedin.data.transform.patch.request.PatchTree)1 RemoveFieldOp (com.linkedin.data.transform.patch.request.RemoveFieldOp)1 SetFieldOp (com.linkedin.data.transform.patch.request.SetFieldOp)1 PatchTreeTestModel (com.linkedin.restli.client.util.test.PatchTreeTestModel)1 Test (org.testng.annotations.Test)1