#!/bin/bash

#

# This is a simple sanity test to satisfy the RHEL8.1 onboard gating

# requirement.

ret=0

cx2="ConnectX VPI PCIe 2.0"

cx3="[ConnectX-3]"

device_list=$(lspci | grep 'Mellanox' | grep -v 'Virtual' | grep -v 'DMA controller' | awk '{print $1}')

# Skip loop if device list is empty

[[ -z "${dev// }" ]] || for dev in "$device_list"; do

    # View Device Info

    lspci -vvv -s "$dev"

    dev_name=$(lspci -s "$dev")

    # Test Mstflint Tools

    mstflint -d "$dev" q

    let ret=$ret+$?

    # mstconfig does not support < 4th Gen devices (e.g. ConnectX-2)

    if [[ "$dev_name" != *"$cx2"* ]]; then

        mstconfig -d "${dev}" q

        let ret=$ret+$?

    fi

    mstvpd "$dev" 2>/dev/null

    let ret=$ret+$?

    mstregdump "$dev" >/dev/null

    let ret=$ret+$?

    # mstfwmanager/mstfwreset do not support ConnectX-2 or ConnectX-3

    if [[ "$dev_name" != *"$cx2"* && "$dev_name" != *"$cx3"* ]]; then

        mstfwmanager -d "$dev" --query

        let ret=$ret+$?

        mstfwreset -d "$dev" q

        let ret=$ret+$?

    fi

done

# end test - it is possible no Mellanox devices were available at all to test,

# in which case we automatically mark the test PASS (exit 0)

exit $ret