// Test the retrieving with an error.
func TestRetrieveError(t *testing.T) {
assert := audit.NewTestingAssertion(t, true)
// Environment.
ctr := 0
efunc := func() (interface{}, error) {
ctr++
return nil, fmt.Errorf("ouch %d", ctr)
}
cv := cache.NewCachedValue(efunc, 25*time.Millisecond)
defer cv.Remove()
retrieve := func() error { _, err := cv.Value(); return err }
// Asserts.
assert.ErrorMatch(retrieve(), "ouch 1")
assert.ErrorMatch(retrieve(), "ouch 2")
assert.ErrorMatch(retrieve(), "ouch 3")
}
// Test the retrieving with a panic.
func TestRetrievePanic(t *testing.T) {
assert := audit.NewTestingAssertion(t, true)
// Environment.
ctr := 0
pfunc := func() (interface{}, error) {
ctr++
panic(fmt.Sprintf("ouch %d", ctr))
}
cv := cache.NewCachedValue(pfunc, 25*time.Millisecond)
defer cv.Remove()
retrieve := func() error { _, err := cv.Value(); return err }
// Asserts.
assert.ErrorMatch(retrieve(), `.* cannot retrieve cached value: ouch 1`)
assert.ErrorMatch(retrieve(), `.* cannot retrieve cached value: ouch 2`)
assert.ErrorMatch(retrieve(), `.* cannot retrieve cached value: ouch 3`)
}
// Test the normal retrieving without errors.
func TestNormalRetrieve(t *testing.T) {
assert := audit.NewTestingAssertion(t, true)
// Environment.
ctr := 0
count := func() (interface{}, error) {
ctr++
return ctr, nil
}
cv := cache.NewCachedValue(count, 25*time.Millisecond)
defer cv.Remove()
retrieve := func() int { v, _ := cv.Value(); return v.(int) }
// Asserts.
assert.Equal(retrieve(), 1)
assert.Equal(retrieve(), 1)
time.Sleep(100 * time.Millisecond)
assert.Equal(retrieve(), 2)
time.Sleep(100 * time.Millisecond)
assert.Equal(retrieve(), 3)
assert.Equal(retrieve(), 3)
}