constellation/coordinator/pubapi/vpn_test.go
2022-06-03 11:55:18 +02:00

59 lines
1.4 KiB
Go

package pubapi
import (
"context"
"errors"
"testing"
"github.com/edgelesssys/constellation/coordinator/logging"
"github.com/edgelesssys/constellation/coordinator/peer"
"github.com/edgelesssys/constellation/coordinator/pubapi/pubproto"
"github.com/edgelesssys/constellation/coordinator/role"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
"go.uber.org/zap/zaptest"
)
func TestGetVPNPeers(t *testing.T) {
wantedPeers := []peer.Peer{
{
PublicIP: "192.0.2.1",
VPNIP: "10.118.0.1",
VPNPubKey: []byte{0x1, 0x2, 0x3},
Role: role.Coordinator,
},
}
testCases := map[string]struct {
coreGetPeersErr error
wantErr bool
}{
"GetVPNPeers works": {},
"GetVPNPeers fails if core cannot retrieve VPN peers": {
coreGetPeersErr: errors.New("failed to get peers"),
wantErr: true,
},
}
for name, tc := range testCases {
t.Run(name, func(t *testing.T) {
assert := assert.New(t)
require := require.New(t)
logger := zaptest.NewLogger(t)
cor := &fakeCore{peers: wantedPeers, GetPeersErr: tc.coreGetPeersErr}
api := New(logger, &logging.NopLogger{}, cor, nil, nil, nil, nil)
defer api.Close()
resp, err := api.GetVPNPeers(context.Background(), &pubproto.GetVPNPeersRequest{})
if tc.wantErr {
assert.Error(err)
return
}
require.NoError(err)
peers := peer.FromPubProto(resp.Peers)
assert.Equal(wantedPeers, peers)
})
}
}