tests/test-cacheextents.sh

   1 #!/usr/bin/env bash
   2 # nbdkit
   3 # Copyright (C) 2019 Red Hat Inc.
   4 #
   5 # Redistribution and use in source and binary forms, with or without
   6 # modification, are permitted provided that the following conditions are
   7 # met:
   8 #
   9 # * Redistributions of source code must retain the above copyright
  10 # notice, this list of conditions and the following disclaimer.
  11 #
  12 # * Redistributions in binary form must reproduce the above copyright
  13 # notice, this list of conditions and the following disclaimer in the
  14 # documentation and/or other materials provided with the distribution.
  15 #
  16 # * Neither the name of Red Hat nor the names of its contributors may be
  17 # used to endorse or promote products derived from this software without
  18 # specific prior written permission.
  19 #
  20 # THIS SOFTWARE IS PROVIDED BY RED HAT AND CONTRIBUTORS ''AS IS'' AND
  21 # ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  22 # THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
  23 # PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL RED HAT OR
  24 # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  25 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  26 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF
  27 # USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  28 # ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
  29 # OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT
  30 # OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31 # SUCH DAMAGE.
  32
  33 source ./functions.sh
  34 set -x
  35 set -e
  36
  37 requires grep --version
  38 requires qemu-img --version
  39 requires qemu-io --version
  40
  41 sock="$(mktemp -u)"
  42 sockurl="nbd+unix:///?socket=$sock"
  43 pidfile="test-cacheextents.pid"
  44 accessfile="test-cacheextents-access.log"
  45 accessfile_full="$PWD/test-cacheextents-access.log"
  46 files="$pidfile $sock"
  47 rm -f $files $accessfile
  48 cleanup_fn rm -f $files
  49
  50 # Intentionally using EOF rather than 'EOF' so that we can pass in the
  51 # $accessfile_full
  52 start_nbdkit \
  53     -P $pidfile \
  54     -U $sock \
  55     --filter=cacheextents \
  56     sh - <<EOF
  57 echo "Call: \$@" >>$accessfile_full
  58 size=4M
  59 block_size=\$((1024*1024))
  60 case "\$1" in
  61   thread_model) echo parallel ;;
  62   get_size) echo \$size ;;
  63   can_extents) ;;
  64   extents)
  65     echo "extents request: \$@" >>$accessfile_full
  66     offset=\$((\$4 / \$block_size))
  67     count=\$((\$3 / \$block_size))
  68     length=\$((\$offset + \$count))
  69     for i in \$(seq \$offset \$length); do
  70       echo \${i}M \$block_size \$((i%4)) >>$accessfile_full
  71       echo \${i}M \$block_size \$((i%4))
  72     done
  73     ;;
  74   pread) dd if=/dev/zero count=\$3 iflag=count_bytes ;;
  75   can_write) ;;
  76   pwrite) dd of=/dev/null ;;
  77   can_trim) ;;
  78   trim) ;;
  79   can_zero) ;;
  80   zero) ;;
  81   *) exit 2 ;;
  82 esac
  83 EOF
  84
  85
  86 test_me() {
  87     num_accesses=$1
  88     shift
  89
  90     qemu-io -f raw "$@" "$sockurl"
  91     test "$(grep -c "^extents request: " $accessfile)" -eq "$num_accesses"
  92     ret=$?
  93     rm -f "$accessfile"
  94     return $ret
  95 }
  96
  97 # First one causes caching, the rest should be returned from cache.
  98 test_me 1 -c 'map' -c 'map' -c 'map'
  99 # First one is still cached from last time, discard should kill the cache, then
 100 # one request should go through.
 101 test_me 1 -c 'map' -c 'discard 0 1' -c 'map'
 102 # Same as above, only this time the cache is killed before all the operations as
 103 # well.  This is used from now on to clear the cache as it seems nicer and
 104 # faster than running new nbdkit for each test.
 105 test_me 2 -c 'discard 0 1' -c 'map' -c 'discard 0 1' -c 'map'
 106 # Write should kill the cache as well.
 107 test_me 2 -c 'discard 0 1' -c 'map' -c 'write 0 1' -c 'map'
 108 # Alloc should use cached data from map
 109 test_me 1 -c 'discard 0 1' -c 'map' -c 'alloc 0'
 110 # Read should not kill the cache
 111 test_me 1 -c 'discard 0 1' -c 'map' -c 'read 0 1' -c 'map' -c 'alloc 0'