Support for Default tags

client/configuration.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"net/http"
 	"os"
+	"strings"
 
 	"github.com/infobloxopen/bloxone-go-client/internal"
 )
@@ -18,11 +19,11 @@ const (
 	HeaderClient        = "x-infoblox-client"
 	HeaderSDK           = "x-infoblox-sdk"
 	HeaderAuthorization = "Authorization"
+	version             = "0.1"
+	sdkIdentifier       = "golang-sdk"
+	clientIdentifier    = "automation"
 )
 
-const version = "0.1"
-const sdkIdentifier = "golang-sdk"
-
 // Configuration stores the configuration of the API client
 type Configuration struct {
 	// ClientName is the name of the client using the SDK.
@@ -43,6 +44,9 @@ type Configuration struct {
 	// HTTPClient to use for the SDK.
 	// Optional. The default HTTPClient will be used if not provided.
 	HTTPClient *http.Client
+
+	// Default global tags the client can set for all requests.
+	DefaultTags map[string]string
 }
 
 func (c Configuration) internal(basePath string) (*internal.Configuration, error) {
@@ -83,12 +87,26 @@ func (c Configuration) internal(basePath string) (*internal.Configuration, error
 
 	userAgent := fmt.Sprintf("bloxone-%s/%s", sdkIdentifier, version)
 
-	return &internal.Configuration{
+	ic := &internal.Configuration{
 		DefaultHeader:    defaultHeaders,
 		UserAgent:        userAgent,
 		Debug:            false,
 		OperationServers: nil,
 		Servers:          []internal.ServerConfiguration{{URL: cspURL}},
 		HTTPClient:       httpClient,
-	}, nil
+		DefaultTags:      make(map[string]string),
+	}
+	// Add default tags set
+	if c.DefaultTags != nil {
+		ic.AddDefaultTags(c.DefaultTags)
+	}
+
+	// setting up custom tag to identify the client
+	dfTags := make(map[string]string)
+	// Extract client from ClientName string
+	// Format: <client>/version#commit
+	dfTags[clientIdentifier] = strings.Split(c.ClientName, "/")[0]
+	ic.AddDefaultTags(dfTags)
+
+	return ic, nil
 }

client/configuration_test.go

@@ -0,0 +1,133 @@
+package client
+
+import (
+    "fmt"
+    "net/http"
+    "os"
+    "reflect"
+    "testing"
+
+    "github.com/infobloxopen/bloxone-go-client/internal"
+)
+
+func TestConfiguration_internal(t *testing.T) {
+    type fields struct {
+        ClientName  string
+        CSPURL      string
+        APIKey      string
+        HTTPClient  *http.Client
+        DefaultTags map[string]string
+    }
+    type args struct {
+        basePath string
+    }
+    tests := []struct {
+        name    string
+        fields  fields
+        args    args
+        want    *internal.Configuration
+        wantErr bool
+    }{
+        {
+            "empty API key",
+            fields{
+                APIKey: "",
+            },
+            args{basePath: ""},
+            nil,
+            true,
+        },
+        {"empty clientName",
+            fields{
+                ClientName: "",
+            },
+            args{basePath: ""},
+            nil,
+            true,
+        },
+        {
+            "empty DefaultTags",
+            fields{
+                ClientName: "terraform/v1.1#yug278872h",
+                APIKey:     "12323455",
+            },
+            args{basePath: ""},
+            &internal.Configuration{
+                DefaultHeader: map[string]string{
+                    HeaderAuthorization: "Token 12323455",
+                    HeaderClient:        "terraform/v1.1#yug278872h",
+                    HeaderSDK:           sdkIdentifier,
+                },
+                Debug:      false,
+                UserAgent:  fmt.Sprintf("bloxone-%s/%s", sdkIdentifier, version),
+                Servers:    []internal.ServerConfiguration{{URL: "https://csp.infoblox.com"}},
+                HTTPClient: http.DefaultClient,
+                DefaultTags: map[string]string{
+                    clientIdentifier: "terraform",
+                },
+            },
+            false,
+        },
+        {
+            "DefaultTags provided",
+            fields{
+                CSPURL:     "https://stage.csp.infoblox.com",
+                ClientName: "terraformv1.1#yug278872h",
+                APIKey:     "12323455",
+                DefaultTags: map[string]string{
+                    "site": "A",
+                    "env":  "test",
+                },
+            },
+            args{basePath: ""},
+            &internal.Configuration{
+                DefaultHeader: map[string]string{
+                    HeaderAuthorization: "Token 12323455",
+                    HeaderClient:        "terraformv1.1#yug278872h",
+                    HeaderSDK:           sdkIdentifier,
+                },
+                Debug:      false,
+                UserAgent:  fmt.Sprintf("bloxone-%s/%s", sdkIdentifier, version),
+                Servers:    []internal.ServerConfiguration{{URL: "https://stage.csp.infoblox.com"}},
+                HTTPClient: http.DefaultClient,
+                DefaultTags: map[string]string{
+                    clientIdentifier: "terraformv1.1#yug278872h",
+                    "site":           "A",
+                    "env":            "test",
+                },
+            },
+            false,
+        },
+    }
+    for _, tt := range tests {
+        t.Run(tt.name, func(t *testing.T) {
+            var curEnvVal string
+            c := Configuration{
+                ClientName:  tt.fields.ClientName,
+                CSPURL:      tt.fields.CSPURL,
+                APIKey:      tt.fields.APIKey,
+                HTTPClient:  tt.fields.HTTPClient,
+                DefaultTags: tt.fields.DefaultTags,
+            }
+            if c.CSPURL != "" {
+                curEnvVal = os.Getenv(ENVBloxOneCSPURL)
+                t.Setenv(ENVBloxOneCSPURL, c.CSPURL)
+            }
+
+            got, err := c.internal(tt.args.basePath)
+            if (err != nil) != tt.wantErr {
+                t.Errorf("internal() error = %v, wantErr %v", err, tt.wantErr)
+                return
+            }
+            if !reflect.DeepEqual(got, tt.want) {
+                t.Errorf("internal() got = %v, want %v", got, tt.want)
+            }
+            t.Cleanup(func() {
+                // Set it to the value prior to executing the test
+                if c.CSPURL != "" {
+                    t.Setenv(ENVBloxOneCSPURL, curEnvVal)
+                }
+            })
+        })
+    }
+}