use of com.yahoo.document.Document in project vespa by vespa-engine.
the class JsonReaderTestCase method emptyDocTest.
@Test
public final void emptyDocTest() throws IOException {
InputStream rawDoc = new ByteArrayInputStream(Utf8.toBytes("{\"put\": \"id:unittest:smoke::whee\"," + " \"fields\": {}}"));
JsonReader r = new JsonReader(types, rawDoc, parserFactory);
DocumentParseInfo parseInfo = r.parseDocument().get();
DocumentType docType = r.readDocumentType(parseInfo.documentId);
DocumentPut put = new DocumentPut(new Document(docType, parseInfo.documentId));
new VespaJsonDocumentReader().readPut(parseInfo.fieldsBuffer, put);
assertEquals("id:unittest:smoke::whee", parseInfo.documentId.toString());
}
use of com.yahoo.document.Document in project vespa by vespa-engine.
the class JsonReaderTestCase method testMapStringToArrayOfInt.
@Test
public final void testMapStringToArrayOfInt() throws IOException {
InputStream rawDoc = new ByteArrayInputStream(Utf8.toBytes("{\"put\": \"id:unittest:testMapStringToArrayOfInt::whee\"," + " \"fields\": { \"actualMapStringToArrayOfInt\": { \"bamse\": [1, 2, 3] }}}"));
JsonReader r = new JsonReader(types, rawDoc, parserFactory);
DocumentParseInfo parseInfo = r.parseDocument().get();
DocumentType docType = r.readDocumentType(parseInfo.documentId);
DocumentPut put = new DocumentPut(new Document(docType, parseInfo.documentId));
new VespaJsonDocumentReader().readPut(parseInfo.fieldsBuffer, put);
Document doc = put.getDocument();
FieldValue f = doc.getFieldValue("actualMapStringToArrayOfInt");
assertSame(MapFieldValue.class, f.getClass());
MapFieldValue<?, ?> m = (MapFieldValue<?, ?>) f;
Array<?> a = (Array<?>) m.get(new StringFieldValue("bamse"));
assertEquals(3, a.size());
assertEquals(new IntegerFieldValue(1), a.get(0));
assertEquals(new IntegerFieldValue(2), a.get(1));
assertEquals(new IntegerFieldValue(3), a.get(2));
}
use of com.yahoo.document.Document in project vespa by vespa-engine.
the class JsonReaderTestCase method assertTensorField.
private static Tensor assertTensorField(String expectedTensor, DocumentPut put, String tensorFieldName) {
final Document doc = put.getDocument();
assertEquals("testtensor", doc.getId().getDocType());
assertEquals(TENSOR_DOC_ID, doc.getId().toString());
TensorFieldValue fieldValue = (TensorFieldValue) doc.getFieldValue(doc.getField(tensorFieldName));
assertEquals(Tensor.from(expectedTensor), fieldValue.getTensor().get());
return fieldValue.getTensor().get();
}
use of com.yahoo.document.Document in project vespa by vespa-engine.
the class JsonReaderTestCase method testParsingWithoutTensorField.
@Test
public void testParsingWithoutTensorField() {
Document doc = createPutWithoutTensor().getDocument();
assertEquals("testtensor", doc.getId().getDocType());
assertEquals("id:unittest:testtensor::0", doc.getId().toString());
TensorFieldValue fieldValue = (TensorFieldValue) doc.getFieldValue(doc.getField("mappedtensorfield"));
assertNull(fieldValue);
}
use of com.yahoo.document.Document in project vespa by vespa-engine.
the class ReferenceFieldValueSerializationTestCase method empty_reference_serialization_matches_cpp.
@Test
public void empty_reference_serialization_matches_cpp() throws IOException {
final Fixture fixture = new Fixture();
final Document document = fixture.createDocumentWithReference(fixture.createEmptyReferenceFieldValue());
SerializationTestUtils.assertSerializationMatchesCpp(Fixture.CROSS_LANGUAGE_PATH, "empty_reference", document, fixture.documentFactory);
}
Aggregations