mirror of
https://github.com/featurebasedb/featurebase.git
synced 2026-10-09 04:17:51 +00:00
moved to internal/test
This commit is contained in:
Todd Gruben
parent
b36b3b16ff
commit
aa21890503
2 changed files with 191 additions and 185 deletions
190
internal/test/querygenerator.go
Normal file
190
internal/test/querygenerator.go
Normal file
|
|
@ -0,0 +1,190 @@
|
|||
package test
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"github.com/pilosa/pilosa/pql"
|
||||
)
|
||||
|
||||
type Args map[string]interface{}
|
||||
|
||||
type Calls []*pql.Call
|
||||
|
||||
func PQL(calls ...*pql.Call) *pql.Query {
|
||||
return &pql.Query{Calls: calls}
|
||||
}
|
||||
|
||||
func Row(frame string, row int) *pql.Call {
|
||||
return &pql.Call{
|
||||
Name: "Row",
|
||||
Args: Args{
|
||||
"frame": frame,
|
||||
"row": row,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func mutationArgs(args ...interface{}) Args {
|
||||
rargs := make(Args)
|
||||
for _, arg := range args {
|
||||
switch v := arg.(type) {
|
||||
case int:
|
||||
rargs["column"] = v
|
||||
case string:
|
||||
if strings.Contains(v, "=") {
|
||||
parts := strings.Split(v, "=")
|
||||
rargs["frame"] = parts[0]
|
||||
i, _ := strconv.ParseInt(parts[1], 10, 64)
|
||||
rargs["value"] = i
|
||||
} else {
|
||||
rargs["timestamp"] = v
|
||||
}
|
||||
default:
|
||||
fmt.Printf("wat %T!\n", v)
|
||||
}
|
||||
}
|
||||
|
||||
return rargs
|
||||
}
|
||||
|
||||
func Set(args ...interface{}) *pql.Call {
|
||||
return &pql.Call{Name: "Set", Args: mutationArgs(args...)}
|
||||
}
|
||||
|
||||
func Clear(args ...interface{}) *pql.Call {
|
||||
return &pql.Call{Name: "Clear", Args: mutationArgs(args...)}
|
||||
}
|
||||
|
||||
func magic(args ...interface{}) (Args, Calls) {
|
||||
var (
|
||||
rargs Args
|
||||
calls Calls
|
||||
)
|
||||
|
||||
for _, arg := range args {
|
||||
switch v := arg.(type) {
|
||||
case Args:
|
||||
rargs = v
|
||||
case []*pql.Call:
|
||||
calls = append(calls, v...)
|
||||
default:
|
||||
fmt.Printf("wat %T!\n", v)
|
||||
}
|
||||
}
|
||||
|
||||
return rargs, calls
|
||||
}
|
||||
func Count(args ...*pql.Call) *pql.Call {
|
||||
kvargs, children := magic(args)
|
||||
return &pql.Call{Name: "Count", Args: kvargs, Children: children}
|
||||
}
|
||||
|
||||
func Union(args ...*pql.Call) *pql.Call {
|
||||
kvargs, children := magic(args)
|
||||
return &pql.Call{Name: "Union", Args: kvargs, Children: children}
|
||||
}
|
||||
|
||||
func Intersect(args ...*pql.Call) *pql.Call {
|
||||
kvargs, children := magic(args)
|
||||
return &pql.Call{Name: "Intersect", Args: kvargs, Children: children}
|
||||
}
|
||||
|
||||
func Difference(args ...*pql.Call) *pql.Call {
|
||||
kvargs, children := magic(args)
|
||||
return &pql.Call{Name: "Difference", Args: kvargs, Children: children}
|
||||
}
|
||||
func Xor(args ...*pql.Call) *pql.Call {
|
||||
kvargs, children := magic(args)
|
||||
return &pql.Call{Name: "Xor", Args: kvargs, Children: children}
|
||||
}
|
||||
|
||||
func Between(frame string, min, max int) *pql.Call {
|
||||
return &pql.Call{
|
||||
Name: "Range",
|
||||
Args: Args{
|
||||
"Op": pql.BETWEEN,
|
||||
"Value": []int{min, max},
|
||||
},
|
||||
}
|
||||
}
|
||||
func Lt(frame string, column int) *pql.Call {
|
||||
return &pql.Call{
|
||||
Name: "Range",
|
||||
Args: Args{
|
||||
"Op": pql.LT,
|
||||
"Value": column,
|
||||
},
|
||||
}
|
||||
}
|
||||
func Lte(frame string, column int) *pql.Call {
|
||||
return &pql.Call{
|
||||
Name: "Range",
|
||||
Args: Args{
|
||||
"Op": pql.LTE,
|
||||
"Value": column,
|
||||
},
|
||||
}
|
||||
}
|
||||
func Gt(frame string, column int) *pql.Call {
|
||||
return &pql.Call{
|
||||
Name: "Range",
|
||||
Args: Args{
|
||||
"Op": pql.GT,
|
||||
"Value": column,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func Gte(frame string, column int) *pql.Call {
|
||||
return &pql.Call{
|
||||
Name: "Range",
|
||||
Args: Args{
|
||||
"Op": pql.GTE,
|
||||
"Value": column,
|
||||
},
|
||||
}
|
||||
}
|
||||
func CompareCall(a, b *pql.Call) bool {
|
||||
if a.Name != b.Name {
|
||||
return false
|
||||
}
|
||||
for k, i := range a.Args {
|
||||
switch v := i.(type) {
|
||||
case []int:
|
||||
bside := b.Args[k]
|
||||
for j := range v {
|
||||
if v[j] != bside.([]int)[j] {
|
||||
return false
|
||||
}
|
||||
|
||||
}
|
||||
default:
|
||||
if b.Args[k] != i {
|
||||
return false
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if len(a.Children) == len(b.Children) {
|
||||
for i := range a.Children {
|
||||
if !CompareCall(a.Children[i], b.Children[i]) {
|
||||
return false
|
||||
}
|
||||
}
|
||||
} else {
|
||||
return false
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func Compare(a, b *pql.Query) bool {
|
||||
for i := range a.Calls {
|
||||
if !CompareCall(a.Calls[i], b.Calls[i]) {
|
||||
return false
|
||||
}
|
||||
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
|
@ -1,194 +1,10 @@
|
|||
package pilosa_test
|
||||
package test
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strconv"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/pilosa/pilosa/pql"
|
||||
)
|
||||
|
||||
type Args map[string]interface{}
|
||||
|
||||
type Calls []*pql.Call
|
||||
|
||||
func PQL(calls ...*pql.Call) *pql.Query {
|
||||
return &pql.Query{Calls: calls}
|
||||
}
|
||||
|
||||
func Row(frame string, row int) *pql.Call {
|
||||
return &pql.Call{
|
||||
Name: "Row",
|
||||
Args: Args{
|
||||
"frame": frame,
|
||||
"row": row,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func mutationArgs(args ...interface{}) Args {
|
||||
rargs := make(Args)
|
||||
for _, arg := range args {
|
||||
switch v := arg.(type) {
|
||||
case int:
|
||||
rargs["column"] = v
|
||||
case string:
|
||||
if strings.Contains(v, "=") {
|
||||
parts := strings.Split(v, "=")
|
||||
rargs["frame"] = parts[0]
|
||||
i, _ := strconv.ParseInt(parts[1], 10, 64)
|
||||
rargs["value"] = i
|
||||
} else {
|
||||
rargs["timestamp"] = v
|
||||
}
|
||||
default:
|
||||
fmt.Printf("wat %T!\n", v)
|
||||
}
|
||||
}
|
||||
|
||||
return rargs
|
||||
}
|
||||
|
||||
func Set(args ...interface{}) *pql.Call {
|
||||
return &pql.Call{Name: "Set", Args: mutationArgs(args...)}
|
||||
}
|
||||
|
||||
func Clear(args ...interface{}) *pql.Call {
|
||||
return &pql.Call{Name: "Clear", Args: mutationArgs(args...)}
|
||||
}
|
||||
|
||||
func magic(args ...interface{}) (Args, Calls) {
|
||||
var (
|
||||
rargs Args
|
||||
calls Calls
|
||||
)
|
||||
|
||||
for _, arg := range args {
|
||||
switch v := arg.(type) {
|
||||
case Args:
|
||||
rargs = v
|
||||
case []*pql.Call:
|
||||
calls = append(calls, v...)
|
||||
default:
|
||||
fmt.Printf("wat %T!\n", v)
|
||||
}
|
||||
}
|
||||
|
||||
return rargs, calls
|
||||
}
|
||||
func Count(args ...*pql.Call) *pql.Call {
|
||||
kvargs, children := magic(args)
|
||||
return &pql.Call{Name: "Count", Args: kvargs, Children: children}
|
||||
}
|
||||
|
||||
func Union(args ...*pql.Call) *pql.Call {
|
||||
kvargs, children := magic(args)
|
||||
return &pql.Call{Name: "Union", Args: kvargs, Children: children}
|
||||
}
|
||||
|
||||
func Intersect(args ...*pql.Call) *pql.Call {
|
||||
kvargs, children := magic(args)
|
||||
return &pql.Call{Name: "Intersect", Args: kvargs, Children: children}
|
||||
}
|
||||
|
||||
func Difference(args ...*pql.Call) *pql.Call {
|
||||
kvargs, children := magic(args)
|
||||
return &pql.Call{Name: "Difference", Args: kvargs, Children: children}
|
||||
}
|
||||
func Xor(args ...*pql.Call) *pql.Call {
|
||||
kvargs, children := magic(args)
|
||||
return &pql.Call{Name: "Xor", Args: kvargs, Children: children}
|
||||
}
|
||||
|
||||
func Between(frame string, min, max int) *pql.Call {
|
||||
return &pql.Call{
|
||||
Name: "Range",
|
||||
Args: Args{
|
||||
"Op": pql.BETWEEN,
|
||||
"Value": []int{min, max},
|
||||
},
|
||||
}
|
||||
}
|
||||
func Lt(frame string, column int) *pql.Call {
|
||||
return &pql.Call{
|
||||
Name: "Range",
|
||||
Args: Args{
|
||||
"Op": pql.LT,
|
||||
"Value": column,
|
||||
},
|
||||
}
|
||||
}
|
||||
func Lte(frame string, column int) *pql.Call {
|
||||
return &pql.Call{
|
||||
Name: "Range",
|
||||
Args: Args{
|
||||
"Op": pql.LTE,
|
||||
"Value": column,
|
||||
},
|
||||
}
|
||||
}
|
||||
func Gt(frame string, column int) *pql.Call {
|
||||
return &pql.Call{
|
||||
Name: "Range",
|
||||
Args: Args{
|
||||
"Op": pql.GT,
|
||||
"Value": column,
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func Gte(frame string, column int) *pql.Call {
|
||||
return &pql.Call{
|
||||
Name: "Range",
|
||||
Args: Args{
|
||||
"Op": pql.GTE,
|
||||
"Value": column,
|
||||
},
|
||||
}
|
||||
}
|
||||
func CompareCall(a, b *pql.Call) bool {
|
||||
if a.Name != b.Name {
|
||||
return false
|
||||
}
|
||||
for k, i := range a.Args {
|
||||
switch v := i.(type) {
|
||||
case []int:
|
||||
bside := b.Args[k]
|
||||
for j := range v {
|
||||
if v[j] != bside.([]int)[j] {
|
||||
return false
|
||||
}
|
||||
|
||||
}
|
||||
default:
|
||||
if b.Args[k] != i {
|
||||
return false
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
if len(a.Children) == len(b.Children) {
|
||||
for i := range a.Children {
|
||||
if !CompareCall(a.Children[i], b.Children[i]) {
|
||||
return false
|
||||
}
|
||||
}
|
||||
} else {
|
||||
return false
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func Compare(a, b *pql.Query) bool {
|
||||
for i := range a.Calls {
|
||||
if !CompareCall(a.Calls[i], b.Calls[i]) {
|
||||
return false
|
||||
}
|
||||
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
||||
func TestPQL_Generator(t *testing.T) {
|
||||
t.Run("pql.Query generator", func(t *testing.T) {
|
||||
Loading…
Add table
Reference in a new issue