1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54
|
package x509bundle_test
import (
"testing"
"github.com/spiffe/go-spiffe/v2/bundle/x509bundle"
"github.com/stretchr/testify/require"
)
var (
b1 = x509bundle.New(td)
b2 = x509bundle.New(td2)
)
func TestNewSet(t *testing.T) {
s := x509bundle.NewSet(b1)
require.True(t, s.Has(td))
s = x509bundle.NewSet(b1, b2)
require.True(t, s.Has(td))
require.True(t, s.Has(td2))
}
func TestAdd(t *testing.T) {
s := x509bundle.NewSet()
require.False(t, s.Has(td))
s.Add(b1)
require.True(t, s.Has(td))
}
func TestRemove(t *testing.T) {
s := x509bundle.NewSet(b1)
require.True(t, s.Has(td))
s.Remove(td2)
require.True(t, s.Has(td))
s.Remove(td)
require.False(t, s.Has(td))
}
func TestHas(t *testing.T) {
s := x509bundle.NewSet(b1)
require.False(t, s.Has(td2))
require.True(t, s.Has(td))
}
func TestSetGetX509BundleForTrustDomain(t *testing.T) {
s := x509bundle.NewSet(b1)
_, err := s.GetX509BundleForTrustDomain(td2)
require.EqualError(t, err, `x509bundle: no X.509 bundle for trust domain "domain2.test"`)
b, err := s.GetX509BundleForTrustDomain(td)
require.NoError(t, err)
require.Equal(t, b1, b)
}
|