# vim:set ft= ts=4 sw=4 et fdm=marker:
use lib 'lib';
use Test::Nginx::Socket::Lua;

#worker_connections(1014);
#master_process_enabled(1);
#log_level('warn');

#repeat_each(2);

plan tests => repeat_each() * (blocks() * 3 + 17);

#no_diff();
no_long_string();
#master_on();
#workers(2);

run_tests();

__DATA__

=== TEST 1: string key, int value
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32)
            dogs:set("bah", 10502)
            local val = dogs:get("foo")
            ngx.say(val, " ", type(val))
            val = dogs:get("bah")
            ngx.say(val, " ", type(val))
        ';
    }
--- request
GET /test
--- response_body
32 number
10502 number
--- no_error_log
[error]



=== TEST 2: string key, floating-point value
--- http_config
    lua_shared_dict cats 1m;
--- config
    location = /test {
        content_by_lua '
            local cats = ngx.shared.cats
            cats:set("foo", 3.14159)
            cats:set("baz", 1.28)
            cats:set("baz", 3.96)
            local val = cats:get("foo")
            ngx.say(val, " ", type(val))
            val = cats:get("baz")
            ngx.say(val, " ", type(val))
        ';
    }
--- request
GET /test
--- response_body
3.14159 number
3.96 number
--- no_error_log
[error]



=== TEST 3: string key, boolean value
--- http_config
    lua_shared_dict cats 1m;
--- config
    location = /test {
        content_by_lua '
            local cats = ngx.shared.cats
            cats:set("foo", true)
            cats:set("bar", false)
            local val = cats:get("foo")
            ngx.say(val, " ", type(val))
            val = cats:get("bar")
            ngx.say(val, " ", type(val))
        ';
    }
--- request
GET /test
--- response_body
true boolean
false boolean
--- no_error_log
[error]



=== TEST 4: number keys, string values
--- http_config
    lua_shared_dict cats 1m;
--- config
    location = /test {
        content_by_lua '
            local cats = ngx.shared.cats
            ngx.say(cats:set(1234, "cat"))
            ngx.say(cats:set("1234", "dog"))
            ngx.say(cats:set(256, "bird"))
            ngx.say(cats:get(1234))
            ngx.say(cats:get("1234"))
            local val = cats:get("256")
            ngx.say(val, " ", type(val))
        ';
    }
--- request
GET /test
--- response_body
truenilfalse
truenilfalse
truenilfalse
dog
dog
bird string
--- no_error_log
[error]



=== TEST 5: different-size values set to the same key
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", "hello")
            ngx.say(dogs:get("foo"))
            dogs:set("foo", "hello, world")
            ngx.say(dogs:get("foo"))
            dogs:set("foo", "hello")
            ngx.say(dogs:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
hello
hello, world
hello
--- no_error_log
[error]



=== TEST 6: expired entries (can be auto-removed by get)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32, 0.01)
            ngx.location.capture("/sleep/0.01")
            ngx.say(dogs:get("foo"))
        ';
    }
    location ~ '^/sleep/(.+)' {
        echo_sleep $1;
    }
--- request
GET /test
--- response_body
nil
--- no_error_log
[error]



=== TEST 7: expired entries (can NOT be auto-removed by get)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("bar", 56, 0.001)
            dogs:set("baz", 78, 0.001)
            dogs:set("foo", 32, 0.01)
            ngx.location.capture("/sleep/0.012")
            ngx.say(dogs:get("foo"))
        ';
    }
    location ~ '^/sleep/(.+)' {
        echo_sleep $1;
    }
--- request
GET /test
--- response_body
nil
--- no_error_log
[error]



=== TEST 8: not yet expired entries
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32, 0.5)
            ngx.location.capture("/sleep/0.01")
            ngx.say(dogs:get("foo"))
        ';
    }
    location ~ '^/sleep/(.+)' {
        echo_sleep $1;
    }
--- request
GET /test
--- response_body
32
--- no_error_log
[error]



=== TEST 9: forcibly override other valid entries
--- http_config
    lua_shared_dict dogs 100k;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local i = 0
            while i < 1000 do
                i = i + 1
                local val = string.rep(" hello", 10) .. i
                local res, err, forcible = dogs:set("key_" .. i, val)
                if not res or forcible then
                    ngx.say(res, " ", err, " ", forcible)
                    break
                end
            end
            ngx.say("abort at ", i)
            ngx.say("cur value: ", dogs:get("key_" .. i))
            if i > 1 then
                ngx.say("1st value: ", dogs:get("key_1"))
            end
            if i > 2 then
                ngx.say("2nd value: ", dogs:get("key_2"))
            end
        ';
    }
--- pipelined_requests eval
["GET /test", "GET /test"]
--- response_body eval
my $a = "true nil true\nabort at (353|705)\ncur value: " . (" hello" x 10) . "\\1\n1st value: nil\n2nd value: " . (" hello" x 10) . "2\n";
[qr/$a/,
"true nil true\nabort at 1\ncur value: " . (" hello" x 10) . "1\n"
]
--- no_error_log
[error]



=== TEST 10: forcibly override other valid entries and test LRU
--- http_config
    lua_shared_dict dogs 100k;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local i = 0
            while i < 1000 do
                i = i + 1
                local val = string.rep(" hello", 10) .. i
                if i == 10 then
                    dogs:get("key_1")
                end
                local res, err, forcible = dogs:set("key_" .. i, val)
                if not res or forcible then
                    ngx.say(res, " ", err, " ", forcible)
                    break
                end
            end
            ngx.say("abort at ", i)
            ngx.say("cur value: ", dogs:get("key_" .. i))
            if i > 1 then
                ngx.say("1st value: ", dogs:get("key_1"))
            end
            if i > 2 then
                ngx.say("2nd value: ", dogs:get("key_2"))
            end
        ';
    }
--- pipelined_requests eval
["GET /test", "GET /test"]
--- response_body eval
my $a = "true nil true\nabort at (353|705)\ncur value: " . (" hello" x 10) . "\\1\n1st value: " . (" hello" x 10) . "1\n2nd value: nil\n";
[qr/$a/,
"true nil true\nabort at 2\ncur value: " . (" hello" x 10) . "2\n1st value: " . (" hello" x 10) . "1\n"
]
--- no_error_log
[error]



=== TEST 11: dogs and cats dicts
--- http_config
    lua_shared_dict dogs 1m;
    lua_shared_dict cats 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local cats = ngx.shared.cats
            dogs:set("foo", 32)
            cats:set("foo", "hello, world")
            ngx.say(dogs:get("foo"))
            ngx.say(cats:get("foo"))
            dogs:set("foo", 56)
            ngx.say(dogs:get("foo"))
            ngx.say(cats:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
32
hello, world
56
hello, world
--- no_error_log
[error]



=== TEST 12: get non-existent keys
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            ngx.say(dogs:get("foo"))
            ngx.say(dogs:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
nil
nil
--- no_error_log
[error]



=== TEST 13: not feed the object into the call
--- SKIP
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local rc, err = pcall(dogs.set, "foo", 3, 0.01)
            ngx.say(rc, " ", err)
            rc, err = pcall(dogs.set, "foo", 3)
            ngx.say(rc, " ", err)
            rc, err = pcall(dogs.get, "foo")
            ngx.say(rc, " ", err)
        ';
    }
--- request
GET /test
--- response_body
false bad argument #1 to '?' (userdata expected, got string)
false expecting 3, 4 or 5 arguments, but only seen 2
false expecting exactly two arguments, but only seen 1
--- no_error_log
[error]



=== TEST 14: too big value
--- http_config
    lua_shared_dict dogs 50k;
--- config
    location = /test {
        content_by_lua '
            collectgarbage("collect")
            local dogs = ngx.shared.dogs
            local res, err, forcible = dogs:set("foo", string.rep("helloworld", 10000))
            ngx.say(res, " ", err, " ", forcible)
        ';
    }
--- request
GET /test
--- response_body
false no memory false
--- log_level: info
--- no_error_log
[error]
[crit]
ngx_slab_alloc() failed: no memory in lua_shared_dict zone



=== TEST 15: set too large key
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local key = string.rep("a", 65535)
            local rc, err = dogs:set(key, "hello")
            ngx.say(rc, " ", err)
            ngx.say(dogs:get(key))

            key = string.rep("a", 65536)
            ok, err = dogs:set(key, "world")
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")

        ';
    }
--- request
GET /test
--- response_body
true nil
hello
not ok: key too long
--- no_error_log
[error]



=== TEST 16: bad value type
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:set("foo", dogs)
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")
        ';
    }
--- request
GET /test
--- response_body
not ok: bad value type
--- no_error_log
[error]



=== TEST 17: delete after setting values
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32)
            ngx.say(dogs:get("foo"))
            dogs:delete("foo")
            ngx.say(dogs:get("foo"))
            dogs:set("foo", "hello, world")
            ngx.say(dogs:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
32
nil
hello, world
--- no_error_log
[error]



=== TEST 18: delete at first
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:delete("foo")
            ngx.say(dogs:get("foo"))
            dogs:set("foo", "hello, world")
            ngx.say(dogs:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
nil
hello, world
--- no_error_log
[error]



=== TEST 19: set nil after setting values
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32)
            ngx.say(dogs:get("foo"))
            dogs:set("foo", nil)
            ngx.say(dogs:get("foo"))
            dogs:set("foo", "hello, world")
            ngx.say(dogs:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
32
nil
hello, world
--- no_error_log
[error]



=== TEST 20: set nil at first
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", nil)
            ngx.say(dogs:get("foo"))
            dogs:set("foo", "hello, world")
            ngx.say(dogs:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
nil
hello, world
--- no_error_log
[error]



=== TEST 21: fail to allocate memory
--- http_config
    lua_shared_dict dogs 100k;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local i = 0
            while i < 1000 do
                i = i + 1
                local val = string.rep("hello", i )
                local res, err, forcible = dogs:set("key_" .. i, val)
                if not res or forcible then
                    ngx.say(res, " ", err, " ", forcible)
                    break
                end
            end
            ngx.say("abort at ", i)
        ';
    }
--- request
GET /test
--- response_body_like
^true nil true\nabort at (?:139|140)$
--- no_error_log
[error]



=== TEST 22: string key, int value (write_by_lua)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        rewrite_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32)
            dogs:set("bah", 10502)
            local val = dogs:get("foo")
            ngx.say(val, " ", type(val))
            val = dogs:get("bah")
            ngx.say(val, " ", type(val))
        ';
        content_by_lua return;
    }
--- request
GET /test
--- response_body
32 number
10502 number
--- no_error_log
[error]



=== TEST 23: string key, int value (access_by_lua)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        access_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32)
            dogs:set("bah", 10502)
            local val = dogs:get("foo")
            ngx.say(val, " ", type(val))
            val = dogs:get("bah")
            ngx.say(val, " ", type(val))
        ';
        content_by_lua return;
    }
--- request
GET /test
--- response_body
32 number
10502 number
--- no_error_log
[error]



=== TEST 24: string key, int value (set_by_lua)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        set_by_lua $res '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32)
            return dogs:get("foo")
        ';
        echo $res;
    }
--- request
GET /test
--- response_body
32
--- no_error_log
[error]



=== TEST 25: string key, int value (header_by_lua)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        echo hello;
        header_filter_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32)
            ngx.header["X-Foo"] = dogs:get("foo")
        ';
    }
--- request
GET /test
--- response_headers
X-Foo: 32
--- response_body
hello
--- no_error_log
[error]



=== TEST 26: too big value (forcible)
--- http_config
    lua_shared_dict dogs 50k;
--- config
    location = /test {
        content_by_lua '
            collectgarbage("collect")
            local dogs = ngx.shared.dogs
            dogs:set("bah", "hello")
            local res, err, forcible = dogs:set("foo", string.rep("helloworld", 10000))
            ngx.say(res, " ", err, " ", forcible)
        ';
    }
--- request
GET /test
--- response_body
false no memory true
--- log_level: info
--- no_error_log
[error]
[crit]
ngx_slab_alloc() failed: no memory in lua_shared_dict zone



=== TEST 27: add key (key exists)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32)
            local res, err, forcible = dogs:add("foo", 10502)
            ngx.say("add: ", res, " ", err, " ", forcible)
            ngx.say("foo = ", dogs:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
add: false exists false
foo = 32
--- no_error_log
[error]



=== TEST 28: add key (key not exists)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("bah", 32)
            local res, err, forcible = dogs:add("foo", 10502)
            ngx.say("add: ", res, " ", err, " ", forcible)
            ngx.say("foo = ", dogs:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
add: true nil false
foo = 10502
--- no_error_log
[error]



=== TEST 29: add key (key expired)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("bar", 32, 0.001)
            dogs:set("baz", 32, 0.001)
            dogs:set("foo", 32, 0.001)
            ngx.location.capture("/sleep/0.002")
            local res, err, forcible = dogs:add("foo", 10502)
            ngx.say("add: ", res, " ", err, " ", forcible)
            ngx.say("foo = ", dogs:get("foo"))
        ';
    }
    location ~ ^/sleep/(.+) {
        echo_sleep $1;
    }
--- request
GET /test
--- response_body
add: true nil false
foo = 10502
--- no_error_log
[error]



=== TEST 30: add key (key expired and value size unmatched)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("bar", 32, 0.001)
            dogs:set("baz", 32, 0.001)
            dogs:set("foo", "hi", 0.001)
            ngx.location.capture("/sleep/0.002")
            local res, err, forcible = dogs:add("foo", "hello")
            ngx.say("add: ", res, " ", err, " ", forcible)
            ngx.say("foo = ", dogs:get("foo"))
        ';
    }
    location ~ ^/sleep/(.+) {
        echo_sleep $1;
    }
--- request
GET /test
--- response_body
add: true nil false
foo = hello
--- no_error_log
[error]



=== TEST 31: incr key (key exists)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32)
            local res, err, forcible = dogs:replace("foo", 10502)
            ngx.say("replace: ", res, " ", err, " ", forcible)
            ngx.say("foo = ", dogs:get("foo"))

            local res, err, forcible = dogs:replace("foo", "hello")
            ngx.say("replace: ", res, " ", err, " ", forcible)
            ngx.say("foo = ", dogs:get("foo"))

        ';
    }
--- request
GET /test
--- response_body
replace: true nil false
foo = 10502
replace: true nil false
foo = hello
--- no_error_log
[error]



=== TEST 32: replace key (key not exists)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("bah", 32)
            local res, err, forcible = dogs:replace("foo", 10502)
            ngx.say("replace: ", res, " ", err, " ", forcible)
            ngx.say("foo = ", dogs:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
replace: false not found false
foo = nil
--- no_error_log
[error]



=== TEST 33: replace key (key expired)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("bar", 3, 0.001)
            dogs:set("baz", 2, 0.001)
            dogs:set("foo", 32, 0.001)
            ngx.location.capture("/sleep/0.002")
            local res, err, forcible = dogs:replace("foo", 10502)
            ngx.say("replace: ", res, " ", err, " ", forcible)
            ngx.say("foo = ", dogs:get("foo"))
        ';
    }
    location ~ ^/sleep/(.+) {
        echo_sleep $1;
    }
--- request
GET /test
--- response_body
replace: false not found false
foo = nil
--- no_error_log
[error]



=== TEST 34: replace key (key expired and value size unmatched)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("bar", 32, 0.001)
            dogs:set("baz", 32, 0.001)
            dogs:set("foo", "hi", 0.001)
            ngx.location.capture("/sleep/0.002")
            local rc, err, forcible = dogs:replace("foo", "hello")
            ngx.say("replace: ", rc, " ", err, " ", forcible)
            ngx.say("foo = ", dogs:get("foo"))
        ';
    }
    location ~ ^/sleep/(.+) {
        echo_sleep $1;
    }
--- request
GET /test
--- response_body
replace: false not found false
foo = nil
--- no_error_log
[error]



=== TEST 35: incr key (key exists)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32)
            local res, err = dogs:incr("foo", 10502)
            ngx.say("incr: ", res, " ", err)
            ngx.say("foo = ", dogs:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
incr: 10534 nil
foo = 10534
--- no_error_log
[error]



=== TEST 36: replace key (key not exists)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("bah", 32)
            local res, err = dogs:incr("foo", 2)
            ngx.say("incr: ", res, " ", err)
            ngx.say("foo = ", dogs:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
incr: nil not found
foo = nil
--- no_error_log
[error]



=== TEST 37: replace key (key expired)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("bar", 3, 0.001)
            dogs:set("baz", 2, 0.001)
            dogs:set("foo", 32, 0.001)
            ngx.location.capture("/sleep/0.002")
            local res, err = dogs:incr("foo", 10502)
            ngx.say("incr: ", res, " ", err)
            ngx.say("foo = ", dogs:get("foo"))
        ';
    }
    location ~ ^/sleep/(.+) {
        echo_sleep $1;
    }
--- request
GET /test
--- response_body
incr: nil not found
foo = nil
--- no_error_log
[error]



=== TEST 38: incr key (incr by 0)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32)
            local res, err = dogs:incr("foo", 0)
            ngx.say("incr: ", res, " ", err)
            ngx.say("foo = ", dogs:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
incr: 32 nil
foo = 32
--- no_error_log
[error]



=== TEST 39: incr key (incr by floating point number)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32)
            local res, err = dogs:incr("foo", 0.14)
            ngx.say("incr: ", res, " ", err)
            ngx.say("foo = ", dogs:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
incr: 32.14 nil
foo = 32.14
--- no_error_log
[error]



=== TEST 40: incr key (incr by negative numbers)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32)
            local res, err = dogs:incr("foo", -0.14)
            ngx.say("incr: ", res, " ", err)
            ngx.say("foo = ", dogs:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
incr: 31.86 nil
foo = 31.86
--- no_error_log
[error]



=== TEST 41: incr key (original value is not number)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", true)
            local res, err = dogs:incr("foo", -0.14)
            ngx.say("incr: ", res, " ", err)
            ngx.say("foo = ", dogs:get("foo"))
        ';
    }
--- request
GET /test
--- response_body
incr: nil not a number
foo = true
--- no_error_log
[error]



=== TEST 42: get and set with flags
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32, 0, 199)
            dogs:set("bah", 10502, 202)
            local val, flags = dogs:get("foo")
            ngx.say(val, " ", type(val))
            ngx.say(flags, " ", type(flags))
            val, flags = dogs:get("bah")
            ngx.say(val, " ", type(val))
            ngx.say(flags, " ", type(flags))
        ';
    }
--- request
GET /test
--- response_body
32 number
199 number
10502 number
nil nil
--- no_error_log
[error]



=== TEST 43: expired entries (can be auto-removed by get), with flags set
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32, 0.01, 255)
            ngx.location.capture("/sleep/0.01")
            local res, flags = dogs:get("foo")
            ngx.say("res = ", res, ", flags = ", flags)
        ';
    }
    location ~ '^/sleep/(.+)' {
        echo_sleep $1;
    }
--- request
GET /test
--- response_body
res = nil, flags = nil
--- no_error_log
[error]



=== TEST 44: flush_all
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /t {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32)
            dogs:set("bah", 10502)

            local val = dogs:get("foo")
            ngx.say(val, " ", type(val))
            val = dogs:get("bah")
            ngx.say(val, " ", type(val))

            dogs:flush_all()

            val = dogs:get("foo")
            ngx.say(val, " ", type(val))
            val = dogs:get("bah")
            ngx.say(val, " ", type(val))
        ';
    }
--- request
GET /t
--- response_body
32 number
10502 number
nil nil
nil nil
--- no_error_log
[error]



=== TEST 45: flush_expires
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /t {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", "x", 1)
            dogs:set("bah", "y", 0)
            dogs:set("bar", "z", 100)

            ngx.sleep(1.5)

            local num = dogs:flush_expired()
            ngx.say(num)
        ';
    }
--- request
GET /t
--- response_body
1
--- no_error_log
[error]



=== TEST 46: flush_expires with number
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /t {
        content_by_lua '
            local dogs = ngx.shared.dogs

            for i=1,100 do
                dogs:set(tostring(i), "x", 1)
            end

            dogs:set("bah", "y", 0)
            dogs:set("bar", "z", 100)

            ngx.sleep(1.5)

            local num = dogs:flush_expired(42)
            ngx.say(num)
        ';
    }
--- request
GET /t
--- response_body
42
--- no_error_log
[error]



=== TEST 47: flush_expires an empty dict
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /t {
        content_by_lua '
            local dogs = ngx.shared.dogs

            local num = dogs:flush_expired()
            ngx.say(num)
        ';
    }
--- request
GET /t
--- response_body
0
--- no_error_log
[error]



=== TEST 48: flush_expires a dict without expired items
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /t {
        content_by_lua '
            local dogs = ngx.shared.dogs

            dogs:set("bah", "y", 0)
            dogs:set("bar", "z", 100)

            local num = dogs:flush_expired()
            ngx.say(num)
        ';
    }
--- request
GET /t
--- response_body
0
--- no_error_log
[error]



=== TEST 49: list all keys in a shdict
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /t {
        content_by_lua '
            local dogs = ngx.shared.dogs

            dogs:set("bah", "y", 0)
            dogs:set("bar", "z", 0)
            local keys = dogs:get_keys()
            ngx.say(#keys)
            table.sort(keys)
            for _,k in ipairs(keys) do
                ngx.say(k)
            end
        ';
    }
--- request
GET /t
--- response_body
2
bah
bar
--- no_error_log
[error]



=== TEST 50: list keys in a shdict with limit
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /t {
        content_by_lua '
            local dogs = ngx.shared.dogs

            dogs:set("bah", "y", 0)
            dogs:set("bar", "z", 0)
            local keys = dogs:get_keys(1)
            ngx.say(#keys)
        ';
    }
--- request
GET /t
--- response_body
1
--- no_error_log
[error]



=== TEST 51: list all keys in a shdict with expires
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /t {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", "x", 1)
            dogs:set("bah", "y", 0)
            dogs:set("bar", "z", 100)

            ngx.sleep(1.5)

            local keys = dogs:get_keys()
            ngx.say(#keys)
        ';
    }
--- request
GET /t
--- response_body
2
--- no_error_log
[error]



=== TEST 52: list keys in a shdict with limit larger than number of keys
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /t {
        content_by_lua '
            local dogs = ngx.shared.dogs

            dogs:set("bah", "y", 0)
            dogs:set("bar", "z", 0)
            local keys = dogs:get_keys(3)
            ngx.say(#keys)
        ';
    }
--- request
GET /t
--- response_body
2
--- no_error_log
[error]



=== TEST 53: list keys in an empty shdict
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /t {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local keys = dogs:get_keys()
            ngx.say(#keys)
        ';
    }
--- request
GET /t
--- response_body
0
--- no_error_log
[error]



=== TEST 54: list keys in an empty shdict with a limit
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /t {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local keys = dogs:get_keys(4)
            ngx.say(#keys)
        ';
    }
--- request
GET /t
--- response_body
0
--- no_error_log
[error]



=== TEST 55: list all keys in a shdict with all keys expired
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /t {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", "x", 1)
            dogs:set("bah", "y", 1)
            dogs:set("bar", "z", 1)

            ngx.sleep(1.5)

            local keys = dogs:get_keys()
            ngx.say(#keys)
        ';
    }
--- request
GET /t
--- response_body
0
--- no_error_log
[error]



=== TEST 56: list all keys in a shdict with more than 1024 keys with no limit set
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /t {
        content_by_lua '
            local dogs = ngx.shared.dogs
            for i=1,2048 do
                dogs:set(tostring(i), i)
            end
            local keys = dogs:get_keys()
            ngx.say(#keys)
        ';
    }
--- request
GET /t
--- response_body
1024
--- no_error_log
[error]



=== TEST 57: list all keys in a shdict with more than 1024 keys with 0 limit set
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /t {
        content_by_lua '
            local dogs = ngx.shared.dogs
            for i=1,2048 do
                dogs:set(tostring(i), i)
            end
            local keys = dogs:get_keys(0)
            ngx.say(#keys)
        ';
    }
--- request
GET /t
--- response_body
2048
--- no_error_log
[error]



=== TEST 58: safe_set
--- http_config
    lua_shared_dict dogs 100k;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local i = 0
            while i < 1000 do
                i = i + 1
                local val = string.rep(" hello", 10) .. i
                local res, err = dogs:safe_set("key_" .. i, val)
                if not res then
                    ngx.say(res, " ", err)
                    break
                end
            end
            ngx.say("abort at ", i)
            ngx.say("cur value: ", dogs:get("key_" .. i))
            if i > 1 then
                ngx.say("1st value: ", dogs:get("key_1"))
            end
            if i > 2 then
                ngx.say("2nd value: ", dogs:get("key_2"))
            end
        ';
    }
--- pipelined_requests eval
["GET /test", "GET /test"]
--- response_body eval
my $a = "false no memory\nabort at (353|705)\ncur value: nil\n1st value: " . (" hello" x 10) . "1\n2nd value: " . (" hello" x 10) . "2\n";
[qr/$a/, qr/$a/]
--- no_error_log
[error]



=== TEST 59: safe_add
--- http_config
    lua_shared_dict dogs 100k;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local i = 0
            while i < 1000 do
                i = i + 1
                local val = string.rep(" hello", 10) .. i
                local res, err = dogs:safe_add("key_" .. i, val)
                if not res then
                    ngx.say(res, " ", err)
                    break
                end
            end
            ngx.say("abort at ", i)
            ngx.say("cur value: ", dogs:get("key_" .. i))
            if i > 1 then
                ngx.say("1st value: ", dogs:get("key_1"))
            end
            if i > 2 then
                ngx.say("2nd value: ", dogs:get("key_2"))
            end
        ';
    }
--- pipelined_requests eval
["GET /test", "GET /test"]
--- response_body eval
my $a = "false no memory\nabort at (353|705)\ncur value: nil\n1st value: " . (" hello" x 10) . "1\n2nd value: " . (" hello" x 10) . "2\n";
[qr/$a/,
q{false exists
abort at 1
cur value:  hello hello hello hello hello hello hello hello hello hello1
}
]
--- no_error_log
[error]



=== TEST 60: get_stale: expired entries can still be fetched
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 32, 0.01)
            dogs:set("blah", 33, 0.1)
            ngx.sleep(0.02)
            local val, flags, stale = dogs:get_stale("foo")
            ngx.say(val, ", ", flags, ", ", stale)
            local val, flags, stale = dogs:get_stale("blah")
            ngx.say(val, ", ", flags, ", ", stale)
        ';
    }
--- request
GET /test
--- response_body
32, nil, true
33, nil, false
--- no_error_log
[error]



=== TEST 61: set nil key
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:set(nil, 32)
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")
        ';
    }
--- request
GET /test
--- response_body
not ok: nil key
--- no_error_log
[error]



=== TEST 62: set bad zone argument
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs.set(nil, "foo", 32)
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")
        ';
    }
--- request
GET /test
--- response_body_like: 500 Internal Server Error
--- error_code: 500
--- error_log
bad "zone" argument



=== TEST 63: set empty string keys
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:set("", 32)
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")
        ';
    }
--- request
GET /test
--- response_body
not ok: empty key
--- no_error_log
[error]



=== TEST 64: get bad zone argument
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs.get(nil, "foo")
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")
        ';
    }
--- request
GET /test
--- response_body_like: 500 Internal Server Error
--- error_code: 500
--- error_log
bad "zone" argument



=== TEST 65: get nil key
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:get(nil)
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")
        ';
    }
--- request
GET /test
--- response_body
not ok: nil key
--- no_error_log
[error]



=== TEST 66: get empty key
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:get("")
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")
        ';
    }
--- request
GET /test
--- response_body
not ok: empty key
--- no_error_log
[error]



=== TEST 67: get a too-long key
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:get(string.rep("a", 65536))
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")
        ';
    }
--- request
GET /test
--- response_body
not ok: key too long
--- no_error_log
[error]



=== TEST 68: set & get large values
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:set("foo", string.rep("helloworld", 1024))
            if not ok then
                ngx.say("set not ok: ", err)
                return
            end
            ngx.say("set ok")

            local data, err = dogs:get("foo")
            if data == nil and err then
                ngx.say("get not ok: ", err)
                return
            end
            ngx.say("get ok: ", #data)

        ';
    }
--- request
GET /test
--- response_body
set ok
get ok: 10240
--- no_error_log
[error]



=== TEST 69: get_stale nil key
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:get_stale(nil)
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")
        ';
    }
--- request
GET /test
--- response_body
not ok: nil key
--- no_error_log
[error]



=== TEST 70: get_stale empty key
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:get_stale("")
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")
        ';
    }
--- request
GET /test
--- response_body
not ok: empty key
--- no_error_log
[error]



=== TEST 71: get_stale number key
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:set(1024, "hello")
            if not ok then
                ngx.say("set not ok: ", err)
                return
            end
            ngx.say("set ok")
            local data, err = dogs:get_stale(1024)
            if not ok then
                ngx.say("get_stale not ok: ", err)
                return
            end
            ngx.say("get_stale: ", data)
        ';
    }
--- request
GET /test
--- response_body
set ok
get_stale: hello
--- no_error_log
[error]



=== TEST 72: get_stale a too-long key
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:get_stale(string.rep("a", 65536))
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")
        ';
    }
--- request
GET /test
--- response_body
not ok: key too long
--- no_error_log
[error]



=== TEST 73: get_stale a non-existent key
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local data, err = dogs:get_stale("not_found")
            if data == nil and err then
                ngx.say("get not ok: ", err)
                return
            end
            ngx.say("get ok: ", data)
        ';
    }
--- request
GET /test
--- response_body
get ok: nil
--- no_error_log
[error]



=== TEST 74: set & get_stale large values
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:set("foo", string.rep("helloworld", 1024))
            if not ok then
                ngx.say("set not ok: ", err)
                return
            end
            ngx.say("set ok")

            local data, err, stale = dogs:get_stale("foo")
            if data == nil and err then
                ngx.say("get not ok: ", err)
                return
            end
            ngx.say("get_stale ok: ", #data, ", stale: ", stale)

        ';
    }
--- request
GET /test
--- response_body
set ok
get_stale ok: 10240, stale: false
--- no_error_log
[error]



=== TEST 75: set & get_stale boolean values (true)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:set("foo", true)
            if not ok then
                ngx.say("set not ok: ", err)
                return
            end
            ngx.say("set ok")

            local data, err, stale = dogs:get_stale("foo")
            if data == nil and err then
                ngx.say("get not ok: ", err)
                return
            end
            ngx.say("get_stale ok: ", data, ", stale: ", stale)

        ';
    }
--- request
GET /test
--- response_body
set ok
get_stale ok: true, stale: false
--- no_error_log
[error]



=== TEST 76: set & get_stale boolean values (false)
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:set("foo", false)
            if not ok then
                ngx.say("set not ok: ", err)
                return
            end
            ngx.say("set ok")

            local data, err, stale = dogs:get_stale("foo")
            if data == nil and err then
                ngx.say("get not ok: ", err)
                return
            end
            ngx.say("get_stale ok: ", data, ", stale: ", stale)

        ';
    }
--- request
GET /test
--- response_body
set ok
get_stale ok: false, stale: false
--- no_error_log
[error]



=== TEST 77: set & get_stale with a flag
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:set("foo", false, 0, 325)
            if not ok then
                ngx.say("set not ok: ", err)
                return
            end
            ngx.say("set ok")

            local data, err, stale = dogs:get_stale("foo")
            if data == nil and err then
                ngx.say("get not ok: ", err)
                return
            end
            flags = err
            ngx.say("get_stale ok: ", data, ", flags: ", flags,
                    ", stale: ", stale)

        ';
    }
--- request
GET /test
--- response_body
set ok
get_stale ok: false, flags: 325, stale: false
--- no_error_log
[error]



=== TEST 78: incr nil key
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:incr(nil, 32)
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")
        ';
    }
--- request
GET /test
--- response_body
not ok: nil key
--- no_error_log
[error]



=== TEST 79: incr bad zone argument
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs.incr(nil, "foo", 32)
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")
        ';
    }
--- request
GET /test
--- response_body_like: 500 Internal Server Error
--- error_code: 500
--- error_log
bad "zone" argument



=== TEST 80: incr empty string keys
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:incr("", 32)
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")
        ';
    }
--- request
GET /test
--- response_body
not ok: empty key
--- no_error_log
[error]



=== TEST 81: incr too long key
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local key = string.rep("a", 65536)
            local ok, err = dogs:incr(key, 32)
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")

        ';
    }
--- request
GET /test
--- response_body
not ok: key too long
--- no_error_log
[error]



=== TEST 82: incr number key
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local key = 56
            local ok, err = dogs:set(key, 1)
            if not ok then
                ngx.say("set not ok: ", err)
                return
            end
            ngx.say("set ok")
            ok, err = dogs:incr(key, 32)
            if not ok then
                ngx.say("incr not ok: ", err)
                return
            end
            ngx.say("incr ok")
            local data, err = dogs:get(key)
            if data == nil and err then
                ngx.say("get not ok: ", err)
                return
            end
            local flags = err
            ngx.say("get ok: ", data, ", flags: ", flags)

        ';
    }
--- request
GET /test
--- response_body
set ok
incr ok
get ok: 33, flags: nil
--- no_error_log
[error]



=== TEST 83: incr a number-like string key
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local key = 56
            local ok, err = dogs:set(key, 1)
            if not ok then
                ngx.say("set not ok: ", err)
                return
            end
            ngx.say("set ok")
            ok, err = dogs:incr(key, "32")
            if not ok then
                ngx.say("incr not ok: ", err)
                return
            end
            ngx.say("incr ok")
            local data, err = dogs:get(key)
            if data == nil and err then
                ngx.say("get not ok: ", err)
                return
            end
            local flags = err
            ngx.say("get ok: ", data, ", flags: ", flags)

        ';
    }
--- request
GET /test
--- response_body
set ok
incr ok
get ok: 33, flags: nil
--- no_error_log
[error]



=== TEST 84: add nil values
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            local ok, err = dogs:add("foo", nil)
            if not ok then
                ngx.say("not ok: ", err)
                return
            end
            ngx.say("ok")
        ';
    }
--- request
GET /test
--- response_body
not ok: attempt to add or replace nil values
--- no_error_log
[error]



=== TEST 85: replace key with exptime
--- http_config
    lua_shared_dict dogs 1m;
--- config
    location = /test {
        content_by_lua '
            local dogs = ngx.shared.dogs
            dogs:set("foo", 2, 0)
            dogs:replace("foo", 32, 0.01)
            local data = dogs:get("foo")
            ngx.say("get foo: ", data)
            ngx.location.capture("/sleep/0.02")
            local res, err, forcible = dogs:replace("foo", 10502)
            ngx.say("replace: ", res, " ", err, " ", forcible)
            ngx.say("foo = ", dogs:get("foo"))
        ';
    }
    location ~ ^/sleep/(.+) {
        echo_sleep $1;
    }
--- request
GET /test
--- response_body
get foo: 32
replace: false not found false
foo = nil
--- no_error_log
[error]

