48 lines
1.0 KiB
Go
48 lines
1.0 KiB
Go
package httpclient
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/terraform/version"
|
|
)
|
|
|
|
func TestUserAgentString_env(t *testing.T) {
|
|
expectedBase := fmt.Sprintf(userAgentFormat, version.Version)
|
|
if oldenv, isSet := os.LookupEnv(uaEnvVar); isSet {
|
|
defer os.Setenv(uaEnvVar, oldenv)
|
|
} else {
|
|
defer os.Unsetenv(uaEnvVar)
|
|
}
|
|
|
|
for i, c := range []struct {
|
|
expected string
|
|
additional string
|
|
}{
|
|
{expectedBase, ""},
|
|
{expectedBase, " "},
|
|
{expectedBase, " \n"},
|
|
|
|
{fmt.Sprintf("%s test/1", expectedBase), "test/1"},
|
|
{fmt.Sprintf("%s test/2", expectedBase), "test/2 "},
|
|
{fmt.Sprintf("%s test/3", expectedBase), " test/3 "},
|
|
{fmt.Sprintf("%s test/4", expectedBase), "test/4 \n"},
|
|
} {
|
|
t.Run(fmt.Sprintf("%d", i), func(t *testing.T) {
|
|
if c.additional == "" {
|
|
os.Unsetenv(uaEnvVar)
|
|
} else {
|
|
os.Setenv(uaEnvVar, c.additional)
|
|
}
|
|
|
|
actual := UserAgentString()
|
|
|
|
if c.expected != actual {
|
|
t.Fatalf("Expected User-Agent '%s' does not match '%s'", c.expected, actual)
|
|
}
|
|
})
|
|
}
|
|
|
|
}
|