github.com/Azure/aad-pod-identity@v1.8.17/version/version_test.go (about)

     1  package version
     2  
     3  import (
     4  	"fmt"
     5  	"strings"
     6  	"testing"
     7  )
     8  
     9  func TestVersion(t *testing.T) {
    10  	BuildDate = "Now"
    11  	GitCommit = "Commit"
    12  	NMIVersion = "NMI version"
    13  	expectedUserAgentStr := fmt.Sprintf("aad-pod-identity/%s/%s/%s/%s", "NMI", NMIVersion, GitCommit, BuildDate)
    14  	gotUserAgentStr := GetUserAgent("NMI", NMIVersion)
    15  	if !strings.EqualFold(expectedUserAgentStr, gotUserAgentStr) {
    16  		t.Fatalf("got unexpected user agent string: %s. Expected: %s.", gotUserAgentStr, expectedUserAgentStr)
    17  	}
    18  }
    19  
    20  func TestGetUserAgent(t *testing.T) {
    21  	BuildDate = "now"
    22  	GitCommit = "commit"
    23  	NMIVersion = "version"
    24  
    25  	tests := []struct {
    26  		name              string
    27  		customUserAgent   string
    28  		expectedUserAgent string
    29  	}{
    30  		{
    31  			name:              "default NMI user agent",
    32  			customUserAgent:   "",
    33  			expectedUserAgent: "aad-pod-identity/NMI/version/commit/now",
    34  		},
    35  		{
    36  			name:              "default NMI user agent and custom user agent",
    37  			customUserAgent:   "managedBy:aks",
    38  			expectedUserAgent: "aad-pod-identity/NMI/version/commit/now managedBy:aks",
    39  		},
    40  	}
    41  
    42  	for _, test := range tests {
    43  		t.Run(test.name, func(t *testing.T) {
    44  			customUserAgent = &test.customUserAgent
    45  			actualUserAgent := GetUserAgent("NMI", NMIVersion)
    46  			if !strings.EqualFold(test.expectedUserAgent, actualUserAgent) {
    47  				t.Fatalf("got unexpected user agent string: %s. Expected: %s.", test.expectedUserAgent, actualUserAgent)
    48  			}
    49  		})
    50  	}
    51  }