Add --base-url flag for connector testability

cmd/baton-splunk/main.go

@@ -67,6 +67,7 @@ func getConnector(ctx context.Context, c *cfg.Splunk) (types.ConnectorServer, er
 			Unsafe:  c.Unsafe,
 			Verbose: c.Verbose,
 			Cloud:   c.Cloud,
+			BaseURL: c.BaseUrl,
 		},
 		c.Deployments,
 	)

pkg/config/config.go

@@ -41,6 +41,12 @@ var (
 		"deployments",
 		field.WithDescription("Limit syncing to specific deployments by specifying cloud deployment names or IP addresses of on-premise deployments."),
 	)
+	BaseURL = field.StringField(
+		"base-url",
+		field.WithDescription("Override the Splunk API URL (for testing)"),
+		field.WithHidden(true),
+		field.WithExportTarget(field.ExportTargetCLIOnly),
+	)
 )
 
 //go:generate go run ./gen
@@ -52,4 +58,5 @@ var Config = field.NewConfiguration([]field.SchemaField{
 	Verbose,
 	Cloud,
 	Deployments,
+	BaseURL,
 })

pkg/connector/connector.go

@@ -107,6 +107,7 @@ type CLIConfig struct {
 	Unsafe  bool
 	Verbose bool
 	Cloud   bool
+	BaseURL string
 }
 
 // New returns the Splunk connector.
@@ -134,7 +135,7 @@ func New(ctx context.Context, auth string, config CLIConfig, deployments []strin
 	}
 
 	return &Splunk{
-		client:      splunk.NewClient(httpClient, auth, config.Cloud),
+		client:      splunk.NewClient(httpClient, auth, config.Cloud, config.BaseURL),
 		verbose:     config.Verbose,
 		cloud:       config.Cloud,
 		deployments: deployments,

pkg/splunk/client.go

@@ -35,6 +35,7 @@
 	Auth       string
 	Cloud      bool
 	Deployment string
+	BaseURL    string
 }
 
 type PaginationData struct {
@@ -53,12 +54,13 @@
 	PaginationData `json:"paging"`
 }
 
-func NewClient(httpClient *http.Client, auth string, cloud bool) *Client {
+func NewClient(httpClient *http.Client, auth string, cloud bool, baseURL string) *Client {
 	return &Client{
 		httpClient: httpClient,
 		Auth:       auth,
 		Cloud:      cloud,
 		Deployment: Localhost,
+		BaseURL:    baseURL,
 	}
 }
 
@@ -76,11 +78,13 @@
 
 // GetUrl returns the full URL for the given endpoint based on platform.
 func (c *Client) CreateUrl(endpoint string) string {
+	if c.BaseURL != "" {
+		return strings.TrimSuffix(c.BaseURL, "/") + endpoint
+	}
 	if c.Cloud {
 		return fmt.Sprintf(CloudBaseURL, c.Deployment) + endpoint
-	} else {
-		return fmt.Sprintf(BaseURL, c.Deployment) + endpoint
 	}
+	return fmt.Sprintf(BaseURL, c.Deployment) + endpoint
 }
 
 func (c *Client) IsCloudPlatform() bool {
@@ -389,7 +393,7 @@
 	req.Header.Set("content-type", "application/json")
 	req.Header.Set("Authorization", c.Auth)
 
 	rawResponse, err := c.httpClient.Do(req)
 	if err != nil {
 		return err
 	}