@@ -6,6 +6,7 @@ package system
6
6
import (
7
7
"context"
8
8
"errors"
9
+ "fmt"
9
10
"net"
10
11
"testing"
11
12
@@ -73,14 +74,14 @@ func (m *mockMetadata) ReverseLookupHost() (string, error) {
73
74
return args .String (0 ), args .Error (1 )
74
75
}
75
76
76
- func (m * mockMetadata ) HostIPs () ([]net. IP , error ) {
77
+ func (m * mockMetadata ) HostIPs () ([]system. InterfaceIP , error ) {
77
78
args := m .MethodCalled ("HostIPs" )
78
- return args .Get (0 ).([]net. IP ), args .Error (1 )
79
+ return args .Get (0 ).([]system. InterfaceIP ), args .Error (1 )
79
80
}
80
81
81
- func (m * mockMetadata ) HostMACs () ([]net. HardwareAddr , error ) {
82
+ func (m * mockMetadata ) HostMACs () ([]system. InterfaceMAC , error ) {
82
83
args := m .MethodCalled ("HostMACs" )
83
- return args .Get (0 ).([]net. HardwareAddr ), args .Error (1 )
84
+ return args .Get (0 ).([]system. InterfaceMAC ), args .Error (1 )
84
85
}
85
86
86
87
func (m * mockMetadata ) CPUInfo (_ context.Context ) ([]cpu.InfoStat , error ) {
@@ -89,11 +90,23 @@ func (m *mockMetadata) CPUInfo(_ context.Context) ([]cpu.InfoStat, error) {
89
90
}
90
91
91
92
var (
92
- testIPsAttribute = []any {"192.168.1.140" , "fe80::abc2:4a28:737a:609e" }
93
- testIPsAddresses = []net.IP {net .ParseIP (testIPsAttribute [0 ].(string )), net .ParseIP (testIPsAttribute [1 ].(string ))}
93
+ testIPs = []system.InterfaceIP {
94
+ {IP : net .ParseIP ("192.168.1.140" ), InterfaceName : "eth0" },
95
+ {IP : net .ParseIP ("fe80::abc2:4a28:737a:609e" ), InterfaceName : "eth1" },
96
+ }
97
+ testIPsAttribute = []any {
98
+ fmt .Sprintf ("Interface: %s, IP: %s" , "eth0" , "192.168.1.140" ),
99
+ fmt .Sprintf ("Interface: %s, IP: %s" , "eth1" , "fe80::abc2:4a28:737a:609e" ),
100
+ }
94
101
95
- testMACsAttribute = []any {"00-00-00-00-00-01" , "DE-AD-BE-EF-00-00" }
96
- testMACsAddresses = []net.HardwareAddr {{0x00 , 0x00 , 0x00 , 0x00 , 0x00 , 0x01 }, {0xDE , 0xAD , 0xBE , 0xEF , 0x00 , 0x00 }}
102
+ testMACs = []system.InterfaceMAC {
103
+ {MAC : net.HardwareAddr {0x00 , 0x00 , 0x00 , 0x00 , 0x00 , 0x01 }, InterfaceName : "eth0" },
104
+ {MAC : net.HardwareAddr {0xDE , 0xAD , 0xBE , 0xEF , 0x00 , 0x00 }, InterfaceName : "eth1" },
105
+ }
106
+ testMACsAttribute = []any {
107
+ fmt .Sprintf ("Interface: %s, MAC: %s" , "eth0" , "00-00-00-00-00-01" ),
108
+ fmt .Sprintf ("Interface: %s, MAC: %s" , "eth1" , "DE-AD-BE-EF-00-00" ),
109
+ }
97
110
)
98
111
99
112
func TestNewDetector (t * testing.T ) {
@@ -129,12 +142,12 @@ func TestToIEEERA(t *testing.T) {
129
142
expected string
130
143
}{
131
144
{
132
- addr : testMACsAddresses [ 0 ] ,
133
- expected : testMACsAttribute [ 0 ].( string ) ,
145
+ addr : net. HardwareAddr { 0x00 , 0x00 , 0x00 , 0x00 , 0x00 , 0x01 } ,
146
+ expected : "00-00-00-00-00-01" ,
134
147
},
135
148
{
136
- addr : testMACsAddresses [ 1 ] ,
137
- expected : testMACsAttribute [ 1 ].( string ) ,
149
+ addr : net. HardwareAddr { 0xDE , 0xAD , 0xBE , 0xEF , 0x00 , 0x00 } ,
150
+ expected : "DE-AD-BE-EF-00-00" ,
138
151
},
139
152
}
140
153
@@ -164,8 +177,8 @@ func TestDetectFQDNAvailable(t *testing.T) {
164
177
md .On ("OSVersion" ).Return ("22.04.2 LTS (Jammy Jellyfish)" , nil )
165
178
md .On ("HostID" ).Return ("2" , nil )
166
179
md .On ("HostArch" ).Return ("amd64" , nil )
167
- md .On ("HostIPs" ).Return (testIPsAddresses , nil )
168
- md .On ("HostMACs" ).Return (testMACsAddresses , nil )
180
+ md .On ("HostIPs" ).Return (testIPs , nil )
181
+ md .On ("HostMACs" ).Return (testMACs , nil )
169
182
170
183
detector := newTestDetector (md , []string {"dns" }, allEnabledConfig ())
171
184
res , schemaURL , err := detector .Detect (context .Background ())
@@ -222,8 +235,8 @@ func TestEnableHostID(t *testing.T) {
222
235
mdHostname .On ("OSVersion" ).Return ("22.04.2 LTS (Jammy Jellyfish)" , nil )
223
236
mdHostname .On ("HostID" ).Return ("3" , nil )
224
237
mdHostname .On ("HostArch" ).Return ("amd64" , nil )
225
- mdHostname .On ("HostIPs" ).Return (testIPsAddresses , nil )
226
- mdHostname .On ("HostMACs" ).Return (testMACsAddresses , nil )
238
+ mdHostname .On ("HostIPs" ).Return (testIPs , nil )
239
+ mdHostname .On ("HostMACs" ).Return (testMACs , nil )
227
240
228
241
detector := newTestDetector (mdHostname , []string {"dns" , "os" }, allEnabledConfig ())
229
242
res , schemaURL , err := detector .Detect (context .Background ())
@@ -253,8 +266,8 @@ func TestUseHostname(t *testing.T) {
253
266
mdHostname .On ("OSVersion" ).Return ("22.04.2 LTS (Jammy Jellyfish)" , nil )
254
267
mdHostname .On ("HostID" ).Return ("1" , nil )
255
268
mdHostname .On ("HostArch" ).Return ("amd64" , nil )
256
- mdHostname .On ("HostIPs" ).Return (testIPsAddresses , nil )
257
- mdHostname .On ("HostMACs" ).Return (testMACsAddresses , nil )
269
+ mdHostname .On ("HostIPs" ).Return (testIPs , nil )
270
+ mdHostname .On ("HostMACs" ).Return (testMACs , nil )
258
271
259
272
detector := newTestDetector (mdHostname , []string {"os" }, allEnabledConfig ())
260
273
res , schemaURL , err := detector .Detect (context .Background ())
@@ -286,8 +299,8 @@ func TestDetectError(t *testing.T) {
286
299
mdFQDN .On ("Hostname" ).Return ("" , errors .New ("err" ))
287
300
mdFQDN .On ("HostID" ).Return ("" , errors .New ("err" ))
288
301
mdFQDN .On ("HostArch" ).Return ("amd64" , nil )
289
- mdFQDN .On ("HostIPs" ).Return (testIPsAddresses , nil )
290
- mdFQDN .On ("HostMACs" ).Return (testMACsAddresses , nil )
302
+ mdFQDN .On ("HostIPs" ).Return (testIPs , nil )
303
+ mdFQDN .On ("HostMACs" ).Return (testMACs , nil )
291
304
292
305
detector := newTestDetector (mdFQDN , []string {"dns" }, allEnabledConfig ())
293
306
res , schemaURL , err := detector .Detect (context .Background ())
@@ -303,8 +316,8 @@ func TestDetectError(t *testing.T) {
303
316
mdHostname .On ("Hostname" ).Return ("" , errors .New ("err" ))
304
317
mdHostname .On ("HostID" ).Return ("" , errors .New ("err" ))
305
318
mdHostname .On ("HostArch" ).Return ("amd64" , nil )
306
- mdHostname .On ("HostIPs" ).Return (testIPsAddresses , nil )
307
- mdHostname .On ("HostMACs" ).Return (testMACsAddresses , nil )
319
+ mdHostname .On ("HostIPs" ).Return (testIPs , nil )
320
+ mdHostname .On ("HostMACs" ).Return (testMACs , nil )
308
321
309
322
detector = newTestDetector (mdHostname , []string {"os" }, allEnabledConfig ())
310
323
res , schemaURL , err = detector .Detect (context .Background ())
@@ -320,7 +333,7 @@ func TestDetectError(t *testing.T) {
320
333
mdOSType .On ("OSVersion" ).Return ("" , "22.04.2 LTS (Jammy Jellyfish)" )
321
334
mdOSType .On ("HostID" ).Return ("1" , nil )
322
335
mdOSType .On ("HostArch" ).Return ("amd64" , nil )
323
- mdOSType .On ("HostIPs" ).Return (testIPsAddresses , nil )
336
+ mdOSType .On ("HostIPs" ).Return (testIPs , nil )
324
337
325
338
detector = newTestDetector (mdOSType , []string {"os" }, allEnabledConfig ())
326
339
res , schemaURL , err = detector .Detect (context .Background ())
@@ -336,7 +349,7 @@ func TestDetectError(t *testing.T) {
336
349
mdOSVersion .On ("OSVersion" ).Return ("" , errors .New ("err" ))
337
350
mdOSVersion .On ("HostID" ).Return ("1" , nil )
338
351
mdOSVersion .On ("HostArch" ).Return ("amd64" , nil )
339
- mdOSVersion .On ("HostIPs" ).Return (testIPsAddresses , nil )
352
+ mdOSVersion .On ("HostIPs" ).Return (testIPs , nil )
340
353
341
354
detector = newTestDetector (mdOSVersion , []string {"os" }, allEnabledConfig ())
342
355
res , schemaURL , err = detector .Detect (context .Background ())
@@ -352,8 +365,8 @@ func TestDetectError(t *testing.T) {
352
365
mdHostID .On ("OSVersion" ).Return ("22.04.2 LTS (Jammy Jellyfish)" , nil )
353
366
mdHostID .On ("HostID" ).Return ("" , errors .New ("err" ))
354
367
mdHostID .On ("HostArch" ).Return ("arm64" , nil )
355
- mdHostID .On ("HostIPs" ).Return (testIPsAddresses , nil )
356
- mdHostID .On ("HostMACs" ).Return (testMACsAddresses , nil )
368
+ mdHostID .On ("HostIPs" ).Return (testIPs , nil )
369
+ mdHostID .On ("HostMACs" ).Return (testMACs , nil )
357
370
358
371
detector = newTestDetector (mdHostID , []string {"os" }, allEnabledConfig ())
359
372
res , schemaURL , err = detector .Detect (context .Background ())
@@ -378,8 +391,8 @@ func TestDetectCPUInfo(t *testing.T) {
378
391
md .On ("OSVersion" ).Return ("22.04.2 LTS (Jammy Jellyfish)" , nil )
379
392
md .On ("HostID" ).Return ("2" , nil )
380
393
md .On ("HostArch" ).Return ("amd64" , nil )
381
- md .On ("HostIPs" ).Return (testIPsAddresses , nil )
382
- md .On ("HostMACs" ).Return (testMACsAddresses , nil )
394
+ md .On ("HostIPs" ).Return (testIPs , nil )
395
+ md .On ("HostMACs" ).Return (testMACs , nil )
383
396
md .On ("CPUInfo" ).Return ([]cpu.InfoStat {{Family : "some" }}, nil )
384
397
385
398
cfg := allEnabledConfig ()
0 commit comments