github.com/blong14/gache@v0.0.0-20240124023949-89416fd8bbfa/sql/sql_test.go

package sql

import (
	"strings"
	"testing"

	gdb "github.com/blong14/gache/internal/db"
)

func TestParse(t *testing.T) {
	tests := map[string]*gdb.Query{
		"select * from default;": {
			Header: gdb.QueryHeader{
				Inst:      gdb.GetRange,
				TableName: []byte("default"),
			},
		},
		"select * from default limit 10;": {
			Header: gdb.QueryHeader{
				Inst:      gdb.GetRange,
				TableName: []byte("default"),
			},
			KeyRange: gdb.KeyRange{Limit: 10},
		},

		"select * from default where key = __key__;": {
			Header: gdb.QueryHeader{
				Inst:      gdb.GetValue,
				TableName: []byte("default")},
			Key: []byte("__key__"),
		},
		"select * from default where key between aaa and ddd;": {
			Header: gdb.QueryHeader{
				Inst:      gdb.GetRange,
				TableName: []byte("default"),
			},
			KeyRange: gdb.KeyRange{Start: []byte("aaa"), End: []byte("ddd")},
		},
		"select count from default;": {
			Header: gdb.QueryHeader{
				Inst:      gdb.Count,
				TableName: []byte("default"),
			},
		},

		"insert into default set key = _key, value = _value;": {
			Header: gdb.QueryHeader{
				Inst:      gdb.SetValue,
				TableName: []byte("default"),
			},
			Key:   []byte("_key"),
			Value: []byte("_value"),
		},

		"copy default from ./persons.csv;": {
			Header: gdb.QueryHeader{
				Inst:      gdb.Load,
				TableName: []byte("default"),
				FileName:  []byte("./persons.csv"),
			},
		},

		"create table default;": {
			Header: gdb.QueryHeader{
				Inst:      gdb.AddTable,
				TableName: []byte("default"),
			},
		},
	}
	for test, expected := range tests {
		t.Run(test, func(t *testing.T) {
			reader := strings.NewReader(test)
			query, err := parse(reader)
			if err != nil {
				t.Error(err)
			}
			if query.String() != expected.String() {
				t.Errorf("e %s g %s", expected, query)
			}
			t.Log(query.String())
		})
	}
}