DS-querier: define func to fetch instance config settings (#106172)

DS-querier: define func to fetch instance config settings to be passed into executing query
This commit is contained in:
Sarah Zinger
2025-05-29 11:07:40 -04:00
committed by GitHub
parent 69653ea3dc
commit 020780f5ac
4 changed files with 40 additions and 12 deletions
+6 -1
View File
@@ -3,6 +3,7 @@ package query
import (
"bytes"
"context"
"errors"
"fmt"
"net/http"
"net/http/httptest"
@@ -152,7 +153,7 @@ type mockClient struct {
lastCalledWithHeaders *map[string]string
}
func (m mockClient) GetDataSourceClient(ctx context.Context, ref data.DataSourceRef, headers map[string]string) (clientapi.QueryDataClient, error) {
func (m mockClient) GetDataSourceClient(ctx context.Context, ref data.DataSourceRef, headers map[string]string, instanceConfig clientapi.InstanceConfigurationSettings) (clientapi.QueryDataClient, error) {
*m.lastCalledWithHeaders = headers
return nil, fmt.Errorf("mock error")
@@ -169,3 +170,7 @@ func (m mockClient) CallResource(ctx context.Context, req *backend.CallResourceR
func (m mockClient) CheckHealth(ctx context.Context, req *backend.CheckHealthRequest) (*backend.CheckHealthResult, error) {
return nil, nil
}
func (m mockClient) GetInstanceConfigurationSettings(_ context.Context) (clientapi.InstanceConfigurationSettings, error) {
return clientapi.InstanceConfigurationSettings{}, errors.New("get instance configuration settings is not implemented")
}