read-flags.sh

#!/usr/bin/env bash
set -ex

# create a snapshot, then export it and check that setting read flags works
# by looking at --debug-ms output

function clean_up {
    rm -f test.log || true
    rbd snap remove test@snap || true
    rbd rm test || true
}

function test_read_flags {
    local IMAGE=$1
    local SET_BALANCED=$2
    local SET_LOCALIZED=$3
    local EXPECT_BALANCED=$4
    local EXPECT_LOCALIZED=$5

    local EXTRA_ARGS="--log-file test.log --debug-ms 1"
    if [ "$SET_BALANCED" = 'y' ]; then
        EXTRA_ARGS="$EXTRA_ARGS --rbd-balance-snap-reads"
    elif [ "$SET_LOCALIZED" = 'y' ]; then
        EXTRA_ARGS="$EXTRA_ARGS --rbd-localize-snap-reads"
    fi

    rbd export $IMAGE - $EXTRA_ARGS
    if [ "$EXPECT_BALANCED" = 'y' ]; then
        grep -q balance_reads test.log
    else
        grep -L balance_reads test.log | grep -q test.log
    fi
    if [ "$EXPECT_LOCALIZED" = 'y' ]; then
        grep -q localize_reads test.log
    else
        grep -L localize_reads test.log | grep -q test.log
    fi
    rm -f test.log

}

clean_up

trap clean_up INT TERM EXIT

rbd create --image-feature layering -s 6 test
rbd snap create test@snap

# export from snapshot should have read flags in log if they are set
test_read_flags test@snap n y n y

echo OK