package audit

import (
	"testing"
	"time"
)

func TestCheckSSH(t *testing.T) {

	MountPingDB()

	host1 := "102.1.41.3:22" // should evaluate to false
	host2 := "127.0.0.1:22"  // should evaluate to true

	if err := BeginSshPing(host1); err != nil {
		panic(err)
	}

	if err := BeginSshPing(host2); err != nil {
		panic(err)
	}

	time.Sleep(5 * time.Second)

	res, err := CheckSshPing(host1)
	if err != nil {
		panic(err)
	}
	if res {
		panic("check host 1 - res should be false")
	}

	res2, err := CheckSshPing(host2)
	if err != nil {
		panic(err)
	}
	if !res2 {
		panic("check host 2 - res should be true")
	}
}