mirror of
https://github.com/netbirdio/netbird.git
synced 2024-12-25 16:18:49 +01:00
105 lines
2.8 KiB
Go
105 lines
2.8 KiB
Go
|
package iface
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/hex"
|
||
|
|
"fmt"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
"github.com/stretchr/testify/require"
|
||
|
|
"golang.zx2c4.com/wireguard/wgctrl/wgtypes"
|
||
|
|
)
|
||
|
|
|
||
|
|
var ipcFixture = `
|
||
|
|
private_key=e84b5a6d2717c1003a13b431570353dbaca9146cf150c5f8575680feba52027a
|
||
|
|
listen_port=12912
|
||
|
|
public_key=b85996fecc9c7f1fc6d2572a76eda11d59bcd20be8e543b15ce4bd85a8e75a33
|
||
|
|
preshared_key=188515093e952f5f22e865cef3012e72f8b5f0b598ac0309d5dacce3b70fcf52
|
||
|
|
allowed_ip=192.168.4.4/32
|
||
|
|
endpoint=[abcd:23::33%2]:51820
|
||
|
|
public_key=58402e695ba1772b1cc9309755f043251ea77fdcf10fbe63989ceb7e19321376
|
||
|
|
tx_bytes=38333
|
||
|
|
rx_bytes=2224
|
||
|
|
allowed_ip=192.168.4.6/32
|
||
|
|
persistent_keepalive_interval=111
|
||
|
|
endpoint=182.122.22.19:3233
|
||
|
|
public_key=662e14fd594556f522604703340351258903b64f35553763f19426ab2a515c58
|
||
|
|
endpoint=5.152.198.39:51820
|
||
|
|
allowed_ip=192.168.4.10/32
|
||
|
|
allowed_ip=192.168.4.11/32
|
||
|
|
tx_bytes=1212111
|
||
|
|
rx_bytes=1929999999
|
||
|
|
protocol_version=1
|
||
|
|
errno=0
|
||
|
|
|
||
|
|
`
|
||
|
|
|
||
|
|
func Test_findPeerInfo(t *testing.T) {
|
||
|
|
tests := []struct {
|
||
|
|
name string
|
||
|
|
peerKey string
|
||
|
|
searchKeys []string
|
||
|
|
want map[string]string
|
||
|
|
wantErr bool
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "single",
|
||
|
|
peerKey: "58402e695ba1772b1cc9309755f043251ea77fdcf10fbe63989ceb7e19321376",
|
||
|
|
searchKeys: []string{"tx_bytes"},
|
||
|
|
want: map[string]string{
|
||
|
|
"tx_bytes": "38333",
|
||
|
|
},
|
||
|
|
wantErr: false,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "multiple",
|
||
|
|
peerKey: "58402e695ba1772b1cc9309755f043251ea77fdcf10fbe63989ceb7e19321376",
|
||
|
|
searchKeys: []string{"tx_bytes", "rx_bytes"},
|
||
|
|
want: map[string]string{
|
||
|
|
"tx_bytes": "38333",
|
||
|
|
"rx_bytes": "2224",
|
||
|
|
},
|
||
|
|
wantErr: false,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "lastpeer",
|
||
|
|
peerKey: "662e14fd594556f522604703340351258903b64f35553763f19426ab2a515c58",
|
||
|
|
searchKeys: []string{"tx_bytes", "rx_bytes"},
|
||
|
|
want: map[string]string{
|
||
|
|
"tx_bytes": "1212111",
|
||
|
|
"rx_bytes": "1929999999",
|
||
|
|
},
|
||
|
|
wantErr: false,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "peer not found",
|
||
|
|
peerKey: "1111111111111111111111111111111111111111111111111111111111111111",
|
||
|
|
searchKeys: nil,
|
||
|
|
want: nil,
|
||
|
|
wantErr: true,
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "key not found",
|
||
|
|
peerKey: "662e14fd594556f522604703340351258903b64f35553763f19426ab2a515c58",
|
||
|
|
searchKeys: []string{"tx_bytes", "unknown_key"},
|
||
|
|
want: map[string]string{
|
||
|
|
"tx_bytes": "1212111",
|
||
|
|
},
|
||
|
|
wantErr: true,
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, tt := range tests {
|
||
|
|
t.Run(tt.name, func(t *testing.T) {
|
||
|
|
res, err := hex.DecodeString(tt.peerKey)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
key, err := wgtypes.NewKey(res)
|
||
|
|
require.NoError(t, err)
|
||
|
|
|
||
|
|
got, err := findPeerInfo(ipcFixture, key.String(), tt.searchKeys)
|
||
|
|
assert.Equalf(t, tt.wantErr, err != nil, fmt.Sprintf("findPeerInfo(%v, %v, %v)", ipcFixture, key.String(), tt.searchKeys))
|
||
|
|
assert.Equalf(t, tt.want, got, "findPeerInfo(%v, %v, %v)", ipcFixture, key.String(), tt.searchKeys)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|