#!/bin/bash

# Test goss by using it to validate ssh service is online

expected_success_output="Count: 4, Failed: 0, Skipped: 0"
expected_fail_output="Count: 4, Failed: 1, Skipped: 0"


# Start ssh, and set goss acknowledge ssh active is correct
systemctl start ssh

goss add package openssh-server
goss add service ssh


# Confirm that goss validation succeeds with ssh active
initial_valid_output=$(goss validate)
initial_return_code=$?

grep -q "$expected_success_output" <<< "$initial_valid_output"
initial_output_correct=$?

if [[ $initial_output_correct -ne 0 ]]; then
  echo "Initial validation failed, expected:"
  echo $expected_success_output
  echo "found:"
  grep "Count" <<< "$initial_valid_output"
  exit 1
elif [[ $initial_return_code -ne 0 ]]; then
  echo "Initial validation failed: bad return code of $initial_return_code"
  exit 1
fi

# Confirm that goss validation fails with ssh inactive
systemctl stop ssh

invalid_output=$(goss validate)
invalid_return_code=$?

grep -q "$expected_fail_output" <<< "$invalid_output"
invalid_output_correct=$?

if [[ $invalid_output_correct -ne 0 ]]; then
  echo "ssh offline validation failed, expected:"
  echo $expected_fail_output
  echo "found:"
  grep "Count" <<< "$invalid_output"
  exit 1
elif [[ $invalid_return_code -ne 1 ]]; then
  echo "ssh offline validation failed: bad return code of $invalid_return_code"
  exit 1
fi

# Confirm that goss validation succeeds with ssh reactivated
systemctl start ssh

final_valid_output=$(goss validate)
final_return_code=$?

grep -q "$expected_success_output" <<< "$final_valid_output"
final_output_correct=$?

if [[ $final_output_correct -ne 0 ]]; then
  echo "Final validation failed, expected:"
  echo $expected_success_output
  echo "found:"
  grep "Count" <<< "$final_valid_output"
  exit 1
elif [[ $final_return_code -ne 0 ]]; then
  echo "Final validation failed: bad return code of $final_return_code"
  exit 1
fi
