/**
* Tests method setRequestContent of HttpClient.
*
* @throws IOException I/O Exception
*/
@Test
public void writeMultipartMessage() throws IOException {
final HttpRequest req = new HttpRequest();
req.isMultipart = true;
req.payloadAttrs.put("media-type", "multipart/alternative");
req.payloadAttrs.put("boundary", "boundary42");
final Part p1 = new Part();
p1.headers.put("Content-Type", "text/plain; charset=us-ascii");
p1.bodyAttrs.put("media-type", "text/plain");
final String plain = "...plain text....";
p1.bodyContent.add(Str.get(plain + '\n'));
final Part p2 = new Part();
p2.headers.put("Content-Type", "text/richtext");
p2.bodyAttrs.put("media-type", "text/richtext");
final String rich = ".... richtext version...";
p2.bodyContent.add(Str.get(rich));
final Part p3 = new Part();
p3.headers.put("Content-Type", "text/x-whatever");
p3.bodyAttrs.put("media-type", "text/x-whatever");
final String fancy = ".... fanciest formatted version...";
p3.bodyContent.add(Str.get(fancy));
req.parts.add(p1);
req.parts.add(p2);
req.parts.add(p3);
final FakeHttpConnection fakeConn = new FakeHttpConnection(new URL("http://www.test.com"));
HttpClient.setRequestContent(fakeConn.getOutputStream(), req);
final String expResult =
"--boundary42"
+ CRLF
+ "Content-Type: text/plain; charset=us-ascii"
+ CRLF
+ CRLF
+ plain
+ Prop.NL
+ CRLF
+ "--boundary42"
+ CRLF
+ "Content-Type: text/richtext"
+ CRLF
+ CRLF
+ rich
+ CRLF
+ "--boundary42"
+ CRLF
+ "Content-Type: text/x-whatever"
+ CRLF
+ CRLF
+ fancy
+ CRLF
+ "--boundary42--"
+ CRLF;
// Compare results
assertEquals(expResult, fakeConn.getOutputStream().toString());
}
/**
* Tests parsing of multipart request when the contents for each part are set from the $bodies
* parameter.
*
* @throws IOException I/O Exception
* @throws QueryException query exception
*/
@Test
public void parseMultipartReqBodies() throws IOException, QueryException {
final String multiReq =
"<http:request "
+ "xmlns:http='http://expath.org/ns/http-client' "
+ "method='POST' href='"
+ REST_ROOT
+ "'>"
+ "<http:header name='hdr1' value='hdr1val'/>"
+ "<http:header name='hdr2' value='hdr2val'/>"
+ "<http:multipart media-type='multipart/mixed' boundary='xxxx'>"
+ "<http:header name='p1hdr1' value='p1hdr1val'/>"
+ "<http:header name='p1hdr2' value='p1hdr2val'/>"
+ "<http:body media-type='text/plain'/>"
+ "<http:header name='p2hdr1' value='p2hdr1val'/>"
+ "<http:body media-type='text/plain'/>"
+ "<http:body media-type='text/plain'/>"
+ "</http:multipart>"
+ "</http:request>";
final DBNode dbNode1 = new DBNode(new IOContent(multiReq));
final ItemList bodies = new ItemList();
bodies.add(Str.get("Part1"));
bodies.add(Str.get("Part2"));
bodies.add(Str.get("Part3"));
final HttpRequestParser rp = new HttpRequestParser(null);
final HttpRequest r = rp.parse(dbNode1.children().next(), bodies.iter());
assertEquals(2, r.attributes.size());
assertEquals(2, r.headers.size());
assertTrue(r.isMultipart);
assertEquals(3, r.parts.size());
// check parts
final Iterator<Part> i = r.parts.iterator();
Part part = i.next();
assertEquals(2, part.headers.size());
assertEquals(1, part.bodyContent.size());
assertEquals(1, part.bodyAttrs.size());
part = i.next();
assertEquals(1, part.headers.size());
assertEquals(1, part.bodyContent.size());
assertEquals(1, part.bodyAttrs.size());
part = i.next();
assertEquals(0, part.headers.size());
assertEquals(1, part.bodyContent.size());
assertEquals(1, part.bodyAttrs.size());
}
/**
* Runs a query with an external variable declaration.
*
* @throws IOException I/O exception
*/
@Test
public void queryBindSequence() throws IOException {
Query query = session.query("declare variable $a external; $a");
query.bind("a", "1\u00012", "xs:integer");
assertEqual("1", query.next());
assertEqual("2", query.next());
query.close();
query = session.query("declare variable $a external; $a");
query.bind("a", "09\u0002xs:hexBinary\u00012", "xs:integer");
assertEqual("09", query.next());
assertEqual("2", query.next());
query.close();
query = session.query("declare variable $a external; $a");
query.bind("a", Seq.get(new Item[] {Int.get(1), Str.get("X")}, 2));
assertEqual("1", query.next());
assertEqual("X", query.next());
query.close();
query = session.query("declare variable $a external; $a");
query.bind("a", IntSeq.get(new long[] {1, 2}, AtomType.INT));
assertEqual("1", query.next());
assertEqual("2", query.next());
query.close();
query = session.query("declare variable $a external; $a");
query.bind("a", IntSeq.get(new long[] {1, 2}, AtomType.INT), "xs:integer");
assertEqual("1", query.next());
assertEqual("2", query.next());
query.close();
}
/**
* Converts the specified result to an XQuery value.
*
* @param obj result object
* @param qc query context
* @param sc static context
* @return value
* @throws QueryException query exception
*/
public static Value toValue(final Object obj, final QueryContext qc, final StaticContext sc)
throws QueryException {
if (obj == null) return Empty.SEQ;
if (obj instanceof Value) return (Value) obj;
if (obj instanceof Iter) return ((Iter) obj).value();
// find XQuery mapping for specified type
final Type type = type(obj);
if (type != null) return type.cast(obj, qc, sc, null);
// primitive arrays
if (obj instanceof byte[]) return BytSeq.get((byte[]) obj);
if (obj instanceof long[]) return IntSeq.get((long[]) obj, AtomType.ITR);
if (obj instanceof char[]) return Str.get(new String((char[]) obj));
if (obj instanceof boolean[]) return BlnSeq.get((boolean[]) obj);
if (obj instanceof double[]) return DblSeq.get((double[]) obj);
if (obj instanceof float[]) return FltSeq.get((float[]) obj);
// no array: return Java type
if (!obj.getClass().isArray()) return new Jav(obj, qc);
// empty array
final int s = Array.getLength(obj);
if (s == 0) return Empty.SEQ;
// string array
if (obj instanceof String[]) {
final String[] r = (String[]) obj;
final byte[][] b = new byte[r.length][];
for (int v = 0; v < s; v++) b[v] = token(r[v]);
return StrSeq.get(b);
}
// character array
if (obj instanceof char[][]) {
final char[][] r = (char[][]) obj;
final byte[][] b = new byte[r.length][];
for (int v = 0; v < s; v++) b[v] = token(new String(r[v]));
return StrSeq.get(b);
}
// short array
if (obj instanceof short[]) {
final short[] r = (short[]) obj;
final long[] b = new long[r.length];
for (int v = 0; v < s; v++) b[v] = r[v];
return IntSeq.get(b, AtomType.SHR);
}
// integer array
if (obj instanceof int[]) {
final int[] r = (int[]) obj;
final long[] b = new long[r.length];
for (int v = 0; v < s; v++) b[v] = r[v];
return IntSeq.get(b, AtomType.INT);
}
// any other array (also nested ones)
final Object[] objs = (Object[]) obj;
final ValueBuilder vb = new ValueBuilder(objs.length);
for (final Object o : objs) vb.add(toValue(o, qc, sc));
return vb.value();
}
/**
* Tests writing of request content with different combinations of the body attributes media-type
* and method.
*
* @throws IOException IO exception
*/
@Test
public void writeMessage() throws IOException {
// Case 1: No method, media-type='text/xml'
final HttpRequest req1 = new HttpRequest();
final FakeHttpConnection fakeConn1 = new FakeHttpConnection(new URL("http://www.test.com"));
req1.payloadAttrs.put(SerializerOptions.MEDIA_TYPE.name(), "text/xml");
// Node child
final FElem e1 = new FElem("a").add("a");
req1.bodyContent.add(e1);
// String item child
req1.bodyContent.add(Str.get("<b>b</b>"));
HttpClient.setRequestContent(fakeConn1.getOutputStream(), req1);
assertEquals("<a>a</a><b>b</b>", fakeConn1.out.toString(Strings.UTF8));
// Case 2: No method, media-type='text/plain'
final HttpRequest req2 = new HttpRequest();
final FakeHttpConnection fakeConn2 = new FakeHttpConnection(new URL("http://www.test.com"));
req2.payloadAttrs.put(SerializerOptions.MEDIA_TYPE.name(), "text/plain");
// Node child
final FElem e2 = new FElem("a").add("a");
req2.bodyContent.add(e2);
// String item child
req2.bodyContent.add(Str.get("<b>b</b>"));
HttpClient.setRequestContent(fakeConn2.getOutputStream(), req2);
assertEquals("a<b>b</b>", fakeConn2.out.toString());
// Case 3: method='text', media-type='text/xml'
final HttpRequest req3 = new HttpRequest();
final FakeHttpConnection fakeConn3 = new FakeHttpConnection(new URL("http://www.test.com"));
req3.payloadAttrs.put(SerializerOptions.MEDIA_TYPE.name(), "text/xml");
req3.payloadAttrs.put("method", "text");
// Node child
final FElem e3 = new FElem("a").add("a");
req3.bodyContent.add(e3);
// String item child
req3.bodyContent.add(Str.get("<b>b</b>"));
HttpClient.setRequestContent(fakeConn3.getOutputStream(), req3);
assertEquals("a<b>b</b>", fakeConn3.out.toString());
}
@Override
public Item item(final QueryContext qc, final InputInfo ii) throws QueryException {
final Iter ir = iter(qc);
final Item it = ir.next();
if (it == null || ir.size() == 1) return it;
final Item n = ir.next();
if (n != null) {
final ValueBuilder vb = new ValueBuilder().add(it).add(n);
if (ir.next() != null) vb.add(Str.get(DOTS));
throw SEQFOUND_X.get(info, vb.value());
}
return it;
}
@Override
public final Item ebv(final QueryContext qc, final InputInfo ii) throws QueryException {
final Item it;
if (seqType().zeroOrOne()) {
it = item(qc, info);
} else {
final Iter ir = iter(qc);
it = ir.next();
if (it != null && !(it instanceof ANode)) {
final Item n = ir.next();
if (n != null) {
final ValueBuilder vb = new ValueBuilder().add(it).add(n);
if (ir.next() != null) vb.add(Str.get(DOTS));
throw EBV_X.get(info, vb.value());
}
}
}
return it == null ? Bln.FALSE : it;
}
/**
* Tests ResponseHandler.getResponse() with multipart response having preamble and epilogue.
*
* @throws IOException I/O Exception
* @throws Exception exception
*/
@Test
public void multipartRespPreamble() throws Exception {
// Create fake HTTP connection
final FakeHttpConnection conn = new FakeHttpConnection(new URL("http://www.test.com"));
final Map<String, List<String>> hdrs = new HashMap<>();
final List<String> fromVal = new ArrayList<>();
fromVal.add("Nathaniel Borenstein <*****@*****.**>");
// From: Nathaniel Borenstein <*****@*****.**>
hdrs.put("From", fromVal);
final List<String> mimeVal = new ArrayList<>();
mimeVal.add("1.0");
final List<String> toVal = new ArrayList<>();
toVal.add("Ned Freed <*****@*****.**>");
// To: Ned Freed <*****@*****.**>
hdrs.put("To", toVal);
// MIME-Version: 1.0
hdrs.put("MIME-version", mimeVal);
final List<String> subjVal = new ArrayList<>();
subjVal.add("Formatted text mail");
// Subject: Formatted text mail
hdrs.put("Subject", subjVal);
final List<String> contTypeVal = new ArrayList<>();
contTypeVal.add("multipart/mixed");
contTypeVal.add("boundary=\"simple boundary\"");
// Content-Type: multipart/alternative; boundary=boundary42
hdrs.put("Content-Type", contTypeVal);
conn.headers = hdrs;
conn.contentType = "multipart/mixed; boundary=\"simple boundary\"";
// Response to be read
conn.content =
token(
"This is the preamble. "
+ "It is to be ignored, though it"
+ NL
+ "is a handy place for mail composers to include an"
+ CRLF
+ "explanatory note to non-MIME compliant readers."
+ CRLF
+ "--simple boundary"
+ CRLF
+ CRLF
+ "This is implicitly typed plain ASCII text."
+ CRLF
+ "It does NOT end with a linebreak."
+ CRLF
+ "--simple boundary"
+ CRLF
+ "Content-type: text/plain; charset=us-ascii"
+ CRLF
+ CRLF
+ "This is explicitly typed plain ASCII text."
+ CRLF
+ "It DOES end with a linebreak."
+ CRLF
+ CRLF
+ "--simple boundary--"
+ CRLF
+ "This is the epilogue. It is also to be ignored.");
// Get response as sequence of XQuery items
final ItemList returned = new HttpResponse(null, ctx.options).getResponse(conn, true, null);
// Construct expected result
final ItemList expected = new ItemList();
final String response =
"<http:response "
+ "xmlns:http='http://expath.org/ns/http-client' "
+ "status='200' message='OK'>"
+ "<http:header name='Subject' value='Formatted text mail'/>"
+ "<http:header name='To' value='Ned "
+ "Freed <[email protected]>'/>"
+ "<http:header name='Content-Type' value='multipart/mixed;"
+ "boundary="simple boundary"'/>"
+ "<http:header name='MIME-version' value='1.0'/>"
+ "<http:header name='From' value='Nathaniel Borenstein "
+ "<[email protected]>'/>"
+ "<http:multipart media-type='multipart/mixed' "
+ "boundary='simple boundary'>"
+ "<http:body media-type='text/plain'/>"
+ "<http:header name='Content-type' value='text/plain; "
+ "charset=us-ascii'/>"
+ "<http:body media-type='text/plain; charset=us-ascii'/>"
+ "</http:multipart>"
+ "</http:response>";
expected.add(new DBNode(new IOContent(response)).children().next());
expected.add(
Str.get(
"This is implicitly typed plain ASCII text.\n"
+ "It does NOT end with a linebreak.\n"));
expected.add(
Str.get(
"This is explicitly typed plain ASCII text.\n" + "It DOES end with a linebreak.\n\n"));
compare(expected, returned);
}
/**
* Tests ResponseHandler.getResponse() with multipart response.
*
* @throws IOException I/O Exception
* @throws Exception exception
*/
@Test
public void multipartResponse() throws Exception {
// Create fake HTTP connection
final FakeHttpConnection conn = new FakeHttpConnection(new URL("http://www.test.com"));
final Map<String, List<String>> hdrs = new HashMap<>();
final List<String> fromVal = new ArrayList<>();
fromVal.add("Nathaniel Borenstein <*****@*****.**>");
// From: Nathaniel Borenstein <*****@*****.**>
hdrs.put("From", fromVal);
final List<String> mimeVal = new ArrayList<>();
mimeVal.add("1.0");
// MIME-Version: 1.0
hdrs.put("MIME-version", mimeVal);
final List<String> subjVal = new ArrayList<>();
subjVal.add("Formatted text mail");
// Subject: Formatted text mail
hdrs.put("Subject", subjVal);
final List<String> contTypeVal = new ArrayList<>();
contTypeVal.add("multipart/alternative");
contTypeVal.add("boundary=\"boundary42\"");
// Content-Type: multipart/alternative; boundary=boundary42
hdrs.put("Content-Type", contTypeVal);
conn.headers = hdrs;
conn.contentType = "multipart/alternative; boundary=\"boundary42\"";
conn.content =
token(
"--boundary42"
+ CRLF
+ "Content-Type: text/plain; charset=us-ascii"
+ CRLF
+ CRLF
+ "...plain text...."
+ CRLF
+ CRLF
+ "--boundary42"
+ CRLF
+ "Content-Type: text/richtext"
+ CRLF
+ CRLF
+ ".... richtext..."
+ CRLF
+ "--boundary42"
+ CRLF
+ "Content-Type: text/x-whatever"
+ CRLF
+ CRLF
+ ".... fanciest formatted version "
+ CRLF
+ "..."
+ CRLF
+ "--boundary42--");
final ItemList returned = new HttpResponse(null, ctx.options).getResponse(conn, true, null);
// Construct expected result
final ItemList expected = new ItemList();
final String response =
"<http:response "
+ "xmlns:http='http://expath.org/ns/http-client' "
+ "status='200' message='OK'>"
+ "<http:header name='Subject' value='Formatted text mail'/>"
+ "<http:header name='Content-Type' "
+ "value='multipart/alternative;boundary="boundary42"'/>"
+ "<http:header name='MIME-version' value='1.0'/>"
+ "<http:header name='From' value='Nathaniel Borenstein "
+ "<[email protected]>'/>"
+ "<http:multipart media-type='multipart/alternative' "
+ "boundary='boundary42'>"
+ "<http:header name='Content-Type' "
+ "value='text/plain; charset=us-ascii'/>"
+ "<http:body media-type='text/plain; charset=us-ascii'/>"
+ "<http:header name='Content-Type' value='text/richtext'/>"
+ "<http:body media-type='text/richtext'/>"
+ "<http:header name='Content-Type' value='text/x-whatever'/>"
+ "<http:body media-type='text/x-whatever'/>"
+ "</http:multipart>"
+ "</http:response> ";
expected.add(new DBNode(new IOContent(response)).children().next());
expected.add(Str.get("...plain text....\n\n"));
expected.add(Str.get(".... richtext...\n"));
expected.add(Str.get(".... fanciest formatted version \n...\n"));
compare(expected, returned);
}
/**
* Returns the WKT format of a geometry.
*
* @param node xml element containing gml object(s)
* @return Well-Known Text geometry representation
* @throws QueryException query exception
*/
@Deterministic
public Str asText(final ANode node) throws QueryException {
return Str.get(new WKTWriter().write(checkGeo(node)));
}
/**
* Returns a database function for the first node in a node set.
*
* @param n node set
* @param i offset
* @return function string
*/
static String openPre(final Nodes n, final int i) {
return Function._DB_OPEN_PRE.get(Str.get(n.data.meta.name), Int.get(n.pres[i])).toString();
}
@Override
public final Item atomItem(final InputInfo ii) {
return type == NodeType.PI || type == NodeType.COM ? Str.get(string()) : new Atm(string());
}
/**
* Extracts text entries.
*
* @param ctx query context
* @return text entry
* @throws QueryException query exception
*/
private ValueBuilder extractText(final QueryContext ctx) throws QueryException {
final String enc = encoding(2, ARCH_ENCODING, ctx);
final ValueBuilder vb = new ValueBuilder();
for (final byte[] b : extract(ctx)) vb.add(Str.get(encode(b, enc)));
return vb;
}