77 lines
2.3 KiB
Go
77 lines
2.3 KiB
Go
package influxdb
|
|
|
|
import (
|
|
"testing"
|
|
)
|
|
|
|
func TestBuildQueryWithParams(t *testing.T) {
|
|
tests := []struct {
|
|
name string
|
|
table string
|
|
filters map[string]interface{}
|
|
operators map[string]string
|
|
fields []string
|
|
expectedQuery string
|
|
}{
|
|
{
|
|
name: "Basic query with filters and fields",
|
|
table: "candles",
|
|
filters: map[string]interface{}{"s": "'AAPL'", "o": 150.0},
|
|
operators: map[string]string{"o": ">"},
|
|
fields: []string{"s", "o", "h", "l", "c", "v"},
|
|
expectedQuery: "SELECT s, o, h, l, c, v FROM candles WHERE s = 'AAPL' AND o > 150",
|
|
},
|
|
{
|
|
name: "Query with no filters",
|
|
table: "candles",
|
|
filters: map[string]interface{}{},
|
|
operators: map[string]string{},
|
|
fields: []string{"s", "o", "h"},
|
|
expectedQuery: "SELECT s, o, h FROM candles",
|
|
},
|
|
{
|
|
name: "Query with no fields",
|
|
table: "candles",
|
|
filters: map[string]interface{}{"s": "'AAPL'"},
|
|
operators: map[string]string{},
|
|
fields: []string{},
|
|
expectedQuery: "SELECT * FROM candles WHERE s = 'AAPL'",
|
|
},
|
|
{
|
|
name: "Empty table name",
|
|
table: "",
|
|
filters: map[string]interface{}{"s": "'AAPL'"},
|
|
operators: map[string]string{},
|
|
fields: []string{"s", "o"},
|
|
expectedQuery: "SELECT s, o FROM WHERE s = 'AAPL'",
|
|
},
|
|
{
|
|
name: "Special characters in filters",
|
|
table: "candles",
|
|
filters: map[string]interface{}{"name": "'O'Reilly'"},
|
|
operators: map[string]string{},
|
|
fields: []string{"name"},
|
|
expectedQuery: "SELECT name FROM candles WHERE name = 'O'Reilly'",
|
|
},
|
|
{
|
|
name: "Query with interval filters",
|
|
table: "candles",
|
|
filters: map[string]interface{}{"time": "now() - interval '15 minutes'"},
|
|
operators: map[string]string{"time": ">="},
|
|
fields: []string{"*"},
|
|
expectedQuery: "SELECT * FROM candles WHERE time >= now() - interval '15 minutes'",
|
|
},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
query := BuildQueryWithParams(tt.table, tt.filters, tt.operators, tt.fields)
|
|
|
|
if query != tt.expectedQuery {
|
|
t.Errorf("expected query %s, got %s", tt.expectedQuery, query)
|
|
}
|
|
//t.Log(query)
|
|
})
|
|
}
|
|
}
|