aboutsummaryrefslogtreecommitdiffstats
path: root/vbench/src/tests/app_dumpurl/app_dumpurl_test.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'vbench/src/tests/app_dumpurl/app_dumpurl_test.cpp')
-rw-r--r--vbench/src/tests/app_dumpurl/app_dumpurl_test.cpp50
1 files changed, 50 insertions, 0 deletions
diff --git a/vbench/src/tests/app_dumpurl/app_dumpurl_test.cpp b/vbench/src/tests/app_dumpurl/app_dumpurl_test.cpp
new file mode 100644
index 00000000000..48a8ce156f0
--- /dev/null
+++ b/vbench/src/tests/app_dumpurl/app_dumpurl_test.cpp
@@ -0,0 +1,50 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+#include <vespa/vespalib/testkit/testapp.h>
+#include <vbench/test/all.h>
+#include <vespa/vespalib/util/slaveproc.h>
+
+using namespace vbench;
+using vespalib::SlaveProc;
+
+bool endsWith(const Memory &mem, const string &str) {
+ return (mem.size < str.size()) ? false
+ : (strncmp(mem.data + mem.size - str.size(), str.data(), str.size()) == 0);
+}
+
+void readUntil(Input &input, SimpleBuffer &buffer, const string &end) {
+ ByteInput in(input, 256);
+ while (!endsWith(buffer.get(), end)) {
+ int c = in.get();
+ if (c < 0) {
+ return;
+ }
+ buffer.reserve(1).data[0] = c;
+ buffer.commit(1, 0);
+ }
+}
+
+TEST("dumpurl usage") {
+ std::string out;
+ EXPECT_FALSE(SlaveProc::run("../../apps/dumpurl/vbench_dumpurl_app", out));
+ fprintf(stderr, "%s\n", out.c_str());
+}
+
+TEST_MT_F("run dumpurl", 2, ServerSocket()) {
+ if (thread_id == 0) {
+ Stream::UP stream = f1.accept();
+ SimpleBuffer ignore;
+ readUntil(*stream, ignore, "\r\n\r\n");
+ BufferedOutput out(*stream, 256);
+ out.append("HTTP/1.1 200\r\n");
+ out.append("content-length: 4\r\n");
+ out.append("\r\n");
+ out.append("data");
+ } else {
+ std::string out;
+ EXPECT_TRUE(SlaveProc::run(strfmt("../../apps/dumpurl/vbench_dumpurl_app localhost %d /foo",
+ f1.port()).c_str(), out));
+ fprintf(stderr, "%s\n", out.c_str());
+ }
+}
+
+TEST_MAIN_WITH_PROCESS_PROXY() { TEST_RUN_ALL(); }