mirror of
https://github.com/go-shiori/shiori.git
synced 2025-01-15 20:37:44 +08:00
cc7c75116d
* migrate bookmark content route to new http server * new archive page * remove unused go generate comment * database mock * utils cleanup * unused var * domains refactor and tests * fixed secret key type * redirect to login on ui errors * fixed archive folder with storage domain * webroot documentation * some bookmark route tests * fixed error in bookmark domain for non existant bookmarks * centralice errors * add coverage data to unittests * added tests, refactor storage to use afero * removed mock to avoid increasing complexity * using deps to copy files around * remove config usage (to deps) * remove handler-ui file
105 lines
2.4 KiB
Go
105 lines
2.4 KiB
Go
package domains_test
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/go-shiori/shiori/internal/config"
|
|
"github.com/go-shiori/shiori/internal/dependencies"
|
|
"github.com/go-shiori/shiori/internal/domains"
|
|
"github.com/sirupsen/logrus"
|
|
"github.com/spf13/afero"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestDirExists(t *testing.T) {
|
|
fs := afero.NewMemMapFs()
|
|
fs.MkdirAll("foo", 0755)
|
|
|
|
domain := domains.NewStorageDomain(
|
|
&dependencies.Dependencies{
|
|
Config: config.ParseServerConfiguration(context.TODO(), logrus.New()),
|
|
Log: logrus.New(),
|
|
},
|
|
fs,
|
|
)
|
|
|
|
require.True(t, domain.DirExists("foo"))
|
|
require.False(t, domain.DirExists("foo/file"))
|
|
require.False(t, domain.DirExists("bar"))
|
|
}
|
|
|
|
func TestFileExists(t *testing.T) {
|
|
fs := afero.NewMemMapFs()
|
|
fs.MkdirAll("foo", 0755)
|
|
fs.Create("foo/file")
|
|
|
|
domain := domains.NewStorageDomain(
|
|
&dependencies.Dependencies{
|
|
Config: config.ParseServerConfiguration(context.TODO(), logrus.New()),
|
|
Log: logrus.New(),
|
|
},
|
|
fs,
|
|
)
|
|
|
|
require.True(t, domain.FileExists("foo/file"))
|
|
require.False(t, domain.FileExists("bar"))
|
|
}
|
|
|
|
func TestWriteFile(t *testing.T) {
|
|
fs := afero.NewMemMapFs()
|
|
|
|
domain := domains.NewStorageDomain(
|
|
&dependencies.Dependencies{
|
|
Config: config.ParseServerConfiguration(context.TODO(), logrus.New()),
|
|
Log: logrus.New(),
|
|
},
|
|
fs,
|
|
)
|
|
|
|
err := domain.WriteData("foo/file.ext", []byte("foo"))
|
|
require.NoError(t, err)
|
|
require.True(t, domain.FileExists("foo/file.ext"))
|
|
require.True(t, domain.DirExists("foo"))
|
|
handler, err := domain.FS().Open("foo/file.ext")
|
|
require.NoError(t, err)
|
|
defer handler.Close()
|
|
|
|
data, err := afero.ReadAll(handler)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, "foo", string(data))
|
|
}
|
|
|
|
func TestSaveFile(t *testing.T) {
|
|
fs := afero.NewMemMapFs()
|
|
|
|
domain := domains.NewStorageDomain(
|
|
&dependencies.Dependencies{
|
|
Config: config.ParseServerConfiguration(context.TODO(), logrus.New()),
|
|
Log: logrus.New(),
|
|
},
|
|
fs,
|
|
)
|
|
|
|
tempFile, err := os.CreateTemp("", "")
|
|
require.NoError(t, err)
|
|
defer os.Remove(tempFile.Name())
|
|
|
|
_, err = tempFile.WriteString("foo")
|
|
require.NoError(t, err)
|
|
|
|
err = domain.WriteFile("foo/file.ext", tempFile)
|
|
require.NoError(t, err)
|
|
require.True(t, domain.FileExists("foo/file.ext"))
|
|
require.True(t, domain.DirExists("foo"))
|
|
handler, err := domain.FS().Open("foo/file.ext")
|
|
require.NoError(t, err)
|
|
defer handler.Close()
|
|
|
|
data, err := afero.ReadAll(handler)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, "foo", string(data))
|
|
}
|