package exoscale

import (
	"os"
	"testing"
	"time"

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

var (
	exoscaleLiveTest  bool
	exoscaleAPIKey    string
	exoscaleAPISecret string
	exoscaleDomain    string
)

func init() {
	exoscaleAPISecret = os.Getenv("EXOSCALE_API_SECRET")
	exoscaleAPIKey = os.Getenv("EXOSCALE_API_KEY")
	exoscaleDomain = os.Getenv("EXOSCALE_DOMAIN")
	if len(exoscaleAPIKey) > 0 && len(exoscaleAPISecret) > 0 && len(exoscaleDomain) > 0 {
		exoscaleLiveTest = true
	}
}

func restoreEnv() {
	os.Setenv("EXOSCALE_API_KEY", exoscaleAPIKey)
	os.Setenv("EXOSCALE_API_SECRET", exoscaleAPISecret)
}

func TestNewDNSProviderValid(t *testing.T) {
	defer restoreEnv()
	os.Setenv("EXOSCALE_API_KEY", "")
	os.Setenv("EXOSCALE_API_SECRET", "")

	_, err := NewDNSProviderClient("example@example.com", "123", "")
	assert.NoError(t, err)
}

func TestNewDNSProviderValidEnv(t *testing.T) {
	defer restoreEnv()
	os.Setenv("EXOSCALE_API_KEY", "example@example.com")
	os.Setenv("EXOSCALE_API_SECRET", "123")

	_, err := NewDNSProvider()
	assert.NoError(t, err)
}

func TestNewDNSProviderMissingCredErr(t *testing.T) {
	os.Setenv("EXOSCALE_API_KEY", "")
	os.Setenv("EXOSCALE_API_SECRET", "")
	defer restoreEnv()

	_, err := NewDNSProvider()
	assert.EqualError(t, err, "Exoscale: some credentials information are missing: EXOSCALE_API_KEY,EXOSCALE_API_SECRET")
}

func TestExtractRootRecordName(t *testing.T) {
	provider, err := NewDNSProviderClient("example@example.com", "123", "")
	assert.NoError(t, err)

	zone, recordName, err := provider.FindZoneAndRecordName("_acme-challenge.bar.com.", "bar.com")
	assert.NoError(t, err)
	assert.Equal(t, "bar.com", zone)
	assert.Equal(t, "_acme-challenge", recordName)
}

func TestExtractSubRecordName(t *testing.T) {
	provider, err := NewDNSProviderClient("example@example.com", "123", "")
	assert.NoError(t, err)

	zone, recordName, err := provider.FindZoneAndRecordName("_acme-challenge.foo.bar.com.", "foo.bar.com")
	assert.NoError(t, err)
	assert.Equal(t, "bar.com", zone)
	assert.Equal(t, "_acme-challenge.foo", recordName)
}

func TestLiveExoscalePresent(t *testing.T) {
	if !exoscaleLiveTest {
		t.Skip("skipping live test")
	}

	provider, err := NewDNSProviderClient(exoscaleAPIKey, exoscaleAPISecret, "")
	assert.NoError(t, err)

	err = provider.Present(exoscaleDomain, "", "123d==")
	assert.NoError(t, err)

	// Present Twice to handle create / update
	err = provider.Present(exoscaleDomain, "", "123d==")
	assert.NoError(t, err)
}

func TestLiveExoscaleCleanUp(t *testing.T) {
	if !exoscaleLiveTest {
		t.Skip("skipping live test")
	}

	time.Sleep(time.Second * 1)

	provider, err := NewDNSProviderClient(exoscaleAPIKey, exoscaleAPISecret, "")
	assert.NoError(t, err)

	err = provider.CleanUp(exoscaleDomain, "", "123d==")
	assert.NoError(t, err)
}