tests/test-python-thread-model.sh

   1 #!/usr/bin/env bash
   2 # nbdkit
   3 # Copyright Red Hat
   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 -e
  35 set -x
  36
  37 SCRIPT="$SRCDIR/python-thread-model.py"
  38 if ! test -d "$SRCDIR" || ! test -f "$SCRIPT"; then
  39     echo "$0: could not locate python-thread-model.py"
  40     exit 1
  41 fi
  42
  43 skip_if_valgrind "because Python code leaks memory"
  44 requires nbdsh --version
  45
  46 # Check the plugin is loadable.
  47 nbdkit python $SCRIPT --dump-plugin
  48
  49 # This test only works on modern Linux (with pipe2, accept4 etc) where
  50 # we are able to issue parallel requests.  Other platforms have more
  51 # restrictive thread models.
  52 requires sh -c "nbdkit python $SCRIPT --dump-plugin |
  53                 grep '^thread_model=parallel'"
  54
  55 pid=test-python-thread-model.pid
  56 sock=$(mktemp -u /tmp/nbdkit-test-sock.XXXXXX)
  57 files="$out $pid $sock"
  58 rm -f $files
  59 cleanup_fn rm -f $files
  60
  61 start_nbdkit -P $pid -U $sock python $SCRIPT
  62
  63 export sock
  64 nbdsh -c '
  65 import os
  66 import time
  67
  68 h.connect_unix(os.environ["sock"])
  69
  70 # We should be able to issue multiple requests in parallel,
  71 # and the total time taken should not be much more than 10 seconds
  72 # because all sleeps in the plugin should happen in parallel.
  73 start_t = time.time()
  74 for i in range(10):
  75     buf = nbd.Buffer(512)
  76     h.aio_pread(buf, 0)
  77
  78 while h.aio_in_flight() > 0:
  79     h.poll(-1)
  80 end_t = time.time()
  81
  82 t = end_t - start_t
  83 print(t)
  84
  85 # Since we launched 10 requests, if we serialized on them we
  86 # would have waited at least 100 seconds.  We would expect to
  87 # wait around 10 seconds, but for flexibility on slow servers
  88 # any test < 100 should be fine.
  89 assert t <= 50
  90 '