fix: update allowed header prefix (#120)

client_base.go

@@ -236,7 +236,7 @@ func extractAdditionalHeaders(ctx context.Context) map[string]string {
 		// only take headers that start with Firebolt- prefix
 		filteredHeaders := make(map[string]string)
 		for key, value := range additionalHeaders {
-			if strings.HasPrefix(key, "X-Firebolt-") {
+			if strings.HasPrefix(key, "Firebolt-") {
 				filteredHeaders[key] = value
 			}
 		}

client_common_test.go

@@ -61,16 +61,16 @@ func testAdditionalHeaders(t *testing.T, clientFactory func(string) Client) {
 	// Test that additional headers, passed in ctx are respected
 
 	var additionalHeaders = map[string]string{
-		"X-Firebolt-Test-Header": "test",
-		"Ignored-Header":         "ignored",
+		"Firebolt-Test-Header": "test",
+		"Ignored-Header":       "ignored",
 	}
 	server := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
 		if r.URL.Path == ServiceAccountLoginURLSuffix {
 			_, _ = w.Write(getAuthResponse(10000))
 		} else if r.URL.Path == UsernamePasswordURLSuffix {
 			_, _ = w.Write(getAuthResponseV0(10000))
 		} else {
-			if r.Header.Get("X-Firebolt-Test-Header") != "test" {
+			if r.Header.Get("Firebolt-Test-Header") != "test" {
 				t.Errorf("Did not set Firebolt-Test-Header value when passed in ctx")
 			}
 			if r.Header.Get("Ignored-Header") != "" {

client_integration_test.go

@@ -42,7 +42,7 @@ func TestQuerySetStatements(t *testing.T) {
 		context.TODO(),
 		engineUrlMock,
 		query,
-		map[string]string{"time_zone": "America/New_York", "database": databaseMock},
+		map[string]string{"timezone": "America/New_York", "database": databaseMock},
 		connectionControl{},
 	)
 	if err != nil {