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.
50 lines
1.5 KiB
50 lines
1.5 KiB
package testdata |
|
|
|
import ( |
|
"context" |
|
"errors" |
|
"testing" |
|
) |
|
|
|
func TestSingleCache(t *testing.T) { |
|
d := New() |
|
meta := &Article{ID: 1} |
|
getFromCache := func(c context.Context, id int64) (*Article, error) { return meta, nil } |
|
notGetFromCache := func(c context.Context, id int64) (*Article, error) { return nil, errors.New("err") } |
|
getFromSource := func(c context.Context, id int64) (*Article, error) { return meta, nil } |
|
notGetFromSource := func(c context.Context, id int64) (*Article, error) { return meta, errors.New("err") } |
|
addToCache := func(c context.Context, id int64, values *Article) error { return nil } |
|
// get from cache |
|
_singleCacheFunc = getFromCache |
|
_singleRawFunc = notGetFromSource |
|
_singleAddCacheFunc = addToCache |
|
res, err := d.Article(context.TODO(), 1) |
|
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 |
|
_singleCacheFunc = notGetFromCache |
|
_singleRawFunc = getFromSource |
|
res, err = d.Article(context.TODO(), 1) |
|
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 := &Article{ID: -1} |
|
getNullFromCache := func(c context.Context, id int64) (*Article, error) { return nullCache, nil } |
|
_singleCacheFunc = getNullFromCache |
|
_singleRawFunc = notGetFromSource |
|
res, err = d.Article(context.TODO(), 1) |
|
if err != nil { |
|
t.Fatalf("err should be nil, get: %v", err) |
|
} |
|
if res != nil { |
|
t.Fatalf("res should be nil") |
|
} |
|
}
|
|
|