@Override public SimpleField newMessage() { return new SimpleField(); }
public Map<String,String> getMarkings() { this.assureMarkings(); return markings; }
basicExpects(d, key, entry); EasyMock.expect(mockResponseFactory.getField()).andReturn(new SimpleField()); EasyMock.expect(mockResponseFactory.getField()).andReturn(new SimpleField()); foundFields.add(field.getName()); if (field.getName().equals("field1")) { Assert.assertEquals("5", field.getValueString()); } else { Assert.assertEquals("6", 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()); }
public void mergeFrom(Input input, SimpleEvent message) throws IOException { int number; while ((number = input.readFieldNumber(this)) != 0) { switch (number) { case 1: message.metadata = input.mergeObject(null, Metadata.getSchema()); break; case 2: if (message.fields == null) message.fields = new ArrayList<>(); SimpleField f = input.mergeObject(null, SimpleField.getSchema()); message.fields.add(f); break; case 3: if (message.markings == null) message.markings = Maps.newHashMap(); input.mergeObject(message.markings, MapSchema.SCHEMA); default: input.handleUnknownField(number, this); break; } } }
basicExpects(d, key, entry); EasyMock.expect(mockResponseFactory.getField()).andReturn(new SimpleField()); foundFields.add(field.getName()); Assert.assertEquals("5", field.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()); }
public void writeTo(Output output, SimpleEvent message) throws IOException { if (message.metadata != null) { output.writeObject(1, message.metadata, Metadata.getSchema(), false); } if (message.fields != null) { for (SimpleField field : message.fields) { if (field != null) output.writeObject(2, field, SimpleField.getSchema(), true); } } if (message.markings != null) { output.writeObject(3, message.markings, MapSchema.SCHEMA, false); } }
@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()); }
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); }