foundFields.add(field.getName()); Assert.assertEquals("5", field.getValueString());
@Test public void transform_primaryNoMatchSetTest() throws MarkingFunctions.Exception { Key key = new Key("shard", "dataType" + Constants.NULL + "uid"); Value value = new Value(); AbstractMap.SimpleEntry<Key,Value> entry = new AbstractMap.SimpleEntry<>(key, value); Map<String,List<String>> fieldMap = new HashMap<>(); List<String> fieldList = new ArrayList<>(); fieldList.add("field3"); fieldMap.put("field2", fieldList); Document d = new Document(); basicExpects(d, key, entry); PowerMock.replayAll(); transformer = new DocumentTransformer(mockLogic, mockQuery, mockMarkingFunctions, mockResponseFactory, true); transformer.setPrimaryToSecondaryFieldMap(fieldMap); SimpleEvent event = (SimpleEvent) transformer.transform(entry); PowerMock.verifyAll(); Assert.assertNotNull(event); Assert.assertEquals(1, event.getFields().size()); Assert.assertEquals("field1", event.getFields().get(0).getName()); Assert.assertEquals("5", event.getFields().get(0).getValueString()); }
@Test public void transform_primaryEmptySecondarySetTest() throws MarkingFunctions.Exception { Key key = new Key("shard", "dataType" + Constants.NULL + "uid"); Value value = new Value(); AbstractMap.SimpleEntry<Key,Value> entry = new AbstractMap.SimpleEntry<>(key, value); Map<String,List<String>> fieldMap = new HashMap<>(); List<String> fieldList = Collections.EMPTY_LIST; fieldMap.put("field2", fieldList); Document d = new Document(); basicExpects(d, key, entry); PowerMock.replayAll(); transformer = new DocumentTransformer(mockLogic, mockQuery, mockMarkingFunctions, mockResponseFactory, true); transformer.setPrimaryToSecondaryFieldMap(fieldMap); SimpleEvent event = (SimpleEvent) transformer.transform(entry); PowerMock.verifyAll(); Assert.assertNotNull(event); Assert.assertEquals(1, event.getFields().size()); Assert.assertEquals("field1", event.getFields().get(0).getName()); Assert.assertEquals("5", event.getFields().get(0).getValueString()); }
@Test public void transform_noPrimaryToSecondaryMapSetTest() throws MarkingFunctions.Exception { Key key = new Key("shard", "dataType" + Constants.NULL + "uid"); Value value = new Value(); AbstractMap.SimpleEntry<Key,Value> entry = new AbstractMap.SimpleEntry<>(key, value); Document d = new Document(); basicExpects(d, key, entry); PowerMock.replayAll(); transformer = new DocumentTransformer(mockLogic, mockQuery, mockMarkingFunctions, mockResponseFactory, true); SimpleEvent event = (SimpleEvent) transformer.transform(entry); PowerMock.verifyAll(); Assert.assertNotNull(event); Assert.assertEquals(1, event.getFields().size()); Assert.assertEquals("field1", event.getFields().get(0).getName()); Assert.assertEquals("5", event.getFields().get(0).getValueString()); }