clickhouse

type ClientInterface interface {
	// QueryOne executes a query and returns a single result
	QueryOne(ctx context.Context, query string, dest interface{}) error
	// QueryMany executes a query and returns multiple results
	QueryMany(ctx context.Context, query string, dest interface{}) error
	// Execute runs a query without expecting results
	Execute(ctx context.Context, query string) error
	// BulkInsert performs a bulk insert operation
	BulkInsert(ctx context.Context, table string, data interface{}) error
	// IsStorageEmpty checks if a table has any records matching the given conditions
	IsStorageEmpty(ctx context.Context, table string, conditions map[string]interface{}) (bool, error)
	// Start initializes the client
	Start() error
	// Stop closes the client
	Stop() error
}

type Config struct {
	URL           string        `yaml:"url" validate:"required,url"`
	Cluster       string        `yaml:"cluster"`
	LocalSuffix   string        `yaml:"local_suffix"`
	QueryTimeout  time.Duration `yaml:"query_timeout"`
	InsertTimeout time.Duration `yaml:"insert_timeout"`
	Debug         bool          `yaml:"debug"`
	KeepAlive     time.Duration `yaml:"keep_alive"`
	AdminDatabase string        `yaml:"admin_database"`
	AdminTable    string        `yaml:"admin_table"`
}

type MockCall struct {
	Method string
	Args   []interface{}
}

type MockClient struct {
	// Function fields that can be set by tests
	QueryOneFunc   func(ctx context.Context, query string, dest interface{}) error
	QueryManyFunc  func(ctx context.Context, query string, dest interface{}) error
	ExecuteFunc    func(ctx context.Context, query string) error
	BulkInsertFunc func(ctx context.Context, table string, data interface{}) error
	StartFunc      func() error
	StopFunc       func() error

	// Track calls for assertions
	Calls []MockCall
}