7a4bb863dd
ioutil has been deprecated since go 1.16, this fixes all of the deprecated functions we are using: ioutil.ReadFile -> os.ReadFile ioutil.ReadAll -> io.ReadAll ioutil.WriteFile -> os.WriteFile ioutil.TempFile -> os.CreateTemp ioutil.TempDir -> os.MkdirTemp All of the above are a simple name change, the function arguments and results are exactly the same as before. ioutil.ReadDir -> os.ReadDir now returns a os.DirEntry but the IsDir and Name functions work the same. The difference is that the FileInfo must be retrieved with the Info() function which can also return an error. These were identified by running: golangci-lint run --build-tags=integration ./...
123 lines
2.6 KiB
Go
123 lines
2.6 KiB
Go
package jsondb_test
|
|
|
|
import (
|
|
"os"
|
|
"path"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/osbuild/osbuild-composer/internal/jsondb"
|
|
)
|
|
|
|
type document struct {
|
|
Animal string `json:"animal"`
|
|
CanSwim bool `json:"can-swim"`
|
|
}
|
|
|
|
// If the passed directory is not readable (writable), we should notice on the
|
|
// first read (write).
|
|
func TestDegenerate(t *testing.T) {
|
|
t.Run("no-exist", func(t *testing.T) {
|
|
db := jsondb.New("/non-existant-directory", 0755)
|
|
|
|
var d document
|
|
exist, err := db.Read("one", &d)
|
|
assert.False(t, exist)
|
|
assert.NoError(t, err)
|
|
|
|
err = db.Write("one", &d)
|
|
assert.Error(t, err)
|
|
|
|
l, err := db.List()
|
|
assert.Error(t, err)
|
|
assert.Nil(t, l)
|
|
})
|
|
|
|
t.Run("invalid-json", func(t *testing.T) {
|
|
dir := t.TempDir()
|
|
|
|
db := jsondb.New(dir, 0755)
|
|
|
|
// write-only file
|
|
err := os.WriteFile(path.Join(dir, "one.json"), []byte("{"), 0600)
|
|
require.NoError(t, err)
|
|
|
|
var d document
|
|
_, err = db.Read("one", &d)
|
|
assert.Error(t, err)
|
|
})
|
|
}
|
|
|
|
func TestCorrupt(t *testing.T) {
|
|
dir := t.TempDir()
|
|
|
|
err := os.WriteFile(path.Join(dir, "one.json"), []byte("{"), 0600)
|
|
require.NoError(t, err)
|
|
|
|
db := jsondb.New(dir, 0755)
|
|
var d document
|
|
_, err = db.Read("one", &d)
|
|
require.Error(t, err)
|
|
}
|
|
|
|
func TestRead(t *testing.T) {
|
|
dir := t.TempDir()
|
|
|
|
err := os.WriteFile(path.Join(dir, "one.json"), []byte("true"), 0600)
|
|
require.NoError(t, err)
|
|
|
|
db := jsondb.New(dir, 0755)
|
|
|
|
var b bool
|
|
exists, err := db.Read("one", &b)
|
|
require.NoError(t, err)
|
|
require.True(t, exists)
|
|
require.True(t, b)
|
|
|
|
// nil means don't deserialize
|
|
exists, err = db.Read("one", nil)
|
|
require.NoError(t, err)
|
|
require.True(t, exists)
|
|
|
|
b = false
|
|
exists, err = db.Read("two", &b)
|
|
require.NoError(t, err)
|
|
require.False(t, exists)
|
|
require.False(t, b)
|
|
|
|
// nil means don't deserialize
|
|
exists, err = db.Read("two", nil)
|
|
require.NoError(t, err)
|
|
require.False(t, exists)
|
|
}
|
|
|
|
func TestMultiple(t *testing.T) {
|
|
dir := t.TempDir()
|
|
|
|
perm := os.FileMode(0600)
|
|
documents := map[string]document{
|
|
"one": document{"octopus", true},
|
|
"two": document{"zebra", false},
|
|
"three": document{"clownfish", true},
|
|
}
|
|
|
|
db := jsondb.New(dir, perm)
|
|
|
|
for name, doc := range documents {
|
|
err := db.Write(name, doc)
|
|
require.NoError(t, err)
|
|
}
|
|
names, err := db.List()
|
|
require.NoError(t, err)
|
|
require.ElementsMatch(t, []string{"one", "two", "three"}, names)
|
|
|
|
for name, doc := range documents {
|
|
var d document
|
|
exist, err := db.Read(name, &d)
|
|
require.NoError(t, err)
|
|
require.True(t, exist)
|
|
require.Equalf(t, doc, d, "error retrieving document '%s'", name)
|
|
}
|
|
}
|