Files
vikunja/pkg/modules/keyvalue/keyvalue_test.go
2025-07-17 16:19:13 +02:00

84 lines
2.1 KiB
Go

// Vikunja is a to-do list application to facilitate your life.
// Copyright 2018-present Vikunja and contributors. All rights reserved.
//
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU Affero General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU Affero General Public License for more details.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <https://www.gnu.org/licenses/>.
package keyvalue
import (
"errors"
"testing"
"code.vikunja.io/api/pkg/modules/keyvalue/memory"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestRememberReturnsExisting(t *testing.T) {
store = memory.NewStorage()
err := Put("foo", "bar")
require.NoError(t, err)
called := false
val, err := Remember("foo", func() (interface{}, error) {
called = true
return "baz", nil
})
require.NoError(t, err)
assert.Equal(t, "bar", val)
assert.False(t, called)
}
func TestRememberComputesAndStores(t *testing.T) {
store = memory.NewStorage()
called := 0
val, err := Remember("foo", func() (interface{}, error) {
called++
return "bar", nil
})
require.NoError(t, err)
assert.Equal(t, "bar", val)
assert.Equal(t, 1, called)
v, exists, err := Get("foo")
require.NoError(t, err)
assert.True(t, exists)
assert.Equal(t, "bar", v)
val, err = Remember("foo", func() (interface{}, error) {
called++
return "baz", nil
})
require.NoError(t, err)
assert.Equal(t, "bar", val)
assert.Equal(t, 1, called)
}
func TestRememberErrorDoesNotStore(t *testing.T) {
store = memory.NewStorage()
_, err := Remember("foo", func() (interface{}, error) {
return nil, errors.New("fail")
})
require.Error(t, err)
_, exists, err2 := Get("foo")
require.NoError(t, err2)
assert.False(t, exists)
}