You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
209 lines
8.7 KiB
Go
209 lines
8.7 KiB
Go
package legacy
|
|
|
|
import (
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/rqlite/rqlite/command"
|
|
)
|
|
|
|
func Test_SimpleExecute(t *testing.T) {
|
|
// "CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT, age INTEGER)" timings=true
|
|
b := []byte{123, 34, 115, 117, 98, 34, 58, 123, 34, 113, 117, 101, 114, 105, 101, 115, 34, 58, 91, 34, 67, 82, 69, 65, 84, 69, 32, 84, 65, 66, 76, 69, 32, 102, 111, 111, 32, 40, 105, 100, 32, 73, 78, 84, 69, 71, 69, 82, 32, 78, 79, 84, 32, 78, 85, 76, 76, 32, 80, 82, 73, 77, 65, 82, 89, 32, 75, 69, 89, 44, 32, 110, 97, 109, 101, 32, 84, 69, 88, 84, 44, 32, 97, 103, 101, 32, 73, 78, 84, 69, 71, 69, 82, 41, 34, 93, 44, 34, 80, 97, 114, 97, 109, 101, 116, 101, 114, 115, 34, 58, 91, 110, 117, 108, 108, 93, 44, 34, 116, 105, 109, 105, 110, 103, 115, 34, 58, 116, 114, 117, 101, 125, 125}
|
|
|
|
var c command.Command
|
|
var er command.ExecuteRequest
|
|
if err := Unmarshal(b, &c); err != nil {
|
|
t.Fatalf("failed to Unmarshal: %s", err)
|
|
}
|
|
|
|
if c.Type != command.Command_COMMAND_TYPE_EXECUTE {
|
|
t.Fatalf("incorrect command type: %s", c.Type)
|
|
}
|
|
if err := command.UnmarshalSubCommand(&c, &er); err != nil {
|
|
t.Fatalf("failed to Unmarshal subcommand: %s", err)
|
|
}
|
|
if !er.Timings {
|
|
t.Fatalf("timings not set")
|
|
}
|
|
if er.Request.Transaction {
|
|
t.Fatalf("transaction set")
|
|
}
|
|
if n := len(er.Request.Statements); n != 1 {
|
|
t.Fatalf("incorrect number of statments: %d", n)
|
|
}
|
|
if s := er.Request.Statements[0].Sql; s != "CREATE TABLE foo (id INTEGER NOT NULL PRIMARY KEY, name TEXT, age INTEGER)" {
|
|
t.Fatalf("incorrect SQL: %s", s)
|
|
}
|
|
}
|
|
|
|
func Test_SimpleQuery(t *testing.T) {
|
|
// "SELECT * FROM foo", timings=true
|
|
b := []byte{123, 34, 116, 121, 112, 34, 58, 49, 44, 34, 115, 117, 98, 34, 58, 123, 34, 113, 117, 101, 114, 105, 101, 115, 34, 58, 91, 34, 83, 69, 76, 69, 67, 84, 32, 42, 32, 70, 82, 79, 77, 32, 102, 111, 111, 34, 93, 44, 34, 80, 97, 114, 97, 109, 101, 116, 101, 114, 115, 34, 58, 91, 110, 117, 108, 108, 93, 44, 34, 116, 105, 109, 105, 110, 103, 115, 34, 58, 116, 114, 117, 101, 125, 125}
|
|
|
|
var c command.Command
|
|
var qr command.QueryRequest
|
|
if err := Unmarshal(b, &c); err != nil {
|
|
t.Fatalf("failed to Unmarshal: %s", err)
|
|
}
|
|
|
|
if c.Type != command.Command_COMMAND_TYPE_QUERY {
|
|
t.Fatalf("incorrect command type: %s", c.Type)
|
|
}
|
|
if err := command.UnmarshalSubCommand(&c, &qr); err != nil {
|
|
t.Fatalf("failed to Unmarshal subcommand: %s", err)
|
|
}
|
|
if !qr.Timings {
|
|
t.Fatalf("timings not set")
|
|
}
|
|
if qr.Request.Transaction {
|
|
t.Fatalf("transaction set")
|
|
}
|
|
if n := len(qr.Request.Statements); n != 1 {
|
|
t.Fatalf("incorrect number of statments: %d", n)
|
|
}
|
|
if s := qr.Request.Statements[0].Sql; s != "SELECT * FROM foo" {
|
|
t.Fatalf("incorrect SQL: %s", s)
|
|
}
|
|
}
|
|
|
|
func Test_SingleParameterized(t *testing.T) {
|
|
// ["INSERT INTO foo(name, age) VALUES(?, ?)", "fiona", 20], timings=true
|
|
b := []byte{123, 34, 115, 117, 98, 34, 58, 123, 34, 113, 117, 101, 114, 105, 101, 115, 34, 58, 91, 34, 73, 78, 83, 69, 82, 84, 32, 73, 78, 84, 79, 32, 102, 111, 111, 40, 110, 97, 109, 101, 44, 32, 97, 103, 101, 41, 32, 86, 65, 76, 85, 69, 83, 40, 63, 44, 32, 63, 41, 34, 93, 44, 34, 80, 97, 114, 97, 109, 101, 116, 101, 114, 115, 34, 58, 91, 91, 34, 102, 105, 111, 110, 97, 34, 44, 50, 48, 93, 93, 44, 34, 116, 105, 109, 105, 110, 103, 115, 34, 58, 116, 114, 117, 101, 125, 125}
|
|
|
|
var c command.Command
|
|
var er command.ExecuteRequest
|
|
if err := Unmarshal(b, &c); err != nil {
|
|
t.Fatalf("failed to Unmarshal: %s", err)
|
|
}
|
|
|
|
if c.Type != command.Command_COMMAND_TYPE_EXECUTE {
|
|
t.Fatalf("incorrect command type: %s", c.Type)
|
|
}
|
|
if err := command.UnmarshalSubCommand(&c, &er); err != nil {
|
|
t.Fatalf("failed to Unmarshal subcommand: %s", err)
|
|
}
|
|
if !er.Timings {
|
|
t.Fatalf("timings not set")
|
|
}
|
|
if er.Request.Transaction {
|
|
t.Fatalf("transaction set")
|
|
}
|
|
if n := len(er.Request.Statements); n != 1 {
|
|
t.Fatalf("incorrect number of statments: %d", n)
|
|
}
|
|
if s := er.Request.Statements[0].Sql; s != "INSERT INTO foo(name, age) VALUES(?, ?)" {
|
|
t.Fatalf("incorrect SQL: %s", s)
|
|
}
|
|
if l := len(er.Request.Statements[0].Parameters); l != 2 {
|
|
t.Fatalf("incorrect number of parameters: %d", l)
|
|
}
|
|
if v := er.Request.Statements[0].Parameters[0].GetS(); v != "fiona" {
|
|
t.Fatalf("incorrect value for 1st parameter: %s", v)
|
|
}
|
|
if v := er.Request.Statements[0].Parameters[1].GetD(); v != 20 {
|
|
t.Fatalf("incorrect value for 2nd parameter: %f", v)
|
|
}
|
|
}
|
|
|
|
func Test_MultipleParameterized(t *testing.T) {
|
|
// ["INSERT INTO foo(name, age) VALUES(?, ?)", "fiona", 20], ["INSERT INTO bar(name, age, address) VALUES(?, ?, ?)", "declan", 5, "galway"], timings=true
|
|
b := []byte{123, 34, 115, 117, 98, 34, 58, 123, 34, 113, 117, 101, 114, 105, 101, 115, 34, 58, 91, 34, 73, 78, 83, 69, 82, 84, 32, 73, 78, 84, 79, 32, 102, 111, 111, 40, 110, 97, 109, 101, 44, 32, 97, 103, 101, 41, 32, 86, 65, 76, 85, 69, 83, 40, 63, 44, 32, 63, 41, 34, 44, 34, 73, 78, 83, 69, 82, 84, 32, 73, 78, 84, 79, 32, 98, 97, 114, 40, 110, 97, 109, 101, 44, 32, 97, 103, 101, 44, 32, 97, 100, 100, 114, 101, 115, 115, 41, 32, 86, 65, 76, 85, 69, 83, 40, 63, 44, 32, 63, 44, 32, 63, 41, 34, 93, 44, 34, 80, 97, 114, 97, 109, 101, 116, 101, 114, 115, 34, 58, 91, 91, 34, 102, 105, 111, 110, 97, 34, 44, 50, 48, 93, 44, 91, 34, 100, 101, 99, 108, 97, 110, 34, 44, 53, 44, 34, 103, 97, 108, 119, 97, 121, 34, 93, 93, 44, 34, 116, 105, 109, 105, 110, 103, 115, 34, 58, 116, 114, 117, 101, 125, 125}
|
|
|
|
var c command.Command
|
|
var er command.ExecuteRequest
|
|
if err := Unmarshal(b, &c); err != nil {
|
|
t.Fatalf("failed to Unmarshal: %s", err)
|
|
}
|
|
|
|
if c.Type != command.Command_COMMAND_TYPE_EXECUTE {
|
|
t.Fatalf("incorrect command type: %s", c.Type)
|
|
}
|
|
if err := command.UnmarshalSubCommand(&c, &er); err != nil {
|
|
t.Fatalf("failed to Unmarshal subcommand: %s", err)
|
|
}
|
|
if !er.Timings {
|
|
t.Fatalf("timings not set")
|
|
}
|
|
if er.Request.Transaction {
|
|
t.Fatalf("transaction set")
|
|
}
|
|
if n := len(er.Request.Statements); n != 2 {
|
|
t.Fatalf("incorrect number of statments: %d", n)
|
|
}
|
|
|
|
if s := er.Request.Statements[0].Sql; s != "INSERT INTO foo(name, age) VALUES(?, ?)" {
|
|
t.Fatalf("incorrect SQL: %s", s)
|
|
}
|
|
if l := len(er.Request.Statements[0].Parameters); l != 2 {
|
|
t.Fatalf("incorrect number of parameters: %d", l)
|
|
}
|
|
if v := er.Request.Statements[0].Parameters[0].GetS(); v != "fiona" {
|
|
t.Fatalf("incorrect value for 1st parameter: %s", v)
|
|
}
|
|
if v := er.Request.Statements[0].Parameters[1].GetD(); v != 20 {
|
|
t.Fatalf("incorrect value for 2nd parameter: %f", v)
|
|
}
|
|
|
|
if s := er.Request.Statements[1].Sql; s != "INSERT INTO bar(name, age, address) VALUES(?, ?, ?)" {
|
|
t.Fatalf("incorrect SQL: %s", s)
|
|
}
|
|
if l := len(er.Request.Statements[1].Parameters); l != 3 {
|
|
t.Fatalf("incorrect number of parameters: %d", l)
|
|
}
|
|
if v := er.Request.Statements[1].Parameters[0].GetS(); v != "declan" {
|
|
t.Fatalf("incorrect value for 1st parameter: %s", v)
|
|
}
|
|
if v := er.Request.Statements[1].Parameters[1].GetD(); v != 5 {
|
|
t.Fatalf("incorrect value for 2nd parameter: %f", v)
|
|
}
|
|
if v := er.Request.Statements[1].Parameters[2].GetS(); v != "galway" {
|
|
t.Fatalf("incorrect value for 1st parameter: %s", v)
|
|
}
|
|
}
|
|
|
|
func Test_MetadataSet(t *testing.T) {
|
|
b := []byte{123, 34, 116, 121, 112, 34, 58, 50, 44, 34, 115, 117, 98, 34, 58, 123, 34, 114, 97, 102, 116, 95, 105, 100, 34, 58, 34, 108, 111, 99, 97, 108, 104, 111, 115, 116, 58, 52, 48, 48, 50, 34, 44, 34, 100, 97, 116, 97, 34, 58, 123, 34, 97, 112, 105, 95, 97, 100, 100, 114, 34, 58, 34, 108, 111, 99, 97, 108, 104, 111, 115, 116, 58, 52, 48, 48, 49, 34, 44, 34, 97, 112, 105, 95, 112, 114, 111, 116, 111, 34, 58, 34, 104, 116, 116, 112, 34, 125, 125, 125}
|
|
|
|
var c command.Command
|
|
var ms command.MetadataSet
|
|
|
|
if err := Unmarshal(b, &c); err != nil {
|
|
t.Fatalf("failed to Unmarshal: %s", err)
|
|
}
|
|
|
|
if c.Type != command.Command_COMMAND_TYPE_METADATA_SET {
|
|
t.Fatalf("incorrect command type: %s", c.Type)
|
|
}
|
|
if err := command.UnmarshalSubCommand(&c, &ms); err != nil {
|
|
t.Fatalf("failed to Unmarshal subcommand: %s", err)
|
|
}
|
|
if id := ms.RaftId; id != "localhost:4002" {
|
|
t.Fatalf("incorrect Raft ID: %s", id)
|
|
}
|
|
if !reflect.DeepEqual(ms.Data, map[string]string{"api_addr": "localhost:4001", "api_proto": "http"}) {
|
|
t.Fatalf("map is incorrect: %s", ms.Data)
|
|
}
|
|
}
|
|
|
|
func Test_MetadataDelete(t *testing.T) {
|
|
b := []byte{123, 34, 116, 121, 112, 34, 58, 51, 44, 34, 115, 117, 98, 34, 58, 34, 108, 111, 99, 97, 108, 104, 111, 115, 116, 58, 52, 48, 48, 52, 34, 125}
|
|
|
|
var c command.Command
|
|
var md command.MetadataDelete
|
|
|
|
if err := Unmarshal(b, &c); err != nil {
|
|
t.Fatalf("failed to Unmarshal: %s", err)
|
|
}
|
|
|
|
if c.Type != command.Command_COMMAND_TYPE_METADATA_DELETE {
|
|
t.Fatalf("incorrect command type: %s", c.Type)
|
|
}
|
|
if err := command.UnmarshalSubCommand(&c, &md); err != nil {
|
|
t.Fatalf("failed to Unmarshal subcommand: %s", err)
|
|
}
|
|
if id := md.RaftId; id != "localhost:4004" {
|
|
t.Fatalf("incorrect Raft ID: %s", id)
|
|
}
|
|
}
|