tests/integration/aof.tcl

   1 set defaults { appendonly {yes} appendfilename {appendonly.aof} }
   2 set server_path [tmpdir server.aof]
   3 set aof_path "$server_path/appendonly.aof"
   4
   5 proc append_to_aof {str} {
   6     upvar fp fp
   7     puts -nonewline $fp $str
   8 }
   9
  10 proc create_aof {code} {
  11     upvar fp fp aof_path aof_path
  12     set fp [open $aof_path w+]
  13     uplevel 1 $code
  14     close $fp
  15 }
  16
  17 proc start_server_aof {overrides code} {
  18     upvar defaults defaults srv srv server_path server_path
  19     set config [concat $defaults $overrides]
  20     set srv [start_server [list overrides $config]]
  21     uplevel 1 $code
  22     kill_server $srv
  23 }
  24
  25 tags {"aof"} {
  26     ## Test the server doesn't start when the AOF contains an unfinished MULTI
  27     create_aof {
  28         append_to_aof [formatCommand set foo hello]
  29         append_to_aof [formatCommand multi]
  30         append_to_aof [formatCommand set bar world]
  31     }
  32
  33     start_server_aof [list dir $server_path] {
  34         test "Unfinished MULTI: Server should not have been started" {
  35             assert_equal 0 [is_alive $srv]
  36         }
  37
  38         test "Unfinished MULTI: Server should have logged an error" {
  39             set result [exec cat [dict get $srv stdout] | tail -n1]
  40             assert_match "*Unexpected end of file reading the append only file*" $result
  41         }
  42     }
  43
  44     ## Test that the server exits when the AOF contains a short read
  45     create_aof {
  46         append_to_aof [formatCommand set foo hello]
  47         append_to_aof [string range [formatCommand set bar world] 0 end-1]
  48     }
  49
  50     start_server_aof [list dir $server_path] {
  51         test "Short read: Server should not have been started" {
  52             assert_equal 0 [is_alive $srv]
  53         }
  54
  55         test "Short read: Server should have logged an error" {
  56             set result [exec cat [dict get $srv stdout] | tail -n1]
  57             assert_match "*Bad file format reading the append only file*" $result
  58         }
  59     }
  60
  61     ## Test that redis-check-aof indeed sees this AOF is not valid
  62     test "Short read: Utility should confirm the AOF is not valid" {
  63         catch {
  64             exec src/redis-check-aof $aof_path
  65         } result
  66         assert_match "*not valid*" $result
  67     }
  68
  69     test "Short read: Utility should be able to fix the AOF" {
  70         set result [exec echo y | src/redis-check-aof --fix $aof_path]
  71         assert_match "*Successfully truncated AOF*" $result
  72     }
  73
  74     ## Test that the server can be started using the truncated AOF
  75     start_server_aof [list dir $server_path] {
  76         test "Fixed AOF: Server should have been started" {
  77             assert_equal 1 [is_alive $srv]
  78         }
  79
  80         test "Fixed AOF: Keyspace should contain values that were parsable" {
  81             set client [redis [dict get $srv host] [dict get $srv port]]
  82             assert_equal "hello" [$client get foo]
  83             assert_equal "" [$client get bar]
  84         }
  85     }
  86
  87     ## Test that SPOP (that modifies the client its argc/argv) is correctly free'd
  88     create_aof {
  89         append_to_aof [formatCommand sadd set foo]
  90         append_to_aof [formatCommand sadd set bar]
  91         append_to_aof [formatCommand spop set]
  92     }
  93
  94     start_server_aof [list dir $server_path] {
  95         test "AOF+SPOP: Server should have been started" {
  96             assert_equal 1 [is_alive $srv]
  97         }
  98
  99         test "AOF+SPOP: Set should have 1 member" {
 100             set client [redis [dict get $srv host] [dict get $srv port]]
 101             assert_equal 1 [$client scard set]
 102         }
 103     }
 104
 105     ## Test that EXPIREAT is loaded correctly
 106     create_aof {
 107         append_to_aof [formatCommand rpush list foo]
 108         append_to_aof [formatCommand expireat list 1000]
 109         append_to_aof [formatCommand rpush list bar]
 110     }
 111
 112     start_server_aof [list dir $server_path] {
 113         test "AOF+EXPIRE: Server should have been started" {
 114             assert_equal 1 [is_alive $srv]
 115         }
 116
 117         test "AOF+EXPIRE: List should be empty" {
 118             set client [redis [dict get $srv host] [dict get $srv port]]
 119             assert_equal 0 [$client llen list]
 120         }
 121     }
 122 }