tailscale/logpolicy/logpolicy_test.go

// Copyright (c) 2022 Tailscale Inc & AUTHORS All rights reserved.
// Use of this source code is governed by a BSD-style
// license that can be found in the LICENSE file.

package logpolicy

import (
	"os"
	"reflect"
	"testing"
)

func TestLogHost(t *testing.T) {
	v := reflect.ValueOf(&getLogTargetOnce).Elem()
	reset := func() {
		v.Set(reflect.Zero(v.Type()))
	}
	defer reset()

	tests := []struct {
		env  string
		want string
	}{
		{"", "log.tailscale.io"},
		{"http://foo.com", "foo.com"},
		{"https://foo.com", "foo.com"},
		{"https://foo.com/", "foo.com"},
		{"https://foo.com:123/", "foo.com"},
	}
	for _, tt := range tests {
		reset()
		os.Setenv("TS_LOG_TARGET", tt.env)
		if got := LogHost(); got != tt.want {
			t.Errorf("for env %q, got %q, want %q", tt.env, got, tt.want)
		}
	}
}