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
55
56
57
58
59
60
|
package workloadapi_test
import (
"testing"
"github.com/spiffe/go-spiffe/v2/bundle/jwtbundle"
"github.com/spiffe/go-spiffe/v2/bundle/spiffebundle"
"github.com/spiffe/go-spiffe/v2/bundle/x509bundle"
"github.com/spiffe/go-spiffe/v2/internal/test/fakeworkloadapi"
"github.com/spiffe/go-spiffe/v2/spiffeid"
"github.com/spiffe/go-spiffe/v2/svid/x509svid"
"github.com/spiffe/go-spiffe/v2/workloadapi"
"github.com/stretchr/testify/require"
)
func withAddr(api *fakeworkloadapi.WorkloadAPI) workloadapi.SourceOption {
return workloadapi.WithClientOptions(workloadapi.WithAddr(api.Addr()))
}
func requireBundle(tb testing.TB, source spiffebundle.Source, td spiffeid.TrustDomain, expected *spiffebundle.Bundle) {
actual, err := source.GetBundleForTrustDomain(td)
require.NoError(tb, err)
require.Equal(tb, expected, actual)
}
func requireNoBundle(tb testing.TB, source spiffebundle.Source, td spiffeid.TrustDomain, expected string) {
bundle, err := source.GetBundleForTrustDomain(td)
require.EqualError(tb, err, expected, "SPIFFE bundle should not exist")
require.Nil(tb, bundle)
}
func requireX509Bundle(tb testing.TB, source x509bundle.Source, td spiffeid.TrustDomain, expected *x509bundle.Bundle) {
actual, err := source.GetX509BundleForTrustDomain(td)
require.NoError(tb, err)
require.Equal(tb, expected, actual)
}
func requireNoX509Bundle(tb testing.TB, source x509bundle.Source, td spiffeid.TrustDomain, expected string) {
bundle, err := source.GetX509BundleForTrustDomain(td)
require.EqualError(tb, err, expected, "X.509 bundle should not exist")
require.Nil(tb, bundle)
}
func requireJWTBundle(tb testing.TB, source jwtbundle.Source, td spiffeid.TrustDomain, expected *jwtbundle.Bundle) {
actual, err := source.GetJWTBundleForTrustDomain(td)
require.NoError(tb, err)
require.Equal(tb, expected, actual)
}
func requireNoJWTBundle(tb testing.TB, source jwtbundle.Source, td spiffeid.TrustDomain, expected string) {
bundle, err := source.GetJWTBundleForTrustDomain(td)
require.EqualError(tb, err, expected, "JWT bundle should not exist")
require.Nil(tb, bundle)
}
func requireX509SVID(tb testing.TB, source x509svid.Source, expected *x509svid.SVID) {
actual, err := source.GetX509SVID()
require.NoError(tb, err)
require.Equal(tb, expected, actual)
}
|
