pytest_gen.sh

#!/bin/bash
TESTFILE="$1"
ASSERTRESULT="0"

if [[ -z "${TESTFILE}" ]]; then
  echo "Please specify a testfile."
  exit 1
fi

if ! file "${TESTFILE}" | grep -q "Python script text"; then
  echo "${TESTFILE} doesn't seem to be a Python file."
  exit 1
fi

if ! grep -qo ".*\ = '''" "${TESTFILE}" || ! grep -qo "(AnsibleLintRule)" "${TESTFILE}"; then
  echo "${TESTFILE} doesn't seem to contain any tests."
  exit 1
fi

RULENAME="$(grep -o '\s.*(AnsibleLintRule)' "${TESTFILE}" | sed -e 's/(AnsibleLintRule)//g' -e 's/ //g' )"

grep -o ".*\ = '''" "${TESTFILE}" | sed 's/ =.*//g' | while read -r testinfo; do
  if echo "${testinfo}" | grep -q '^FAIL'; then
    ASSERTRESULT=1
  fi

echo "
    @pytest.mark.parametrize(
        'rule_runner', (${RULENAME},), indirect=['rule_runner']
    )
    def test_${testinfo,,}(rule_runner: Any) -> None:
        \"\"\"This Docstring should be updated.\"\"\"
        results = rule_runner.run_playbook(${testinfo})
        assert len(results) == ${ASSERTRESULT}"
done