datasources: querier: refactor (#108776)

datasources: querier: refactor types
This commit is contained in:
Gábor Farkas
2025-08-04 09:57:41 +02:00
committed by GitHub
parent 81f544e186
commit 6dc0dedc97
6 changed files with 68 additions and 37 deletions
+14 -3
View File
@@ -250,7 +250,18 @@ type mockClient struct {
stubbedFrame *data.Frame
}
func (m mockClient) GetDataSourceClient(ctx context.Context, ref dataapi.DataSourceRef, headers map[string]string, instanceConfig clientapi.InstanceConfigurationSettings) (clientapi.QueryDataClient, error) {
func (m mockClient) GetInstance(ctx context.Context) (clientapi.Instance, error) {
mclient := mockClient{
stubbedFrame: m.stubbedFrame,
}
return mclient, nil
}
func (m mockClient) GetLogger(parent log.Logger) log.Logger {
return parent.New()
}
func (m mockClient) GetDataSourceClient(ctx context.Context, ref dataapi.DataSourceRef, headers map[string]string) (clientapi.QueryDataClient, error) {
mclient := mockClient{
stubbedFrame: m.stubbedFrame,
}
@@ -281,11 +292,11 @@ func (m mockClient) CheckHealth(ctx context.Context, req *backend.CheckHealthReq
return nil, nil
}
func (m mockClient) GetInstanceConfigurationSettings(ctx context.Context) (clientapi.InstanceConfigurationSettings, error) {
func (m mockClient) GetSettings() clientapi.InstanceConfigurationSettings {
return clientapi.InstanceConfigurationSettings{
ExpressionsEnabled: true,
FeatureToggles: featuremgmt.WithFeatures(featuremgmt.FlagSqlExpressions),
}, nil
}
}
type mockLegacyDataSourceLookup struct{}