#!/bin/bash

# FILE_PATH="/usr/lib/mx-common-lib"
FILE_PATH="/usr/lib/mx-common-lib"
source $FILE_PATH

function test_pass_result {
        if [ $1 -eq 0 ]; then
                echo -e "[\e[32mPASS\e[0m] $2"
        else
                echo -e "[\e[31mFAIL\e[0m] $2"
        fi
}

function test_fail_result {
        if [ $1 -eq 0 ]; then
                echo -e "[\e[31mFAIL\e[0m] $2"
        else
                echo -e "[\e[32mPASS\e[0m] $2"
        fi
}

# Test 1: Check if tool is installed
test -e $FILE_PATH
test_pass_result $? "Test 1: Check if tool is installed"

# Test 2: Vaild digit
for i in {0..9}; do
        check_leading_zero_digit $i
        test_pass_result $? "Test 2: Vaild digit $i"
done

# Test 3: Invaild digit
for i in {0..9}; do
        check_leading_zero_digit "0$i"
        test_fail_result $? "Test 3: Invaild digit 0$i"
done

# Test 4: Loaded module
is_module_loaded "kvm"
test_pass_result $? "Test 4: Loaded module: kvm"

# Test 4: Not loaded module
is_module_loaded "not_load_mod"
test_fail_result $? "Test 4: Not loaded module: not_load_mod"
