package ansible

import (
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
)

func TestSSH_NewSSHClient_Good_CustomConfig(t *testing.T) {
	cfg := SSHConfig{
		Host: "localhost",
		Port: 2222,
		User: "root",
	}

	client, err := NewSSHClient(cfg)
	assert.NoError(t, err)
	assert.NotNil(t, client)
	assert.Equal(t, "localhost", client.host)
	assert.Equal(t, 2222, client.port)
	assert.Equal(t, "root", client.user)
	assert.Equal(t, 30*time.Second, client.timeout)
}

func TestSSH_NewSSHClient_Good_Defaults(t *testing.T) {
	cfg := SSHConfig{
		Host: "localhost",
	}

	client, err := NewSSHClient(cfg)
	assert.NoError(t, err)
	assert.Equal(t, 22, client.port)
	assert.Equal(t, "root", client.user)
	assert.Equal(t, 30*time.Second, client.timeout)
}

func TestSSH_SetBecome_Good_DisablesAndClearsState(t *testing.T) {
	client := &SSHClient{}

	client.SetBecome(true, "admin", "secret")
	become, user, password := client.BecomeState()
	assert.True(t, become)
	assert.Equal(t, "admin", user)
	assert.Equal(t, "secret", password)

	client.SetBecome(false, "", "")
	become, user, password = client.BecomeState()
	assert.False(t, become)
	assert.Empty(t, user)
	assert.Empty(t, password)
}