mirror of
https://github.com/EchoVault/SugarDB.git
synced 2025-10-11 19:00:05 +08:00
60 lines
1.6 KiB
Go
60 lines
1.6 KiB
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"strings"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/kelvinmwinuka/memstore/utils"
|
|
)
|
|
|
|
const (
|
|
OK = "+OK\r\n\n"
|
|
)
|
|
|
|
func TestHandleCommand(t *testing.T) {
|
|
server := utils.MockServer{
|
|
Data: utils.MockData{
|
|
Mu: sync.Mutex{},
|
|
Data: make(map[string]interface{}),
|
|
},
|
|
}
|
|
|
|
cw := utils.CustomWriter{}
|
|
writer := bufio.NewWriter(&cw)
|
|
|
|
tests := []struct {
|
|
cmd []string
|
|
expected string
|
|
}{
|
|
// SET test cases
|
|
{[]string{"set", "key1", "value1"}, OK},
|
|
{[]string{"set", "key2", "30"}, OK},
|
|
{[]string{"set", "key3", "3.142"}, OK},
|
|
{[]string{"set", "key4", "part1", "part2", "part3"}, "-Error wrong number of args for SET command\r\n\n"},
|
|
{[]string{"set"}, "-Error wrong number of args for SET command\r\n\n"},
|
|
|
|
// GET test cases
|
|
{[]string{"get", "key1"}, "+value1\r\n\n"},
|
|
{[]string{"get", "key2"}, ":30\r\n\n"},
|
|
{[]string{"get", "key3"}, "+3.142\r\n\n"},
|
|
{[]string{"get", "key4"}, "+nil\r\n\n"},
|
|
{[]string{"get"}, "-Error wrong number of args for GET command\r\n\n"},
|
|
{[]string{"get", "key1", "key2"}, "-Error wrong number of args for GET command\r\n\n"},
|
|
|
|
// MGET test cases
|
|
{[]string{"mget", "key1", "key2", "key3", "key4"}, "*4\r\n$6\r\nvalue1\r\n$2\r\n30\r\n$5\r\n3.142\r\n$3\r\nnil\r\n\n"},
|
|
{[]string{"mget", "key5", "key6"}, "*2\r\n$3\r\nnil\r\n$3\r\nnil\r\n\n"},
|
|
{[]string{"mget"}, "-Error wrong number of args for MGET command\r\n\n"},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
cw.Buf.Reset()
|
|
Plugin.HandleCommand(tt.cmd, &server, writer)
|
|
if tt.expected != cw.Buf.String() {
|
|
t.Errorf("Expected %s, Got %s", strings.TrimSpace(tt.expected), strings.TrimSpace(cw.Buf.String()))
|
|
}
|
|
}
|
|
}
|