#!/usr/bin/perl

use strict;

use Test::More tests => 108;

use FindBin qw($Bin);

use lib "$Bin/lib";

use MemcachedTest;

my $server = new_memcached("-o no_lru_crawler,no_lru_maintainer");

my $sock = $server->sock;

## Output looks like this:

##

## STAT pid 22969

## STAT uptime 13

## STAT time 1259170891

## STAT version 1.4.3

## STAT libevent 1.4.13-stable.

## see doc/protocol.txt for others

# note that auth stats are tested in auth specific tests

my $stats = mem_stats($sock);

# Test number of keys

if (MemcachedTest::enabled_tls_testing()) {

    # when TLS is enabled, stats contains time_since_server_cert_refresh

    is(scalar(keys(%$stats)), 72, "expected count of stats values");

} else {

    is(scalar(keys(%$stats)), 71, "expected count of stats values");

}

# Test initial state

foreach my $key (qw(curr_items total_items bytes cmd_get cmd_set get_hits evictions get_misses get_expired

                 bytes_written delete_hits delete_misses incr_hits incr_misses decr_hits get_flushed

                 decr_misses listen_disabled_num lrutail_reflocked time_in_listen_disabled_us)) {

    is($stats->{$key}, 0, "initial $key is zero");

}

is($stats->{accepting_conns}, 1, "initial accepting_conns is one");

# Do some operations

print $sock "set foo 0 0 6\r\nfooval\r\n";

is(scalar <$sock>, "STORED\r\n", "stored foo");

mem_get_is($sock, "foo", "fooval");

my $stats = mem_stats($sock);

foreach my $key (qw(total_items curr_items cmd_get cmd_set get_hits)) {

    is($stats->{$key}, 1, "after one set/one get $key is 1");

}

my $cache_dump = mem_stats($sock, " cachedump 1 100");

ok(defined $cache_dump->{'foo'}, "got foo from cachedump");

print $sock "delete foo\r\n";

is(scalar <$sock>, "DELETED\r\n", "deleted foo");

my $stats = mem_stats($sock);

is($stats->{delete_hits}, 1);

is($stats->{delete_misses}, 0);

print $sock "delete foo\r\n";

is(scalar <$sock>, "NOT_FOUND\r\n", "shouldn't delete foo again");

my $stats = mem_stats($sock);

is($stats->{delete_hits}, 1);

is($stats->{delete_misses}, 1);

# incr stats

sub check_incr_stats {

    my ($ih, $im, $dh, $dm) = @_;

    my $stats = mem_stats($sock);

    is($stats->{incr_hits}, $ih);

    is($stats->{incr_misses}, $im);

    is($stats->{decr_hits}, $dh);

    is($stats->{decr_misses}, $dm);

}

print $sock "incr i 1\r\n";

is(scalar <$sock>, "NOT_FOUND\r\n", "shouldn't incr a missing thing");

check_incr_stats(0, 1, 0, 0);

print $sock "decr d 1\r\n";

is(scalar <$sock>, "NOT_FOUND\r\n", "shouldn't decr a missing thing");

check_incr_stats(0, 1, 0, 1);

print $sock "set n 0 0 1\r\n0\r\n";

is(scalar <$sock>, "STORED\r\n", "stored n");

print $sock "incr n 3\r\n";

is(scalar <$sock>, "3\r\n", "incr works");

check_incr_stats(1, 1, 0, 1);

print $sock "decr n 1\r\n";

is(scalar <$sock>, "2\r\n", "decr works");

check_incr_stats(1, 1, 1, 1);

# cas stats

sub check_cas_stats {

    my ($ch, $cm, $cb) = @_;

    my $stats = mem_stats($sock);

    is($stats->{cas_hits}, $ch);

    is($stats->{cas_misses}, $cm);

    is($stats->{cas_badval}, $cb);

}

check_cas_stats(0, 0, 0);

print $sock "cas c 0 0 1 99999999\r\nz\r\n";

is(scalar <$sock>, "NOT_FOUND\r\n", "missed cas");

check_cas_stats(0, 1, 0);

print $sock "set c 0 0 1\r\nx\r\n";

is(scalar <$sock>, "STORED\r\n", "stored c");

my ($id, $v) = mem_gets($sock, 'c');

is('x', $v, 'got the expected value');

print $sock "cas c 0 0 1 99999999\r\nz\r\n";

is(scalar <$sock>, "EXISTS\r\n", "missed cas");

check_cas_stats(0, 1, 1);

my ($newid, $v) = mem_gets($sock, 'c');

is('x', $v, 'got the expected value');

print $sock "cas c 0 0 1 $id\r\nz\r\n";

is(scalar <$sock>, "STORED\r\n", "good cas");

check_cas_stats(1, 1, 1);

my ($newid, $v) = mem_gets($sock, 'c');

is('z', $v, 'got the expected value');

my $settings = mem_stats($sock, ' settings');

is(1024, $settings->{'maxconns'});

# we run SSL tests over TCP; hence the domain_socket

# is expected to be NULL.

if (enabled_tls_testing()) {

    is('NULL', $settings->{'domain_socket'});

} else {

    isnt('NULL', $settings->{'domain_socket'});

}

is('on', $settings->{'evictions'});

is('yes', $settings->{'cas_enabled'});

is('no', $settings->{'auth_enabled_sasl'});

print $sock "stats reset\r\n";

is(scalar <$sock>, "RESET\r\n", "good stats reset");

my $stats = mem_stats($sock);

is(0, $stats->{'cmd_get'});

is(0, $stats->{'cmd_set'});

is(0, $stats->{'get_hits'});

is(0, $stats->{'get_misses'});

is(0, $stats->{'get_expired'});

is(0, $stats->{'get_flushed'});

is(0, $stats->{'delete_misses'});

is(0, $stats->{'delete_hits'});

is(0, $stats->{'incr_misses'});

is(0, $stats->{'incr_hits'});

is(0, $stats->{'decr_misses'});

is(0, $stats->{'decr_hits'});

is(0, $stats->{'cas_misses'});

is(0, $stats->{'cas_hits'});

is(0, $stats->{'cas_badval'});

is(0, $stats->{'evictions'});

is(0, $stats->{'reclaimed'});

is(0, $stats->{'lrutail_reflocked'});

# item expired

print $sock "set should_expire 0 2678400 6\r\nfooval\r\n"; #2678400 = 31 days in seconds

is(scalar <$sock>, "STORED\r\n", "set item to expire");

print $sock "get should_expire\r\n";

is(scalar <$sock>, "END\r\n", "item not returned");

my $stats = mem_stats($sock);

is(1, $stats->{'get_expired'}, "get_expired counter is 1");

print $sock "set should_be_flushed 0 0 6\r\nbooval\r\n";

is(scalar <$sock>, "STORED\r\n", "set item to flush");

print $sock "flush_all\r\n";

is(scalar <$sock>, "OK\r\n", "flushed");

print $sock "get should_be_flushed\r\n";

is(scalar <$sock>, "END\r\n", "flushed item not returned");

my $stats = mem_stats($sock);

is($stats->{cmd_flush}, 1, "after one flush cmd_flush is 1");

is($stats->{get_flushed}, 1, "after flush and a get, get_flushed is 1");