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_plugin sh
  38 requires_filter cacheextents
  39 requires grep --version
  40 requires qemu-io --version
  41 requires dd iflag=count_bytes </dev/null
  42
  43 sock=$(mktemp -u /tmp/nbdkit-test-sock.XXXXXX)
  44 sockurl="nbd+unix:///?socket=$sock"
  45 pidfile="test-cacheextents.pid"
  46 accessfile="test-cacheextents-access.log"
  47 accessfile_full="$PWD/test-cacheextents-access.log"
  48 files="$pidfile $sock"
  49 rm -f $files $accessfile
  50 cleanup_fn rm -f $files
  51
  52 export accessfile_full
  53 start_nbdkit \
  54     -P $pidfile \
  55     -U $sock \
  56     --filter=cacheextents \
  57     sh - <<'EOF'
  58 echo "Call: $@" >>$accessfile_full
  59 size=4M
  60 block_size=$((1024*1024))
  61 case "$1" in
  62   thread_model) echo parallel ;;
  63   get_size) echo $size ;;
  64   can_extents) ;;
  65   extents)
  66     echo "extents request: $@" >>$accessfile_full
  67     offset=$(($4 / $block_size))
  68     count=$(($3 / $block_size))
  69     length=$(($offset + $count))
  70     for i in $(seq $offset $length); do
  71       echo ${i}M $block_size $((i%4)) >>$accessfile_full
  72       echo ${i}M $block_size $((i%4))
  73     done
  74     ;;
  75   pread) dd if=/dev/zero count=$3 iflag=count_bytes ;;
  76   can_write) ;;
  77   pwrite) dd of=/dev/null ;;
  78   can_trim) ;;
  79   trim) ;;
  80   can_zero) ;;
  81   zero) ;;
  82   *) exit 2 ;;
  83 esac
  84 EOF
  85
  86
  87 test_me() {
  88     num_accesses=$1
  89     shift
  90
  91     qemu-io -f raw "$@" "$sockurl"
  92     test "$(grep -c "^extents request: " $accessfile)" -eq "$num_accesses"
  93     ret=$?
  94     rm -f "$accessfile"
  95     return $ret
  96 }
  97
  98 # First one causes caching, the rest should be returned from cache.
  99 test_me 1 -c 'map' -c 'map' -c 'map'
 100 # First one is still cached from last time, discard should kill the cache, then
 101 # one request should go through.
 102 test_me 1 -c 'map' -c 'discard 0 1' -c 'map'
 103 # Same as above, only this time the cache is killed before all the operations as
 104 # well.  This is used from now on to clear the cache as it seems nicer and
 105 # faster than running new nbdkit for each test.
 106 test_me 2 -c 'discard 0 1' -c 'map' -c 'discard 0 1' -c 'map'
 107 # Write should kill the cache as well.
 108 test_me 2 -c 'discard 0 1' -c 'map' -c 'write 0 1' -c 'map'
 109 # Alloc should use cached data from map
 110 test_me 1 -c 'discard 0 1' -c 'map' -c 'alloc 0'
 111 # Read should not kill the cache
 112 test_me 1 -c 'discard 0 1' -c 'map' -c 'read 0 1' -c 'map' -c 'alloc 0'