test fixed

This commit is contained in:
Scott Osteen
2024-08-15 16:24:27 -05:00
parent a7f0c07e5d
commit 791e73d0bc

View File

@@ -2770,51 +2770,9 @@ func Test_Generic(t *testing.T) {
}()
client := resp.NewConn(conn)
tests := []struct {
name string
key string
value string
}{
{
name: "1. String",
key: "RandKey1",
value: "value1",
},
{
name: "2. Integer",
key: "RandKey2",
value: "10",
},
{
name: "3. Float",
key: "RandKey3",
value: "3.142",
},
}
// Map to check RANDOMKEY output
var expected map[string]string
// Preset the values
for _, test := range tests {
t.Run(test.name, func(t *testing.T) {
func(key, value string, expected map[string]string) {
err = client.WriteArray([]resp.Value{resp.StringValue("SET"), resp.StringValue(key), resp.StringValue(value)})
if err != nil {
t.Error(err)
}
res, _, err := client.ReadValue()
if err != nil {
t.Error(err)
}
if !strings.EqualFold(res.String(), "ok") {
t.Errorf("expected preset response to be \"OK\", got %s", res.String())
}
expected[key] = ""
}(test.key, test.value, expected)
})
}
// Reuse existing keys to test RANDOMKEY
// All keys used for other tests contain the substring 'Key'
expected := "Key"
// Test RANDOMKEY
if err = client.WriteArray([]resp.Value{resp.StringValue("RANDOMKEY")}); err != nil {
t.Error(err)
@@ -2825,8 +2783,8 @@ func Test_Generic(t *testing.T) {
t.Error(err)
}
if _, ok := expected[res.String()]; !ok {
t.Errorf("expected one of %v, got %s", expected, res.String())
if !strings.Contains(res.String(), expected) {
t.Errorf("expected a key containing substring '%s', got %s", expected, res.String())
}
})