Publish

1 files changed, 246 insertions, 0 deletions

diff --git a/fnet/src/tests/examples/examples_test.cpp b/fnet/src/tests/examples/examples_test.cpp
new file mode 100644
index 00000000000..5debd969a60
--- /dev/null
+++ b/fnet/src/tests/examples/examples_test.cpp
@@ -0,0 +1,246 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+#include <vespa/vespalib/testkit/test_kit.h>
+#include <vespa/vespalib/util/slaveproc.h>
+#include <vespa/vespalib/util/stringfmt.h>
+#include <vespa/vespalib/util/thread.h>
+
+// reserved in vespa/factory/doc/port-ranges.txt
+static const int PORT0 = 18570;
+static const int PORT1 = 18571;
+static const int PORT2 = 18572;
+static const int PORT3 = 18573;
+static const int PORT4 = 18574;
+static const int PORT5 = 18575;
+static const int PORT6 = 18576;
+static const int PORT7 = 18577;
+static const int PORT8 = 18578;
+static const int PORT9 = 18579;
+
+using vespalib::SlaveProc;
+
+bool runProc(SlaveProc &proc, bool &done) {
+    char buf[4096];
+    proc.close(); // close stdin
+    while (proc.running() && !done) {
+        if (!proc.eof()) {
+            uint32_t res = proc.read(buf, sizeof(buf), 10);
+            std::string tmp(buf, res);
+            fprintf(stderr, "%s", tmp.c_str());
+        }
+        vespalib::Thread::sleep(10);
+    }
+    if (done && proc.running()) {
+        kill(proc.getPid(), SIGTERM);
+        return proc.wait(60000);
+    }
+    return !proc.failed();
+}
+
+bool runProc(const std::string &cmd) {
+    bool ok = false;
+    for (size_t retry = 0; !ok && retry < 60; ++retry) {
+        if (retry > 0) {
+            fprintf(stderr, "retrying command in 500ms...\n");
+            vespalib::Thread::sleep(500);
+        }
+        bool done = false;
+        SlaveProc proc(cmd.c_str());
+        ok = runProc(proc, done);
+    }
+    return ok;
+}
+
+TEST("usage") {
+    bool done = false;
+    {
+        SlaveProc proc("../../examples/proxy/fnet_proxy_app");
+        EXPECT_FALSE(runProc(proc, done));
+    }
+    {
+        SlaveProc proc("../../examples/ping/fnet_pingserver_app");
+        EXPECT_FALSE(runProc(proc, done));
+    }
+    {
+        SlaveProc proc("../../examples/ping/fnet_pingclient_app");
+        EXPECT_FALSE(runProc(proc, done));
+    }
+    {
+        SlaveProc proc("../../examples/frt/rpc/fnet_rpc_client_app");
+        EXPECT_FALSE(runProc(proc, done));
+    }
+    {
+        SlaveProc proc("../../examples/frt/rpc/fnet_rpc_server_app");
+        EXPECT_FALSE(runProc(proc, done));
+    }
+    {
+        SlaveProc proc("../../examples/frt/rpc/fnet_echo_client_app");
+        EXPECT_FALSE(runProc(proc, done));
+    }
+    {
+        SlaveProc proc("../../examples/frt/rpc/rpc_info");
+        EXPECT_FALSE(runProc(proc, done));
+    }
+    {
+        SlaveProc proc("../../examples/frt/rpc/rpc_invoke");
+        EXPECT_FALSE(runProc(proc, done));
+    }
+    {
+        SlaveProc proc("../../examples/frt/rpc/fnet_rpc_callback_server_app");
+        EXPECT_FALSE(runProc(proc, done));
+    }
+    {
+        SlaveProc proc("../../examples/frt/rpc/fnet_rpc_callback_client_app");
+        EXPECT_FALSE(runProc(proc, done));
+    }
+    {
+        SlaveProc proc("../../examples/frt/rpc/rpc_proxy");
+        EXPECT_FALSE(runProc(proc, done));
+    }
+}
+
+TEST("timeout") {
+    std::string out;
+    EXPECT_TRUE(SlaveProc::run("../../examples/timeout/fnet_timeout_app", out));
+    fprintf(stderr, "%s\n", out.c_str());
+}
+
+TEST_MT_F("ping", 2, bool()) {
+    if (thread_id == 0) {
+        SlaveProc proc(vespalib::make_string("../../examples/ping/fnet_pingserver_app tcp/%d",
+                                             PORT0).c_str());
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(proc, f1));
+    } else {
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(vespalib::make_string("../../examples/ping/fnet_pingclient_app tcp/localhost:%d",
+                                                  PORT0).c_str()));
+        f1 = true;
+    }
+}
+
+TEST_MT_F("ping times out", 2, bool()) {
+    if (thread_id == 0) {
+        SlaveProc proc(vespalib::make_string("../../examples/frt/rpc/fnet_rpc_server_app tcp/%d",
+                                             PORT0).c_str());
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(proc, f1));
+    } else {
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(vespalib::make_string("../../examples/ping/fnet_pingclient_app tcp/localhost:%d",
+                                                  PORT0).c_str()));
+        f1 = true;
+    }
+}
+
+TEST_MT_F("ping with proxy", 3, bool()) {
+    if (thread_id == 0) {
+        SlaveProc proc(vespalib::make_string("../../examples/ping/fnet_pingserver_app tcp/%d",
+                                             PORT0).c_str());
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(proc, f1));
+    } else if (thread_id == 1) {
+        SlaveProc proc(vespalib::make_string("../../examples/proxy/fnet_proxy_app tcp/%d tcp/localhost:%d",
+                                             PORT1, PORT0).c_str());
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(proc, f1));
+    } else {
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(vespalib::make_string("../../examples/ping/fnet_pingclient_app tcp/localhost:%d",
+                                                  PORT1).c_str()));
+        f1 = true;
+    }
+}
+
+TEST_MT_F("rpc client server", 2, bool()) {
+    if (thread_id == 0) {
+        SlaveProc proc(vespalib::make_string("../../examples/frt/rpc/fnet_rpc_server_app tcp/%d",
+                                             PORT0).c_str());
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(proc, f1));
+    } else {
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(vespalib::make_string("../../examples/frt/rpc/fnet_rpc_client_app tcp/localhost:%d",
+                                                  PORT0).c_str()));
+        f1 = true;
+    }
+}
+
+TEST_MT_F("rpc echo client", 2, bool()) {
+    if (thread_id == 0) {
+        SlaveProc proc(vespalib::make_string("../../examples/frt/rpc/fnet_rpc_server_app tcp/%d",
+                                             PORT0).c_str());
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(proc, f1));
+    } else {
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(vespalib::make_string("../../examples/frt/rpc/fnet_echo_client_app tcp/localhost:%d",
+                                                  PORT0).c_str()));
+        f1 = true;
+    }
+}
+
+TEST_MT_F("rpc info", 2, bool()) {
+    if (thread_id == 0) {
+        SlaveProc proc(vespalib::make_string("../../examples/frt/rpc/fnet_rpc_server_app tcp/%d",
+                                             PORT0).c_str());
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(proc, f1));
+    } else {
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(vespalib::make_string("../../examples/frt/rpc/rpc_info tcp/localhost:%d",
+                                                  PORT0).c_str()));
+        EXPECT_TRUE(runProc(vespalib::make_string("../../examples/frt/rpc/rpc_info tcp/localhost:%d verbose",
+                                                  PORT0).c_str()));
+        f1 = true;
+    }
+}
+
+TEST_MT_F("rpc invoke", 2, bool()) {
+    if (thread_id == 0) {
+        SlaveProc proc(vespalib::make_string("../../examples/frt/rpc/fnet_rpc_server_app tcp/%d",
+                                             PORT0).c_str());
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(proc, f1));
+    } else {
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(vespalib::make_string("../../examples/frt/rpc/rpc_invoke tcp/localhost:%d frt.rpc.echo "
+                                                  "b:1 h:2 i:4 l:8 f:0.5 d:0.25 s:foo",
+                                                  PORT0).c_str()));
+        f1 = true;
+    }
+}
+
+TEST_MT_F("rpc callback client server", 2, bool()) {
+    if (thread_id == 0) {
+        SlaveProc proc(vespalib::make_string("../../examples/frt/rpc/fnet_rpc_callback_server_app tcp/%d",
+                                             PORT0).c_str());
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(proc, f1));
+    } else {
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(vespalib::make_string("../../examples/frt/rpc/fnet_rpc_callback_client_app tcp/localhost:%d",
+                                                  PORT0).c_str()));
+        f1 = true;
+    }
+}
+
+TEST_MT_F("rpc callback client server with proxy", 3, bool()) {
+    if (thread_id == 0) {
+        SlaveProc proc(vespalib::make_string("../../examples/frt/rpc/fnet_rpc_callback_server_app tcp/%d",
+                                             PORT0).c_str());
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(proc, f1));
+    } else if (thread_id == 1) {
+        SlaveProc proc(vespalib::make_string("../../examples/frt/rpc/rpc_proxy tcp/%d tcp/localhost:%d",
+                                             PORT1, PORT0).c_str());
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(proc, f1));
+    } else {
+        TEST_BARRIER();
+        EXPECT_TRUE(runProc(vespalib::make_string("../../examples/frt/rpc/fnet_rpc_callback_client_app tcp/localhost:%d",
+                                                  PORT1).c_str()));
+        f1 = true;
+    }
+}
+
+TEST_MAIN_WITH_PROCESS_PROXY() { TEST_RUN_ALL(); }