121 lines
3.5 KiB
Go
121 lines
3.5 KiB
Go
// Copyright (c) 2025-2026 Probo Inc <hello@getprobo.com>.
|
|
//
|
|
// Permission to use, copy, modify, and/or distribute this software for any
|
|
// purpose with or without fee is hereby granted, provided that the above
|
|
// copyright notice and this permission notice appear in all copies.
|
|
//
|
|
// THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES WITH
|
|
// REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
|
|
// AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY SPECIAL, DIRECT,
|
|
// INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
|
|
// LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
|
|
// OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
|
// PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
package version_test
|
|
|
|
import (
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"go.probo.inc/probo/pkg/version"
|
|
)
|
|
|
|
func TestGetBuildInfo(t *testing.T) {
|
|
info := version.GetBuildInfo()
|
|
|
|
// These should always be present
|
|
assert.NotEmpty(t, info.GoVersion, "Go version should not be empty")
|
|
assert.True(t, strings.HasPrefix(info.GoVersion, "go"), "Go version should start with 'go'")
|
|
|
|
// Version should have a default value
|
|
assert.NotEmpty(t, info.Version, "Version should not be empty")
|
|
|
|
// Commit and BuildDate may be "unknown" in test environment
|
|
assert.NotEmpty(t, info.Commit, "Commit should not be empty")
|
|
assert.NotEmpty(t, info.BuildDate, "BuildDate should not be empty")
|
|
}
|
|
|
|
func TestUserAgent(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
component string
|
|
checks []string
|
|
}{
|
|
{
|
|
name: "ACME client user agent",
|
|
component: "acme-client",
|
|
checks: []string{
|
|
"Probo/",
|
|
"acme-client",
|
|
"Go/go",
|
|
},
|
|
},
|
|
{
|
|
name: "Probod user agent",
|
|
component: "probod",
|
|
checks: []string{
|
|
"Probo/",
|
|
"probod",
|
|
"Go/go",
|
|
},
|
|
},
|
|
{
|
|
name: "Custom component",
|
|
component: "test-component",
|
|
checks: []string{
|
|
"Probo/",
|
|
"test-component",
|
|
"Go/go",
|
|
},
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
ua := version.UserAgent(tt.component)
|
|
require.NotEmpty(t, ua, "User agent should not be empty")
|
|
|
|
for _, check := range tt.checks {
|
|
assert.Contains(t, ua, check, "User agent should contain %s", check)
|
|
}
|
|
|
|
// Check format structure
|
|
// Should be like: Probo/version (component; ...) Go/version
|
|
assert.True(t, strings.HasPrefix(ua, "Probo/"), "User agent should start with Probo/")
|
|
assert.Contains(t, ua, "(", "User agent should have opening parenthesis")
|
|
assert.Contains(t, ua, ")", "User agent should have closing parenthesis")
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestUserAgentConsistency(t *testing.T) {
|
|
// Test that multiple calls return consistent results
|
|
ua1 := version.UserAgent("test")
|
|
ua2 := version.UserAgent("test")
|
|
assert.Equal(t, ua1, ua2, "User agent should be consistent across calls")
|
|
|
|
// Test that different components produce different user agents
|
|
ua3 := version.UserAgent("different")
|
|
assert.NotEqual(t, ua1, ua3, "Different components should produce different user agents")
|
|
|
|
// But they should have the same version info
|
|
parts1 := strings.SplitN(ua1, " ", 2)
|
|
parts3 := strings.SplitN(ua3, " ", 2)
|
|
assert.Equal(t, parts1[0], parts3[0], "Version part should be the same")
|
|
}
|
|
|
|
func BenchmarkUserAgent(b *testing.B) {
|
|
for i := 0; i < b.N; i++ {
|
|
_ = version.UserAgent("benchmark")
|
|
}
|
|
}
|
|
|
|
func BenchmarkGetBuildInfo(b *testing.B) {
|
|
for i := 0; i < b.N; i++ {
|
|
_ = version.GetBuildInfo()
|
|
}
|
|
}
|