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
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
|
//go:build linux
package network
import (
"errors"
"fmt"
"strings"
"testing"
"github.com/vishvananda/netlink"
)
// mockNetworkOps allows us to control the behavior of netlink calls.
type mockNetworkOps struct {
linkByNameFunc func(name string) (netlink.Link, error)
linkSetMTUFunc func(link netlink.Link, mtu int) error
linkSetUpFunc func(link netlink.Link) error
addrAddFunc func(link netlink.Link, addr *netlink.Addr) error
routeAddFunc func(route *netlink.Route) error
routeReplaceFunc func(route *netlink.Route) error
}
func (m *mockNetworkOps) LinkList() ([]netlink.Link, error) { return nil, nil }
func (m *mockNetworkOps) LinkByName(name string) (netlink.Link, error) {
if m.linkByNameFunc != nil {
return m.linkByNameFunc(name)
}
return nil, fmt.Errorf("not implemented")
}
func (m *mockNetworkOps) LinkSetMTU(link netlink.Link, mtu int) error {
if m.linkSetMTUFunc != nil {
return m.linkSetMTUFunc(link, mtu)
}
return nil
}
func (m *mockNetworkOps) LinkSetUp(link netlink.Link) error {
if m.linkSetUpFunc != nil {
return m.linkSetUpFunc(link)
}
return nil
}
func (m *mockNetworkOps) AddrAdd(link netlink.Link, addr *netlink.Addr) error {
if m.addrAddFunc != nil {
return m.addrAddFunc(link, addr)
}
return nil
}
func (m *mockNetworkOps) RouteAdd(route *netlink.Route) error {
if m.routeAddFunc != nil {
return m.routeAddFunc(route)
}
return nil
}
func (m *mockNetworkOps) RouteReplace(route *netlink.Route) error {
if m.routeReplaceFunc != nil {
return m.routeReplaceFunc(route)
}
return nil
}
// mockLink implements netlink.Link.
type mockLink struct {
name string
idx int
}
func (m *mockLink) Type() string {
return "mock"
}
func (m *mockLink) Attrs() *netlink.LinkAttrs {
return &netlink.LinkAttrs{Name: m.name, Index: m.idx}
}
func TestConfigureInterface_Success(t *testing.T) {
t.Parallel()
mock := &mockNetworkOps{
linkByNameFunc: func(name string) (netlink.Link, error) {
return &mockLink{name: name, idx: 1}, nil
},
}
nm := &NetworkManager{Ops: mock}
err := nm.ConfigureInterface("tun0", "10.0.0.1/24", 1420)
if err != nil {
t.Errorf("expected success, got %v", err)
}
}
func TestConfigureInterface_RouteFallback(t *testing.T) {
t.Parallel()
routeAddCalled := false
routeReplaceCalled := false
mock := &mockNetworkOps{
linkByNameFunc: func(name string) (netlink.Link, error) {
return &mockLink{name: name, idx: 1}, nil
},
routeAddFunc: func(route *netlink.Route) error {
routeAddCalled = true
return errors.New("file exists") // Simulate EEXIST
},
routeReplaceFunc: func(route *netlink.Route) error {
routeReplaceCalled = true
return nil
},
}
nm := &NetworkManager{Ops: mock}
err := nm.ConfigureInterface("tun0", "10.0.0.1/24", 1420)
if err != nil {
t.Errorf("expected success after fallback, got %v", err)
}
if !routeAddCalled {
t.Error("expected RouteAdd to be called first")
}
if !routeReplaceCalled {
t.Error("expected RouteReplace to be called after RouteAdd fails with 'file exists'")
}
}
func TestConfigureInterface_RouteFailure(t *testing.T) {
t.Parallel()
mock := &mockNetworkOps{
linkByNameFunc: func(name string) (netlink.Link, error) {
return &mockLink{name: name, idx: 1}, nil
},
routeAddFunc: func(route *netlink.Route) error {
return errors.New("critical network failure")
},
routeReplaceFunc: func(route *netlink.Route) error {
return errors.New("critical network failure")
},
}
nm := &NetworkManager{Ops: mock}
err := nm.ConfigureInterface("tun0", "10.0.0.1/24", 1420)
if err == nil {
t.Error("expected error when both RouteAdd and RouteReplace fail, got nil")
}
if !strings.Contains(err.Error(), "failed to configure default route") {
t.Errorf("expected route error, got: %v", err)
}
}
func TestConfigureInterface_LinkNotFound(t *testing.T) {
t.Parallel()
mock := &mockNetworkOps{
linkByNameFunc: func(name string) (netlink.Link, error) {
return nil, errors.New("no such device")
},
}
nm := &NetworkManager{Ops: mock}
err := nm.ConfigureInterface("nonexistent", "10.0.0.1/24", 1420)
if err == nil {
t.Error("expected error when link is not found, got nil")
}
if !strings.Contains(err.Error(), "failed to find link") {
t.Errorf("expected link not found error, got: %v", err)
}
}
|
