45 lines
1.1 KiB
Go
45 lines
1.1 KiB
Go
|
package cache
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestMemoryCacheMultiAndDeleteMany(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
c := newMemoryCache()
|
||
|
|
|
||
|
|
if err := c.SetMulti(ctx, []Item{
|
||
|
|
{Key: "a", Value: []byte("one"), TTL: time.Minute},
|
||
|
|
{Key: "b", Value: []byte("two"), TTL: time.Minute},
|
||
|
|
}); err != nil {
|
||
|
|
t.Fatalf("SetMulti() error = %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
values, err := c.GetMulti(ctx, []string{"a", "missing", "b"})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("GetMulti() error = %v", err)
|
||
|
|
}
|
||
|
|
if got := string(values["a"]); got != "one" {
|
||
|
|
t.Fatalf("values[a] = %q, want one", got)
|
||
|
|
}
|
||
|
|
if got := string(values["b"]); got != "two" {
|
||
|
|
t.Fatalf("values[b] = %q, want two", got)
|
||
|
|
}
|
||
|
|
if _, ok := values["missing"]; ok {
|
||
|
|
t.Fatalf("missing key should not be returned")
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := c.DeleteMany(ctx, []string{"a", "b"}); err != nil {
|
||
|
|
t.Fatalf("DeleteMany() error = %v", err)
|
||
|
|
}
|
||
|
|
values, err = c.GetMulti(ctx, []string{"a", "b"})
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("GetMulti() after DeleteMany error = %v", err)
|
||
|
|
}
|
||
|
|
if len(values) != 0 {
|
||
|
|
t.Fatalf("values after DeleteMany = %v, want empty", values)
|
||
|
|
}
|
||
|
|
}
|