You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
kratos/tool/kratos-gen-bts/testdata/none_test.go

50 lines
1.4 KiB

package testdata
import (
"context"
"errors"
"testing"
)
func TestNoneCache(t *testing.T) {
d := New()
meta := &Demo{ID: 1}
getFromCache := func(c context.Context) (*Demo, error) { return meta, nil }
notGetFromCache := func(c context.Context) (*Demo, error) { return nil, errors.New("err") }
getFromSource := func(c context.Context) (*Demo, error) { return meta, nil }
notGetFromSource := func(c context.Context) (*Demo, error) { return meta, errors.New("err") }
addToCache := func(c context.Context, values *Demo) error { return nil }
// get from cache
_noneCacheFunc = getFromCache
_noneRawFunc = notGetFromSource
_noneAddCacheFunc = addToCache
res, err := d.None(context.TODO())
if err != nil {
t.Fatalf("err should be nil, get: %v", err)
}
if res.ID != 1 {
t.Fatalf("id should be 1")
}
// get from source
_noneCacheFunc = notGetFromCache
_noneRawFunc = getFromSource
res, err = d.None(context.TODO())
if err != nil {
t.Fatalf("err should be nil, get: %v", err)
}
if res.ID != 1 {
t.Fatalf("id should be 1")
}
// with null cache
nullCache := &Demo{ID: -1}
getNullFromCache := func(c context.Context) (*Demo, error) { return nullCache, nil }
_noneCacheFunc = getNullFromCache
_noneRawFunc = notGetFromSource
res, err = d.None(context.TODO())
if err != nil {
t.Fatalf("err should be nil, get: %v", err)
}
if res != nil {
t.Fatalf("res should be nil")
}
}