make it possible to set a custom http header (#134)

rockset · Aug 23, 2023 · 338f6cd · 338f6cd
1 parent a4f9954
commit 338f6cd
Show file tree

Hide file tree

Showing 45 changed files with 14 additions and 2 deletions.
diff --git a/query_lambdas_test.go b/query_lambdas_test.go
@@ -35,7 +35,7 @@ func TestRockClient_CreateQueryLambda(t *testing.T) {
 // test fixtures
 const (
 	qlName    = "events"
-	qlVersion = "1921bcad817b5cc2"
+	qlVersion = "0eb7783c81ef339e"
 	qlTag     = "test"
 )
 

diff --git a/rockset.go b/rockset.go
@@ -128,6 +128,13 @@ func WithHTTPClient(c *http.Client) RockOption {
 	}
 }
 
+// WithCustomHeader sets a custom HTTP header for all Rockset API calls.
+func WithCustomHeader(key, value string) RockOption {
+	return func(rc *RockConfig) {
+		rc.cfg.AddDefaultHeader(key, value)
+	}
+}
+
 // WithRetry sets the Retrier the RockClient uses to retry requests which return a Error that can be retried.
 // The errors which can be retried are configurable using the ExponentialRetry field RetryableErrorCheck.
 //

diff --git a/rockset_test.go b/rockset_test.go
@@ -132,7 +132,8 @@ func vcrClient(name string) (*rockset.RockClient, func(string) string, error) {
 	}
 
 	vcr := govcr.NewVCR(govcr.NewCassetteLoader(path), settings...)
-	options = append(options, rockset.WithHTTPClient(vcr.HTTPClient()))
+	options = append(options, rockset.WithHTTPClient(vcr.HTTPClient()),
+		rockset.WithCustomHeader("x-rockset-test", "go-client"))
 
 	rc, err := rockset.NewClient(options...)
 	return rc, randFn, err
@@ -168,6 +169,10 @@ func vcrSettings(offline bool) []govcr.Setting {
 				return govcr.DefaultHeaderMatcher(httpRequest, trackRequest)
 			},
 		),
+		govcr.WithTrackRecordingMutators(track.TrackRequestDeleteHeaderKeys(authHeader)),
+		govcr.WithTrackRecordingMutators(track.ResponseDeleteTLS()),
+		govcr.WithTrackReplayingMutators(track.TrackRequestDeleteHeaderKeys(authHeader)),
+		govcr.WithTrackReplayingMutators(track.ResponseDeleteTLS()),
 	}
 	if offline {
 		settings = append(settings, govcr.WithOfflineMode())

diff --git a/vcr/CollectionTestSuite/TestCreateSampleCitiesCollection.cassette.gz b/vcr/CollectionTestSuite/TestCreateSampleCitiesCollection.cassette.gz
diff --git a/vcr/CollectionTestSuite/TestCreateSampleMoviesCollection.cassette.gz b/vcr/CollectionTestSuite/TestCreateSampleMoviesCollection.cassette.gz
diff --git a/vcr/CollectionTestSuite/TestGetCollection.cassette.gz b/vcr/CollectionTestSuite/TestGetCollection.cassette.gz
diff --git a/vcr/CollectionTestSuite/TestListAllCollections.cassette.gz b/vcr/CollectionTestSuite/TestListAllCollections.cassette.gz
diff --git a/vcr/CollectionTestSuite/TestListCollectionsInWorkspace.cassette.gz b/vcr/CollectionTestSuite/TestListCollectionsInWorkspace.cassette.gz
diff --git a/vcr/CollectionTestSuite/TestUpdateCollection.cassette.gz b/vcr/CollectionTestSuite/TestUpdateCollection.cassette.gz
diff --git a/vcr/IntegrationTestSuite/TestCreateGCSIntegration.cassette.gz b/vcr/IntegrationTestSuite/TestCreateGCSIntegration.cassette.gz
diff --git a/vcr/IntegrationTestSuite/TestCreateS3Integration.cassette.gz b/vcr/IntegrationTestSuite/TestCreateS3Integration.cassette.gz
diff --git a/vcr/IntegrationTestSuite/TestGetIntegration.cassette.gz b/vcr/IntegrationTestSuite/TestGetIntegration.cassette.gz
diff --git a/vcr/IntegrationTestSuite/TestListIntegrations.cassette.gz b/vcr/IntegrationTestSuite/TestListIntegrations.cassette.gz
diff --git a/vcr/QueryIntegrationSuite/TestAsyncQuery.cassette.gz b/vcr/QueryIntegrationSuite/TestAsyncQuery.cassette.gz
diff --git a/vcr/QueryIntegrationSuite/TestAsyncQueryWithClientTimeout.cassette.gz b/vcr/QueryIntegrationSuite/TestAsyncQueryWithClientTimeout.cassette.gz
diff --git a/vcr/QueryIntegrationSuite/TestCancelQuery.cassette.gz b/vcr/QueryIntegrationSuite/TestCancelQuery.cassette.gz
diff --git a/vcr/QueryIntegrationSuite/TestListQueries.cassette.gz b/vcr/QueryIntegrationSuite/TestListQueries.cassette.gz
diff --git a/vcr/QueryIntegrationSuite/TestQuery.cassette.gz b/vcr/QueryIntegrationSuite/TestQuery.cassette.gz
diff --git a/vcr/QueryIntegrationSuite/TestQueryWithTimeout.cassette.gz b/vcr/QueryIntegrationSuite/TestQueryWithTimeout.cassette.gz
diff --git a/vcr/QueryIntegrationSuite/TestValidateQuery.cassette.gz b/vcr/QueryIntegrationSuite/TestValidateQuery.cassette.gz
diff --git a/vcr/TestAliasIntegrationSuite.cassette.gz b/vcr/TestAliasIntegrationSuite.cassette.gz
diff --git a/vcr/TestAzureIntegrations.cassette.gz b/vcr/TestAzureIntegrations.cassette.gz
diff --git a/vcr/TestCollectionIntegrationSuite.cassette.gz b/vcr/TestCollectionIntegrationSuite.cassette.gz
diff --git a/vcr/TestDocumentSuite.cassette.gz b/vcr/TestDocumentSuite.cassette.gz
diff --git a/vcr/TestError_IsNotFoundError.cassette.gz b/vcr/TestError_IsNotFoundError.cassette.gz
diff --git a/vcr/TestListViews.cassette.gz b/vcr/TestListViews.cassette.gz
diff --git a/vcr/TestRockClient_CreateQueryLambda.cassette.gz b/vcr/TestRockClient_CreateQueryLambda.cassette.gz
diff --git a/vcr/TestRockClient_GetOrganization.cassette.gz b/vcr/TestRockClient_GetOrganization.cassette.gz
diff --git a/vcr/TestRockClient_GetQueryLambdaVersion.cassette.gz b/vcr/TestRockClient_GetQueryLambdaVersion.cassette.gz
diff --git a/vcr/TestRockClient_GetQueryLambdaVersionByTag.cassette.gz b/vcr/TestRockClient_GetQueryLambdaVersionByTag.cassette.gz
diff --git a/vcr/TestRockClient_ListQueryLambdaTags.cassette.gz b/vcr/TestRockClient_ListQueryLambdaTags.cassette.gz
diff --git a/vcr/TestRockClient_ListQueryLambdaVersions.cassette.gz b/vcr/TestRockClient_ListQueryLambdaVersions.cassette.gz
diff --git a/vcr/TestRockClient_ListQueryLambdas.cassette.gz b/vcr/TestRockClient_ListQueryLambdas.cassette.gz
diff --git a/vcr/TestRockClient_ListQueryLambdas_workspace.cassette.gz b/vcr/TestRockClient_ListQueryLambdas_workspace.cassette.gz
diff --git a/vcr/TestRockClient_Ping.cassette.gz b/vcr/TestRockClient_Ping.cassette.gz
diff --git a/vcr/TestRockClient_withAPIServerEnv.cassette.gz b/vcr/TestRockClient_withAPIServerEnv.cassette.gz
diff --git a/vcr/TestRoleIntegration.cassette.gz b/vcr/TestRoleIntegration.cassette.gz
diff --git a/vcr/TestSuiteAPIKey.cassette.gz b/vcr/TestSuiteAPIKey.cassette.gz
diff --git a/vcr/TestSuiteWorkspace.cassette.gz b/vcr/TestSuiteWorkspace.cassette.gz
diff --git a/vcr/TestTemplate.cassette.gz b/vcr/TestTemplate.cassette.gz
diff --git a/vcr/TestUserIntegration.cassette.gz b/vcr/TestUserIntegration.cassette.gz
diff --git a/vcr/TestViewCRUD.cassette.gz b/vcr/TestViewCRUD.cassette.gz
diff --git a/vcr/TestVirtualInstance.cassette.gz b/vcr/TestVirtualInstance.cassette.gz
diff --git a/vcr/TestVirtualInstanceIntegration.cassette.gz b/vcr/TestVirtualInstanceIntegration.cassette.gz
diff --git a/vcr/example_s3.cassette.gz b/vcr/example_s3.cassette.gz