public SimpleEvent newMessage() { return new SimpleEvent(); }
@Override public String toString() { return getMarkings() + ": " + this.fields; }
@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()); }
@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()); }
private void basicExpects(Document d, Key key, Map.Entry<Key,Value> entry) throws MarkingFunctions.Exception { // AbstractMap.SimpleEntry<Key, Document> documentEntry = new AbstractMap.SimpleEntry<>(key, mockDocument); d.put("field1", new Numeric("5", key, true)); AbstractMap.SimpleEntry<Key,Document> documentEntry = new AbstractMap.SimpleEntry<>(key, d); // Map<String,Attribute<?extends Comparable<?>>> dictionary = new HashMap<>(); // dictionary.put("field1", mockNumeric); SimpleField simpleField = new SimpleField(); SimpleEvent simpleEvent = new SimpleEvent(); PowerMock.mockStatic(DocumentSerialization.class); EasyMock.expect(DocumentSerialization.getDocumentDeserializer(mockQuery)).andReturn(mockDeserializer); EasyMock.expect(mockLogic.getTableName()).andReturn("table1"); EasyMock.expect(mockQuery.getQueryAuthorizations()).andReturn("A,B,C"); EasyMock.expect(mockQuery.findParameter("log.timing.details")).andReturn(new QueryImpl.Parameter("", "")); EasyMock.expect(mockDeserializer.apply(entry)).andReturn(documentEntry); // EasyMock.expect(mockDocument.getDictionary()).andReturn(Collections.EMPTY_MAP); // mockDocument.debugDocumentSize(key); EasyMock.expect(mockMarkingFunctions.translateFromColumnVisibility(key.getColumnVisibilityParsed())).andReturn(Collections.EMPTY_MAP); // EasyMock.expect(mockDocument.getDictionary()).andReturn(dictionary); // EasyMock.expect(mockNumeric.getData()).andReturn("5"); EasyMock.expect(mockResponseFactory.getField()).andReturn(simpleField); EasyMock.expect(mockResponseFactory.getEvent()).andReturn(simpleEvent); // EasyMock.expect(mockDocument.sizeInBytes()).andReturn(1l); }
@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()); }