tests/test-python-thread-model.sh

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