#!/bin/bash

. /usr/share/beakerlib/beakerlib.sh || exit 1


function assert_execution_order(){
    local input_file="$tmp/run/plans${plan}/data/execution_order"
    sed -n 's;^.*execute/data\(.\+\)/data;\1;p' "$input_file" > $tmp/outcome
    rlRun "diff -pu $tmp/outcome $tmp/EXPECTED-EXECUTION" 0 "Verify execution order"
}

function assert_discover_order(){
    local input_file="$rlRun_LOG"
    # Discovered tests are printed with leading spaces
    sed -n 's;^ \+\(/.\+\);\1;p' "$input_file" > $tmp/outcome
    rlRun "diff -pu $tmp/outcome $tmp/EXPECTED-DISCOVERY" 0 "Verify discovery order"
}

function run_test(){
    rlPhaseStartTest "$plan"
        rlRun -s "tmt run --all --id $tmp/run --scratch plans -n $plan$ discover -vvv"
        rlLog "execution_order content:\n$(cat $tmp/run/plans${plan}/data/execution_order)"

        assert_discover_order
        assert_execution_order

        # Now in two parts
        rlRun -s "tmt run --id $tmp/run --scratch plans -n $plan$ discover -vvv"
        assert_discover_order

        rlRun -s "tmt run --all --id $tmp/run plans -n $plan$"
        assert_execution_order
    rlPhaseEnd
}


rlJournalStart
    rlPhaseStartSetup
        rlRun "pushd order"
        rlRun "tmp=\$(mktemp -d)" 0 "Creating tmp directory"
    rlPhaseEnd


    ### New test begins
    plan="/single-without-order-tag"
    cat > $tmp/EXPECTED-DISCOVERY <<EOF
/tests/no-order-0
/tests/no-order-1
/tests/no-order-2
EOF
    cat > $tmp/EXPECTED-EXECUTION <<EOF
/guest/default-0/tests/no-order-0-1
/guest/default-0/tests/no-order-1-2
/guest/default-0/tests/no-order-2-3
EOF
    run_test

    ### New test begins
    plan="/single-without-order-name"
    cat > $tmp/EXPECTED-DISCOVERY <<EOF
/tests/no-order-0
/tests/no-order-1
/tests/no-order-2
EOF
    cat > $tmp/EXPECTED-EXECUTION <<EOF
/guest/default-0/tests/no-order-0-1
/guest/default-0/tests/no-order-1-2
/guest/default-0/tests/no-order-2-3
EOF
    run_test

    ### New test begins
    plan="/single-enumerate"
    cat > $tmp/EXPECTED-DISCOVERY <<EOF
/tests/no-order-2
/tests/no-order-0
/tests/no-order-1
EOF
    cat > $tmp/EXPECTED-EXECUTION <<EOF
/guest/default-0/tests/no-order-2-1
/guest/default-0/tests/no-order-0-2
/guest/default-0/tests/no-order-1-3
EOF
    run_test

    ### New test begins
    plan="/single-enumerate-and-order"
    cat > $tmp/EXPECTED-DISCOVERY <<EOF
/tests/no-order-2
/tests/order-80
/tests/no-order-0
/tests/no-order-1
EOF
    cat > $tmp/EXPECTED-EXECUTION <<EOF
/guest/default-0/tests/no-order-2-1
/guest/default-0/tests/order-80-2
/guest/default-0/tests/no-order-0-3
/guest/default-0/tests/no-order-1-4
EOF
    run_test

    ### New test begins
    plan="/single-order"
    cat > $tmp/EXPECTED-DISCOVERY <<EOF
/tests/order-10
/tests/no-order-0
/tests/no-order-1
/tests/no-order-2
/tests/order-80
EOF
    cat > $tmp/EXPECTED-EXECUTION <<EOF
/guest/default-0/tests/order-10-1
/guest/default-0/tests/no-order-0-2
/guest/default-0/tests/no-order-1-3
/guest/default-0/tests/no-order-2-4
/guest/default-0/tests/order-80-5
EOF
    run_test

    ### New test begins
    plan="/multiple-by-enumerate"
    cat > $tmp/EXPECTED-DISCOVERY <<EOF
/enumerate-and-order/tests/no-order-2
/enumerate-and-order/tests/order-80
/enumerate-and-order/tests/no-order-0
/enumerate-and-order/tests/no-order-1
/by-order-attribute/tests/order-10
/by-order-attribute/tests/no-order-0
/by-order-attribute/tests/no-order-1
/by-order-attribute/tests/no-order-2
/by-order-attribute/tests/order-80
/third/order-20
/third/order-default
EOF
    cat > $tmp/EXPECTED-EXECUTION <<EOF
/guest/default-0/enumerate-and-order/tests/no-order-2-1
/guest/default-0/enumerate-and-order/tests/order-80-2
/guest/default-0/enumerate-and-order/tests/no-order-0-3
/guest/default-0/enumerate-and-order/tests/no-order-1-4
/guest/default-0/by-order-attribute/tests/order-10-5
/guest/default-0/by-order-attribute/tests/no-order-0-6
/guest/default-0/by-order-attribute/tests/no-order-1-7
/guest/default-0/by-order-attribute/tests/no-order-2-8
/guest/default-0/by-order-attribute/tests/order-80-9
/guest/default-0/third/order-20-10
/guest/default-0/third/order-default-11
EOF
    run_test

    ### New test begins
    plan="/multiple-by-order"
    cat > $tmp/EXPECTED-DISCOVERY <<EOF
/order-10/tests/order-10
/order-10/tests/no-order-0
/order-10/tests/no-order-1
/order-10/tests/no-order-2
/order-10/tests/order-80
/order-default/order-20
/order-default/order-default
/order-80/tests/no-order-2
/order-80/tests/order-80
/order-80/tests/no-order-0
/order-80/tests/no-order-1
EOF
    cat > $tmp/EXPECTED-EXECUTION <<EOF
/guest/default-0/order-10/tests/order-10-1
/guest/default-0/order-10/tests/no-order-0-2
/guest/default-0/order-10/tests/no-order-1-3
/guest/default-0/order-10/tests/no-order-2-4
/guest/default-0/order-10/tests/order-80-5
/guest/default-0/order-default/order-20-6
/guest/default-0/order-default/order-default-7
/guest/default-0/order-80/tests/no-order-2-8
/guest/default-0/order-80/tests/order-80-9
/guest/default-0/order-80/tests/no-order-0-10
/guest/default-0/order-80/tests/no-order-1-11
EOF
    run_test

    # Tests defined using the 'shell' discover method should be
    # executed in the order provided in the config file

    plan="/shell-preserve-order"

    cat > $tmp/EXPECTED-DISCOVERY <<EOF
/tests/one
/tests/two
/tests/three
EOF

    cat > $tmp/EXPECTED-EXECUTION <<EOF
/guest/default-0/tests/one-1
/guest/default-0/tests/two-2
/guest/default-0/tests/three-3
EOF
    run_test

    rlPhaseStartCleanup
        rlRun 'rm -rf $tmp' 0 "Remove tmp directory"
    rlPhaseEnd
rlJournalEnd