bin: consolidate clients
[awesomized/libmemcached] / test / tests / bin / memflush.cpp
1 #include "test/lib/common.hpp"
2 #include "test/lib/Shell.hpp"
3 #include "test/lib/Server.hpp"
4 #include "test/lib/Retry.hpp"
5 #include "test/lib/ReturnMatcher.hpp"
6
7 using Catch::Matchers::Contains;
8
9 TEST_CASE("bin/memflush") {
10 Shell sh{string{TESTING_ROOT "/../src/bin"}};
11
12 SECTION("no servers provided") {
13 string output;
14 REQUIRE_FALSE(sh.run("memflush", output));
15 REQUIRE(output == "No Servers provided\n");
16 }
17
18 SECTION("--help") {
19 string output;
20 REQUIRE(sh.run("memflush --help", output));
21 REQUIRE_THAT(output, Contains("memflush"));
22 REQUIRE_THAT(output, Contains("v1"));
23 REQUIRE_THAT(output, Contains("help"));
24 REQUIRE_THAT(output, Contains("version"));
25 REQUIRE_THAT(output, Contains("option"));
26 REQUIRE_THAT(output, Contains("--"));
27 REQUIRE_THAT(output, Contains("="));
28 }
29
30 SECTION("with server") {
31 Server server{MEMCACHED_BINARY, {"-p", random_port_string}};
32 MemcachedPtr memc;
33 LoneReturnMatcher test{*memc};
34
35 REQUIRE(server.ensureListening());
36 auto port = get<int>(server.getSocketOrPort());
37 auto comm = "memflush --servers=localhost:" + to_string(port) + " ";
38
39 REQUIRE_SUCCESS(memcached_server_add(*memc, "localhost", port));
40
41 SECTION("okay") {
42
43 REQUIRE_SUCCESS(memcached_set(*memc, S("key1"), S("val1"), 0, 0));
44 REQUIRE_SUCCESS(memcached_set(*memc, S("key2"), S("val2"), 0, 0));
45
46 this_thread::sleep_for(500ms);
47
48 string output;
49 REQUIRE(sh.run(comm, output));
50 REQUIRE(output.empty());
51
52 memcached_return_t rc;
53 REQUIRE(nullptr == memcached_get(*memc, S("key1"), nullptr, nullptr, &rc));
54 REQUIRE_RC(MEMCACHED_NOTFOUND, rc);
55 REQUIRE(nullptr == memcached_get(*memc, S("key2"), nullptr, nullptr, &rc));
56 REQUIRE_RC(MEMCACHED_NOTFOUND, rc);
57 }
58
59 SECTION("connection failure") {
60 server.signal(SIGKILL);
61 server.wait();
62
63 string output;
64 REQUIRE_FALSE(sh.run(comm, output));
65 REQUIRE_THAT(output,
66 Contains("CONNECTION FAILURE")
67 || Contains("SERVER HAS FAILED")
68 || Contains("SYSTEM ERROR")
69 || Contains("TIMEOUT OCCURRED"));
70 }
71 }
72 }