command_getkeys_test.go

package tests

import (
	"testing"

	"gotest.tools/v3/assert"
)

var getKeysTestCases = []struct {
	name     string
	inCmd    string
	expected interface{}
}{
	{"Set command", "set 1 2 3 4", []interface{}{"1"}},
	{"Get command", "get key", []interface{}{"key"}},
	{"TTL command", "ttl key", []interface{}{"key"}},
	{"Del command", "del 1 2 3 4 5 6", []interface{}{"1", "2", "3", "4", "5", "6"}},
	{"MSET command", "MSET key1 val1 key2 val2", []interface{}{"key1", "key2"}},
	{"Expire command", "expire key time extra", []interface{}{"key"}},
	{"QINTINS command", "QINTINS k 1", []interface{}{"k"}},
	{"BFINIT command", "BFINIT bloom some parameters", []interface{}{"bloom"}},
	{"Ping command", "ping", "ERR the command has no key arguments"},
	{"Invalid Get command", "get", "ERR invalid number of arguments specified for command"},
	{"Abort command", "abort", "ERR the command has no key arguments"},
	{"Invalid command", "NotValidCommand", "ERR invalid command specified"},
}

func TestCommandGetKeys(t *testing.T) {
	conn := getLocalConnection()
	defer conn.Close()

	for _, tc := range getKeysTestCases {
		t.Run(tc.name, func(t *testing.T) {
			result := fireCommand(conn, "COMMAND GETKEYS "+tc.inCmd)
			assert.DeepEqual(t, tc.expected, result)
		})
	}
}

func BenchmarkGetKeysMatch(b *testing.B) {
	conn := getLocalConnection()
	defer conn.Close()

	b.ResetTimer()
	for i := 0; i < b.N; i++ {
		for _, tc := range getKeysTestCases {
			fireCommand(conn, "COMMAND GETKEYS "+tc.inCmd)
		}
	}
}