68 lines
1.6 KiB
Go
68 lines
1.6 KiB
Go
package client
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http/httptest"
|
|
"net/url"
|
|
"os"
|
|
"strings"
|
|
"testing"
|
|
"time"
|
|
|
|
"forge.cadoles.com/arcad/edge/pkg/storage/driver/rpc/server"
|
|
"forge.cadoles.com/arcad/edge/pkg/storage/driver/sqlite"
|
|
"forge.cadoles.com/arcad/edge/pkg/storage/share"
|
|
"forge.cadoles.com/arcad/edge/pkg/storage/share/testsuite"
|
|
"github.com/pkg/errors"
|
|
"gitlab.com/wpetit/goweb/logger"
|
|
)
|
|
|
|
func TestShareStore(t *testing.T) {
|
|
t.Parallel()
|
|
if testing.Verbose() {
|
|
logger.SetLevel(logger.LevelDebug)
|
|
}
|
|
|
|
testsuite.TestStore(t, func(testName string) (share.Store, error) {
|
|
httpServer, err := startNewShareStoreServer(testName)
|
|
if err != nil {
|
|
return nil, errors.WithStack(err)
|
|
}
|
|
|
|
serverAddr := httpServer.Listener.Addr()
|
|
serverURL := &url.URL{
|
|
Host: serverAddr.String(),
|
|
}
|
|
|
|
return NewShareStore(serverURL), nil
|
|
})
|
|
}
|
|
|
|
func getSQLiteShareStore(testName string) (*sqlite.ShareStore, error) {
|
|
filename := strings.ToLower(strings.ReplaceAll(testName, " ", "_"))
|
|
|
|
file := fmt.Sprintf("./testdata/sharestore_test_%s.sqlite", filename)
|
|
|
|
if err := os.Remove(file); err != nil && !errors.Is(err, os.ErrNotExist) {
|
|
return nil, errors.WithStack(err)
|
|
}
|
|
|
|
dsn := fmt.Sprintf("%s?_pragma=foreign_keys(1)&_pragma=busy_timeout=%d", file, (60 * time.Second).Milliseconds())
|
|
store := sqlite.NewShareStore(dsn)
|
|
|
|
return store, nil
|
|
}
|
|
|
|
func startNewShareStoreServer(testName string) (*httptest.Server, error) {
|
|
store, err := getSQLiteShareStore(testName)
|
|
if err != nil {
|
|
return nil, errors.WithStack(err)
|
|
}
|
|
|
|
server := server.NewShareStoreServer(store)
|
|
|
|
httpServer := httptest.NewServer(server)
|
|
|
|
return httpServer, nil
|
|
}
|