// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package vespa

import (
	"fmt"
	"os"
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/vespa-engine/vespa/client/go/internal/admin/trace"
)

func TestDetectHostname(t *testing.T) {
	trace.AdjustVerbosity(0)
	t.Setenv("VESPA_HOSTNAME", "foo.bar")
	got, err := FindOurHostname()
	assert.Nil(t, err)
	assert.Equal(t, "foo.bar", got)
	os.Unsetenv("VESPA_HOSTNAME")
	got, err = findOurHostnameFrom("bar.foo.123")
	fmt.Fprintln(os.Stderr, "findOurHostname from bar.foo.123 returns:", got, "with error:", err)
	assert.NotEqual(t, "", got)
	parts := strings.Split(got, ".")
	if len(parts) > 1 {
		expanded, err2 := findOurHostnameFrom(parts[0])
		fmt.Fprintln(os.Stderr, "findOurHostname from", parts[0], "returns:", expanded, "with error:", err2)
		assert.Equal(t, got, expanded)
	}
	got, err = findOurHostnameFrom("")
	assert.NotEqual(t, "", got)
	fmt.Fprintln(os.Stderr, "findOurHostname('') returns:", got, "with error:", err)
}