chore(internal/provider): GetConnection test
This commit is contained in:
@@ -24,10 +24,10 @@ func Test_Provider_GetConnection(t *testing.T) {
|
|||||||
errWrapped error
|
errWrapped error
|
||||||
errMessage string
|
errMessage string
|
||||||
}{
|
}{
|
||||||
"no server available": {
|
"no server": {
|
||||||
selection: settings.ServerSelection{}.WithDefaults(providers.Expressvpn),
|
selection: settings.ServerSelection{}.WithDefaults(providers.Expressvpn),
|
||||||
errWrapped: utils.ErrNoServerFound,
|
errWrapped: utils.ErrNoServer,
|
||||||
errMessage: "cannot filter servers: no server found: for VPN openvpn; protocol udp",
|
errMessage: "no server",
|
||||||
},
|
},
|
||||||
"no filter": {
|
"no filter": {
|
||||||
servers: []models.Server{
|
servers: []models.Server{
|
||||||
|
|||||||
@@ -26,8 +26,8 @@ func Test_Ivpn_GetConnection(t *testing.T) {
|
|||||||
}{
|
}{
|
||||||
"no server available": {
|
"no server available": {
|
||||||
selection: settings.ServerSelection{}.WithDefaults(providers.Ivpn),
|
selection: settings.ServerSelection{}.WithDefaults(providers.Ivpn),
|
||||||
errWrapped: utils.ErrNoServerFound,
|
errWrapped: utils.ErrNoServer,
|
||||||
errMessage: "cannot filter servers: no server found: for VPN openvpn; protocol udp",
|
errMessage: "no server",
|
||||||
},
|
},
|
||||||
"no filter": {
|
"no filter": {
|
||||||
servers: []models.Server{
|
servers: []models.Server{
|
||||||
|
|||||||
@@ -26,8 +26,8 @@ func Test_Mullvad_GetConnection(t *testing.T) {
|
|||||||
}{
|
}{
|
||||||
"no server available": {
|
"no server available": {
|
||||||
selection: settings.ServerSelection{}.WithDefaults(providers.Mullvad),
|
selection: settings.ServerSelection{}.WithDefaults(providers.Mullvad),
|
||||||
errWrapped: utils.ErrNoServerFound,
|
errWrapped: utils.ErrNoServer,
|
||||||
errMessage: "cannot filter servers: no server found: for VPN openvpn; protocol udp",
|
errMessage: "no server",
|
||||||
},
|
},
|
||||||
"no filter": {
|
"no filter": {
|
||||||
servers: []models.Server{
|
servers: []models.Server{
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
package utils
|
package utils
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"errors"
|
||||||
"math/rand"
|
"math/rand"
|
||||||
|
|
||||||
"github.com/qdm12/gluetun/internal/configuration/settings"
|
"github.com/qdm12/gluetun/internal/configuration/settings"
|
||||||
@@ -24,14 +24,20 @@ func NewConnectionDefaults(openvpnTCPPort, openvpnUDPPort,
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
var ErrNoServer = errors.New("no server")
|
||||||
|
|
||||||
func GetConnection(servers []models.Server,
|
func GetConnection(servers []models.Server,
|
||||||
selection settings.ServerSelection,
|
selection settings.ServerSelection,
|
||||||
defaults ConnectionDefaults,
|
defaults ConnectionDefaults,
|
||||||
randSource rand.Source) (
|
randSource rand.Source) (
|
||||||
connection models.Connection, err error) {
|
connection models.Connection, err error) {
|
||||||
servers, err = FilterServers(servers, selection)
|
if len(servers) == 0 {
|
||||||
if err != nil {
|
return connection, ErrNoServer
|
||||||
return connection, fmt.Errorf("cannot filter servers: %w", err)
|
}
|
||||||
|
|
||||||
|
servers = FilterServers(servers, selection)
|
||||||
|
if len(servers) == 0 {
|
||||||
|
return connection, NoServerFoundError(selection)
|
||||||
}
|
}
|
||||||
|
|
||||||
protocol := getProtocol(selection)
|
protocol := getProtocol(selection)
|
||||||
|
|||||||
@@ -1,9 +1,184 @@
|
|||||||
package utils
|
package utils
|
||||||
|
|
||||||
import "testing"
|
import (
|
||||||
|
"math/rand"
|
||||||
|
"net"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/qdm12/gluetun/internal/configuration/settings"
|
||||||
|
"github.com/qdm12/gluetun/internal/constants"
|
||||||
|
"github.com/qdm12/gluetun/internal/constants/providers"
|
||||||
|
"github.com/qdm12/gluetun/internal/constants/vpn"
|
||||||
|
"github.com/qdm12/gluetun/internal/models"
|
||||||
|
"github.com/stretchr/testify/assert"
|
||||||
|
)
|
||||||
|
|
||||||
func Test_GetConnection(t *testing.T) {
|
func Test_GetConnection(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
// testCases := map[string]struct{}{}
|
testCases := map[string]struct {
|
||||||
|
servers []models.Server
|
||||||
|
serverSelection settings.ServerSelection
|
||||||
|
defaults ConnectionDefaults
|
||||||
|
randSource rand.Source
|
||||||
|
connection models.Connection
|
||||||
|
errWrapped error
|
||||||
|
errMessage string
|
||||||
|
}{
|
||||||
|
"no server": {
|
||||||
|
serverSelection: settings.ServerSelection{}.
|
||||||
|
WithDefaults(providers.Mullvad),
|
||||||
|
errWrapped: ErrNoServer,
|
||||||
|
errMessage: "no server",
|
||||||
|
},
|
||||||
|
"all servers filtered": {
|
||||||
|
servers: []models.Server{
|
||||||
|
{VPN: vpn.Wireguard},
|
||||||
|
{VPN: vpn.Wireguard},
|
||||||
|
},
|
||||||
|
serverSelection: settings.ServerSelection{
|
||||||
|
VPN: vpn.OpenVPN,
|
||||||
|
}.WithDefaults(providers.Mullvad),
|
||||||
|
errWrapped: ErrNoServerFound,
|
||||||
|
errMessage: "no server found: for VPN openvpn; protocol udp",
|
||||||
|
},
|
||||||
|
"server without IPs": {
|
||||||
|
servers: []models.Server{
|
||||||
|
{VPN: vpn.OpenVPN, UDP: true},
|
||||||
|
{VPN: vpn.OpenVPN, UDP: true},
|
||||||
|
},
|
||||||
|
serverSelection: settings.ServerSelection{}.
|
||||||
|
WithDefaults(providers.Mullvad),
|
||||||
|
defaults: ConnectionDefaults{
|
||||||
|
OpenVPNTCPPort: 1,
|
||||||
|
OpenVPNUDPPort: 1,
|
||||||
|
WireguardPort: 1,
|
||||||
|
},
|
||||||
|
errWrapped: ErrNoConnectionToPickFrom,
|
||||||
|
errMessage: "no connection to pick from",
|
||||||
|
},
|
||||||
|
"OpenVPN server with hostname": {
|
||||||
|
servers: []models.Server{
|
||||||
|
{
|
||||||
|
VPN: vpn.OpenVPN,
|
||||||
|
UDP: true,
|
||||||
|
IPs: []net.IP{net.IPv4(1, 1, 1, 1)},
|
||||||
|
Hostname: "name",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
serverSelection: settings.ServerSelection{}.
|
||||||
|
WithDefaults(providers.Mullvad),
|
||||||
|
defaults: NewConnectionDefaults(443, 1194, 58820),
|
||||||
|
randSource: rand.NewSource(0),
|
||||||
|
connection: models.Connection{
|
||||||
|
Type: vpn.OpenVPN,
|
||||||
|
IP: net.IPv4(1, 1, 1, 1),
|
||||||
|
Protocol: constants.UDP,
|
||||||
|
Port: 1194,
|
||||||
|
Hostname: "name",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
"OpenVPN server with x509": {
|
||||||
|
servers: []models.Server{
|
||||||
|
{
|
||||||
|
VPN: vpn.OpenVPN,
|
||||||
|
UDP: true,
|
||||||
|
IPs: []net.IP{net.IPv4(1, 1, 1, 1)},
|
||||||
|
Hostname: "hostname",
|
||||||
|
OvpnX509: "x509",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
serverSelection: settings.ServerSelection{}.
|
||||||
|
WithDefaults(providers.Mullvad),
|
||||||
|
defaults: NewConnectionDefaults(443, 1194, 58820),
|
||||||
|
randSource: rand.NewSource(0),
|
||||||
|
connection: models.Connection{
|
||||||
|
Type: vpn.OpenVPN,
|
||||||
|
IP: net.IPv4(1, 1, 1, 1),
|
||||||
|
Protocol: constants.UDP,
|
||||||
|
Port: 1194,
|
||||||
|
Hostname: "x509",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
"server with IPv4 and IPv6": {
|
||||||
|
servers: []models.Server{
|
||||||
|
{
|
||||||
|
VPN: vpn.OpenVPN,
|
||||||
|
UDP: true,
|
||||||
|
IPs: []net.IP{
|
||||||
|
net.IPv4(1, 1, 1, 1),
|
||||||
|
// All IPv6 is ignored
|
||||||
|
net.IPv6zero,
|
||||||
|
net.IPv6zero,
|
||||||
|
net.IPv6zero,
|
||||||
|
net.IPv6zero,
|
||||||
|
net.IPv6zero,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
serverSelection: settings.ServerSelection{}.
|
||||||
|
WithDefaults(providers.Mullvad),
|
||||||
|
defaults: NewConnectionDefaults(443, 1194, 58820),
|
||||||
|
randSource: rand.NewSource(0),
|
||||||
|
connection: models.Connection{
|
||||||
|
Type: vpn.OpenVPN,
|
||||||
|
IP: net.IPv4(1, 1, 1, 1),
|
||||||
|
Protocol: constants.UDP,
|
||||||
|
Port: 1194,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
"mixed servers": {
|
||||||
|
servers: []models.Server{
|
||||||
|
{
|
||||||
|
VPN: vpn.OpenVPN,
|
||||||
|
UDP: true,
|
||||||
|
IPs: []net.IP{net.IPv4(1, 1, 1, 1)},
|
||||||
|
OvpnX509: "ovpnx509",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
VPN: vpn.Wireguard,
|
||||||
|
UDP: true,
|
||||||
|
IPs: []net.IP{net.IPv4(2, 2, 2, 2)},
|
||||||
|
OvpnX509: "ovpnx509",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
VPN: vpn.OpenVPN,
|
||||||
|
UDP: true,
|
||||||
|
IPs: []net.IP{
|
||||||
|
net.IPv4(3, 3, 3, 3),
|
||||||
|
{1, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1}, // ipv6 ignored
|
||||||
|
},
|
||||||
|
Hostname: "hostname",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
serverSelection: settings.ServerSelection{}.
|
||||||
|
WithDefaults(providers.Mullvad),
|
||||||
|
defaults: NewConnectionDefaults(443, 1194, 58820),
|
||||||
|
randSource: rand.NewSource(0),
|
||||||
|
connection: models.Connection{
|
||||||
|
Type: vpn.OpenVPN,
|
||||||
|
IP: net.IPv4(1, 1, 1, 1),
|
||||||
|
Protocol: constants.UDP,
|
||||||
|
Port: 1194,
|
||||||
|
Hostname: "ovpnx509",
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for name, testCase := range testCases {
|
||||||
|
testCase := testCase
|
||||||
|
t.Run(name, func(t *testing.T) {
|
||||||
|
t.Parallel()
|
||||||
|
|
||||||
|
connection, err := GetConnection(testCase.servers,
|
||||||
|
testCase.serverSelection, testCase.defaults,
|
||||||
|
testCase.randSource)
|
||||||
|
|
||||||
|
assert.Equal(t, testCase.connection, connection)
|
||||||
|
assert.ErrorIs(t, err, testCase.errWrapped)
|
||||||
|
if testCase.errWrapped != nil {
|
||||||
|
assert.EqualError(t, err, testCase.errMessage)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -8,9 +8,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
func FilterServers(servers []models.Server,
|
func FilterServers(servers []models.Server,
|
||||||
selection settings.ServerSelection) (
|
selection settings.ServerSelection) (filtered []models.Server) {
|
||||||
filtered []models.Server, err error,
|
|
||||||
) {
|
|
||||||
for _, server := range servers {
|
for _, server := range servers {
|
||||||
if filterServer(server, selection) {
|
if filterServer(server, selection) {
|
||||||
continue
|
continue
|
||||||
@@ -19,11 +17,7 @@ func FilterServers(servers []models.Server,
|
|||||||
filtered = append(filtered, server)
|
filtered = append(filtered, server)
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(filtered) == 0 {
|
return filtered
|
||||||
return nil, NoServerFoundError(selection)
|
|
||||||
}
|
|
||||||
|
|
||||||
return filtered, nil
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func filterServer(server models.Server,
|
func filterServer(server models.Server,
|
||||||
|
|||||||
@@ -17,13 +17,9 @@ func Test_FilterServers(t *testing.T) {
|
|||||||
servers []models.Server
|
servers []models.Server
|
||||||
selection settings.ServerSelection
|
selection settings.ServerSelection
|
||||||
filtered []models.Server
|
filtered []models.Server
|
||||||
errMessage string
|
|
||||||
errWrapped error
|
|
||||||
}{
|
}{
|
||||||
"no server available": {
|
"no server available": {
|
||||||
selection: settings.ServerSelection{}.WithDefaults(providers.Mullvad),
|
selection: settings.ServerSelection{}.WithDefaults(providers.Mullvad),
|
||||||
errMessage: "no server found: for VPN openvpn; protocol udp",
|
|
||||||
errWrapped: ErrNoServerFound,
|
|
||||||
},
|
},
|
||||||
"no filter": {
|
"no filter": {
|
||||||
servers: []models.Server{
|
servers: []models.Server{
|
||||||
@@ -216,12 +212,7 @@ func Test_FilterServers(t *testing.T) {
|
|||||||
t.Run(name, func(t *testing.T) {
|
t.Run(name, func(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
filtered, err := FilterServers(testCase.servers, testCase.selection)
|
filtered := FilterServers(testCase.servers, testCase.selection)
|
||||||
|
|
||||||
assert.ErrorIs(t, err, testCase.errWrapped)
|
|
||||||
if testCase.errWrapped != nil {
|
|
||||||
assert.EqualError(t, err, testCase.errMessage)
|
|
||||||
}
|
|
||||||
|
|
||||||
assert.Equal(t, testCase.filtered, filtered)
|
assert.Equal(t, testCase.filtered, filtered)
|
||||||
})
|
})
|
||||||
|
|||||||
@@ -28,8 +28,8 @@ func Test_Wevpn_GetConnection(t *testing.T) {
|
|||||||
selection: settings.ServerSelection{
|
selection: settings.ServerSelection{
|
||||||
VPN: vpn.OpenVPN,
|
VPN: vpn.OpenVPN,
|
||||||
}.WithDefaults(providers.Wevpn),
|
}.WithDefaults(providers.Wevpn),
|
||||||
errWrapped: utils.ErrNoServerFound,
|
errWrapped: utils.ErrNoServer,
|
||||||
errMessage: "cannot filter servers: no server found: for VPN openvpn; protocol udp",
|
errMessage: "no server",
|
||||||
},
|
},
|
||||||
"no filter": {
|
"no filter": {
|
||||||
servers: []models.Server{
|
servers: []models.Server{
|
||||||
|
|||||||
@@ -26,8 +26,8 @@ func Test_Windscribe_GetConnection(t *testing.T) {
|
|||||||
}{
|
}{
|
||||||
"no server available": {
|
"no server available": {
|
||||||
selection: settings.ServerSelection{}.WithDefaults(providers.Windscribe),
|
selection: settings.ServerSelection{}.WithDefaults(providers.Windscribe),
|
||||||
errWrapped: utils.ErrNoServerFound,
|
errWrapped: utils.ErrNoServer,
|
||||||
errMessage: "cannot filter servers: no server found: for VPN openvpn; protocol udp",
|
errMessage: "no server",
|
||||||
},
|
},
|
||||||
"no filter": {
|
"no filter": {
|
||||||
servers: []models.Server{
|
servers: []models.Server{
|
||||||
|
|||||||
Reference in New Issue
Block a user