proxy: more await unit tests

test FASTGOOD and some set scenarios

2 files changed, 113 insertions, 1 deletions

diff --git a/t/proxyunits.lua b/t/proxyunits.lua
index 69878f4..8c516d7 100644
--- a/t/proxyunits.lua
+++ b/t/proxyunits.lua
@@ -219,6 +219,47 @@ function mcp_config_routes(zones)
         return rtable[1]
     end
 
+    -- testing different styles of building the table argument for mcp.await()
+    pfx_get["awaitfastgood"] = function(r)
+        local all_zones = {}
+        for k, v in pairs(zones) do
+            all_zones[k] = v
+        end
+
+        local restable = mcp.await(r, all_zones, 2, mcp.AWAIT_FASTGOOD)
+
+        local final_res = restable[1]
+        local count = 0
+        for _, res in pairs(restable) do
+            if res:hit() then
+                final_res = res
+            end
+            count = count + 1
+        end
+
+        return final_res
+    end
+
+    pfx_set["awaitfastgood"] = function(r)
+        local all_zones = {}
+        for _, v in pairs(zones) do
+            table.insert(all_zones, v)
+        end
+
+        local restable = mcp.await(r, all_zones, 2)
+        local count = 0
+        local good_res = restable[1]
+        for _, res in pairs(restable) do
+            if res:ok() then
+                good_res = res
+            end
+            count = count + 1
+        end
+
+        print("Set Response count: " .. count)
+        return good_res
+    end
+
     mcp.attach(mcp.CMD_GET, toproute_factory(pfx_get, "get"))
     mcp.attach(mcp.CMD_SET, toproute_factory(pfx_set, "set"))
     mcp.attach(mcp.CMD_TOUCH, toproute_factory(pfx_touch, "touch"))
diff --git a/t/proxyunits.t b/t/proxyunits.t
index b9b98a6..284cb09 100644
--- a/t/proxyunits.t
+++ b/t/proxyunits.t
@@ -8,6 +8,7 @@ use lib "$Bin/lib";
 use Carp qw(croak);
 use MemcachedTest;
 use IO::Socket qw(AF_INET SOCK_STREAM);
+use IO::Select;
 
 if (!supports_proxy()) {
     plan skip_all => 'proxy not enabled';
@@ -606,7 +607,77 @@ check_version($ps);
     # more AWAIT_FIRST tests? to see how much it waits on/etc.
     # await(r, p, 2, mcp.AWAIT_FASTGOOD)
     # - should return 1 res on good, else wait for N non-error responses
-    # - test three pools, but third returns good. should have returned already
+    $key = "/awaitfastgood/a";
+    $cmd = "get $key\r\n";
+    print $ps $cmd;
+    my $fbe = $mbe[0];
+    is(scalar <$fbe>, $cmd, "awaitfastgood backend req");
+    print $fbe "VALUE $key 0 2\r\nok\r\nEND\r\n";
+    # Should have response after the first hit.
+    is(scalar <$ps>, "VALUE $key 0 2\r\n", "response from await");
+    is(scalar <$ps>, "ok\r\n", "await value");
+    is(scalar <$ps>, "END\r\n", "end from await");
+    for my $be ($mbe[1], $mbe[2]) {
+        is(scalar <$be>, $cmd, "awaitfastgood backend req");
+        print $be "VALUE $key 0 2\r\nok\r\nEND\r\n";
+    }
+
+    # test three pools, second response returns good. should have a hit.
+    print $ps $cmd;
+    for my $be (@mbe) {
+        is(scalar <$be>, $cmd, "awaitfastgood backend req");
+    }
+    $fbe = $mbe[0];
+    print $fbe "END\r\n";
+    $fbe = $mbe[1];
+    print $fbe "VALUE $key 0 2\r\nun\r\nEND\r\n";
+    is(scalar <$ps>, "VALUE $key 0 2\r\n", "response from await");
+    is(scalar <$ps>, "un\r\n", "await value");
+    is(scalar <$ps>, "END\r\n", "end from await");
+    $fbe = $mbe[2];
+    print $fbe "END\r\n";
+
+    # test three pools, but third returns good. should have returned already
+    print $ps $cmd;
+    for my $be ($mbe[0], $mbe[1]) {
+        is(scalar <$be>, $cmd, "awaitfastgood backend req");
+        print $be "END\r\n";
+    }
+    $fbe = $mbe[2];
+    is(scalar <$fbe>, $cmd, "awaitfastgood backend req");
+    print $fbe "VALUE $key 0 2\r\nnu\r\nEND\r\n";
+    is(scalar <$ps>, "END\r\n", "miss from awaitfastgood");
+
+    # Testing a set related to fastgood. waiting for two responses.
+    $cmd = "set $key 0 0 2\r\nmo\r\n";
+    print $ps $cmd;
+    for my $be ($mbe[0], $mbe[1]) {
+        is(scalar <$be>, "set $key 0 0 2\r\n", "set backend req");
+        is(scalar <$be>, "mo\r\n", "set backend data");
+        print $be "STORED\r\n";
+    }
+    is(scalar <$ps>, "STORED\r\n", "got stored from await");
+    $fbe = $mbe[2];
+    is(scalar <$fbe>, "set $key 0 0 2\r\n", "set backend req");
+    is(scalar <$fbe>, "mo\r\n", "set backend data");
+
+    # Testing another set; ensure it isn't returning early.
+    my $s = IO::Select->new();
+    $s->add($ps);
+    print $ps $cmd;
+    for my $be (@mbe) {
+        is(scalar <$be>, "set $key 0 0 2\r\n", "set backend req");
+        is(scalar <$be>, "mo\r\n", "set backend data");
+    }
+    $fbe = $mbe[0];
+    print $fbe "STORED\r\n";
+    my @readable = $s->can_read(0.25);
+    is(scalar @readable, 0, "set doesn't return early");
+    for my $be ($mbe[1], $mbe[2]) {
+        print $be "STORED\r\n";
+    }
+    is(scalar <$ps>, "STORED\r\n", "set completed normally");
+
     # await(r, p, 1, mcp.AWAIT_BACKGROUND) - ensure res without waiting
     $key = "/awaitbg/a";
     $cmd = "get $key\r\n";