mirror of
https://github.com/grafana/grafana.git
synced 2024-11-28 11:44:26 -06:00
72 lines
2.0 KiB
Go
72 lines
2.0 KiB
Go
package store
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"mime/multipart"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/grafana/grafana-plugin-sdk-go/experimental"
|
|
"github.com/grafana/grafana/pkg/models"
|
|
"github.com/grafana/grafana/pkg/services/featuremgmt"
|
|
"github.com/grafana/grafana/pkg/setting"
|
|
"github.com/grafana/grafana/pkg/tsdb/testdatasource"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
var (
|
|
dummyUser = &models.SignedInUser{OrgId: 1}
|
|
)
|
|
|
|
func TestListFiles(t *testing.T) {
|
|
publicRoot, err := filepath.Abs("../../../public")
|
|
require.NoError(t, err)
|
|
roots := []storageRuntime{
|
|
newDiskStorage("public", "Public static files", &StorageLocalDiskConfig{
|
|
Path: publicRoot,
|
|
Roots: []string{
|
|
"/testdata/",
|
|
"/img/icons/",
|
|
"/img/bg/",
|
|
"/gazetteer/",
|
|
"/maps/",
|
|
"/upload/",
|
|
},
|
|
}).setReadOnly(true).setBuiltin(true),
|
|
}
|
|
|
|
store := newStandardStorageService(roots, func(orgId int64) []storageRuntime {
|
|
return make([]storageRuntime, 0)
|
|
})
|
|
frame, err := store.List(context.Background(), dummyUser, "public/testdata")
|
|
require.NoError(t, err)
|
|
|
|
experimental.CheckGoldenJSONFrame(t, "testdata", "public_testdata.golden", frame, true)
|
|
|
|
file, err := store.Read(context.Background(), dummyUser, "public/testdata/js_libraries.csv")
|
|
require.NoError(t, err)
|
|
require.NotNil(t, file)
|
|
|
|
frame, err = testdatasource.LoadCsvContent(bytes.NewReader(file.Contents), file.Name)
|
|
require.NoError(t, err)
|
|
experimental.CheckGoldenJSONFrame(t, "testdata", "public_testdata_js_libraries.golden", frame, true)
|
|
}
|
|
|
|
func TestUpload(t *testing.T) {
|
|
features := featuremgmt.WithFeatures(featuremgmt.FlagStorageLocalUpload)
|
|
path, err := os.Getwd()
|
|
require.NoError(t, err)
|
|
cfg := &setting.Cfg{AppURL: "http://localhost:3000/", DataPath: path}
|
|
s := ProvideService(nil, features, cfg)
|
|
testForm := &multipart.Form{
|
|
Value: map[string][]string{},
|
|
File: map[string][]*multipart.FileHeader{},
|
|
}
|
|
res, err := s.Upload(context.Background(), dummyUser, testForm)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, res.path, "upload")
|
|
}
|