99 lines
2.2 KiB
Go
99 lines
2.2 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package trust_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"storj.io/common/testcontext"
|
|
"storj.io/storj/storagenode/trust"
|
|
)
|
|
|
|
func TestFileSourceString(t *testing.T) {
|
|
source := trust.NewFileSource("/some/path")
|
|
require.Equal(t, "/some/path", source.String())
|
|
}
|
|
|
|
func TestFileSourceIsStatic(t *testing.T) {
|
|
source := trust.NewFileSource("/some/path")
|
|
require.True(t, source.Static(), "file source is unexpectedly not static")
|
|
}
|
|
|
|
func TestFileSourceFetchEntries(t *testing.T) {
|
|
ctx := testcontext.New(t)
|
|
defer ctx.Cleanup()
|
|
|
|
url1 := makeSatelliteURL("domain1.test")
|
|
url2 := makeSatelliteURL("domain2.test")
|
|
|
|
// Prepare a directory with a couple of lists
|
|
goodData := fmt.Sprintf(`
|
|
# Some comment
|
|
%s
|
|
%s
|
|
`, url1.String(), url2.String())
|
|
goodPath := ctx.File("good.txt")
|
|
require.NoError(t, os.WriteFile(goodPath, []byte(goodData), 0644))
|
|
|
|
badData := `BAD`
|
|
badPath := ctx.File("bad.txt")
|
|
require.NoError(t, os.WriteFile(badPath, []byte(badData), 0644))
|
|
|
|
missingPath := ctx.File("missing.txt")
|
|
|
|
for _, tt := range []struct {
|
|
name string
|
|
path string
|
|
errs []string
|
|
entries []trust.Entry
|
|
}{
|
|
{
|
|
name: "bad list",
|
|
path: badPath,
|
|
errs: []string{"file source: invalid satellite URL: must contain an ID"},
|
|
},
|
|
{
|
|
name: "missing list",
|
|
path: missingPath,
|
|
errs: []string{
|
|
fmt.Sprintf("file source: open %s: no such file or directory", missingPath),
|
|
fmt.Sprintf("file source: open %s: The system cannot find the file specified.", missingPath),
|
|
},
|
|
},
|
|
{
|
|
name: "good list",
|
|
path: goodPath,
|
|
entries: []trust.Entry{
|
|
{
|
|
SatelliteURL: url1,
|
|
Authoritative: true,
|
|
},
|
|
{
|
|
SatelliteURL: url2,
|
|
Authoritative: true,
|
|
},
|
|
},
|
|
},
|
|
} {
|
|
tt := tt // quiet linting
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
source := trust.NewFileSource(tt.path)
|
|
entries, err := source.FetchEntries(context.Background())
|
|
if len(tt.errs) > 0 {
|
|
require.Error(t, err)
|
|
require.Contains(t, tt.errs, err.Error())
|
|
return
|
|
}
|
|
require.NoError(t, err)
|
|
assert.Equal(t, tt.entries, entries)
|
|
})
|
|
}
|
|
}
|