package palette

import (
	"testing"
	"time"

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

func TestSessionResultsAlwaysQualifyClientOrigin(t *testing.T) {
	for _, tc := range []struct {
		name   string
		result Result
		want   string
	}{
		{"local only", NewActiveSessionResult(testExactTarget("sample", 1), time.Time{}), "Switch to session sample"},
		{"hybrid local", NewActiveSessionResultWithDisplayOrigin(testExactTarget("sample", 1), time.Time{}, "local"), "Switch to session sample@local"},
		{"serving remote", NewActiveSessionResultWithDisplayOrigin(testExactTarget("sample", 1), time.Time{}, "host-a"), "Switch to session sample@host-a"},
		{"imported local", NewImportedSessionResult("local", "key", "sample", "local", "up", ""), "Switch to session sample@local"},
	} {
		t.Run(tc.name, func(t *testing.T) { require.Equal(t, tc.want, tc.result.DisplayText()) })
	}
}