mirror of
https://git.mirrors.martin98.com/https://github.com/SigNoz/signoz
synced 2025-08-05 09:30:43 +08:00
d1e7cc128f
* fix: inital changes for telemetry store * fix: add tests and use proper config for conn * fix: add telemetry store test * fix: add backward compatibility for old variables and update example conf * fix: move wrapper to telemetry store * fix: no need to pass query for settings * fix: remove redundant config for ch conn * fix: use clickhouse dsn instead * fix: update example config * fix: update backward compatibility code * fix: use hooks in telemetrystore * fix: address minor comments --------- Co-authored-by: Vibhu Pandey <vibhupandey28@gmail.com>
35 lines
817 B
Go
35 lines
817 B
Go
package telemetrystoretest
|
|
|
|
import (
|
|
"github.com/ClickHouse/clickhouse-go/v2"
|
|
cmock "github.com/srikanthccv/ClickHouse-go-mock"
|
|
)
|
|
|
|
// Provider represents a mock telemetry store provider for testing
|
|
type Provider struct {
|
|
mock cmock.ClickConnMockCommon
|
|
}
|
|
|
|
// New creates a new mock telemetry store provider
|
|
func New() (*Provider, error) {
|
|
options := &clickhouse.Options{} // Default options
|
|
mock, err := cmock.NewClickHouseNative(options)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Provider{
|
|
mock: mock,
|
|
}, nil
|
|
}
|
|
|
|
// Clickhouse returns the mock Clickhouse connection
|
|
func (p *Provider) Clickhouse() clickhouse.Conn {
|
|
return p.mock.(clickhouse.Conn)
|
|
}
|
|
|
|
// Mock returns the underlying Clickhouse mock instance for setting expectations
|
|
func (p *Provider) Mock() cmock.ClickConnMockCommon {
|
|
return p.mock
|
|
}
|