64 lines
2.2 KiB
Go
64 lines
2.2 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information.
|
|
|
|
package trust_test
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"storj.io/common/testrand"
|
|
"storj.io/storj/storagenode/trust"
|
|
)
|
|
|
|
func TestSourcesConfig(t *testing.T) {
|
|
var sources trust.Sources
|
|
assert.Equal(t, "trust-sources", sources.Type())
|
|
assert.Equal(t, "", sources.String())
|
|
|
|
// Assert that comma separated sources can be set
|
|
require.NoError(t, sources.Set("some-file,12ugxzSYW7UA1n7jfVyJHV2zL5TL1x5fSYZb48y7jZ1r3bLhbfo@127.0.0.1:12345"))
|
|
assert.Equal(t, "some-file,12ugxzSYW7UA1n7jfVyJHV2zL5TL1x5fSYZb48y7jZ1r3bLhbfo@127.0.0.1:12345", sources.String())
|
|
|
|
// Assert that a failure to set does not modify the current sources
|
|
require.Error(t, sources.Set("@"))
|
|
assert.Equal(t, "some-file,12ugxzSYW7UA1n7jfVyJHV2zL5TL1x5fSYZb48y7jZ1r3bLhbfo@127.0.0.1:12345", sources.String())
|
|
|
|
// Assert the source was sources correctly
|
|
if assert.Len(t, sources, 2) {
|
|
assert.Equal(t, "some-file", sources[0].String())
|
|
assert.Equal(t, "12ugxzSYW7UA1n7jfVyJHV2zL5TL1x5fSYZb48y7jZ1r3bLhbfo@127.0.0.1:12345", sources[1].String())
|
|
}
|
|
}
|
|
|
|
func TestExclusionsConfig(t *testing.T) {
|
|
exclusion0 := fmt.Sprintf("%s@", testrand.NodeID())
|
|
exclusion1 := "foo.test"
|
|
exclusion2 := fmt.Sprintf("%s@bar.test:7777", testrand.NodeID())
|
|
|
|
exclusionConfig := strings.Join([]string{exclusion0, exclusion1, exclusion2}, ",")
|
|
|
|
var exclusions trust.Exclusions
|
|
assert.Equal(t, "trust-exclusions", exclusions.Type())
|
|
assert.Equal(t, "", exclusions.String())
|
|
|
|
// Assert that comma separated exclusions can be set
|
|
require.NoError(t, exclusions.Set(exclusionConfig))
|
|
assert.Equal(t, exclusionConfig, exclusions.String())
|
|
|
|
// Assert that a failure to set does not modify the current exclusions
|
|
require.Error(t, exclusions.Set("foo.test:7777"))
|
|
assert.Equal(t, exclusionConfig, exclusions.String())
|
|
|
|
// Assert the source was sources correctly
|
|
if assert.Len(t, exclusions.Rules, 3) {
|
|
assert.Equal(t, exclusion0, exclusions.Rules[0].String())
|
|
assert.Equal(t, exclusion1, exclusions.Rules[1].String())
|
|
assert.Equal(t, exclusion2, exclusions.Rules[2].String())
|
|
}
|
|
}
|