128 lines
2.9 KiB
Go
128 lines
2.9 KiB
Go
//go:build linux
|
|
// +build linux
|
|
|
|
package aghnet
|
|
|
|
import (
|
|
"io/fs"
|
|
"testing"
|
|
"testing/fstest"
|
|
|
|
"github.com/AdguardTeam/golibs/testutil"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestHasStaticIP(t *testing.T) {
|
|
const ifaceName = "wlan0"
|
|
|
|
const (
|
|
dhcpcd = "etc/dhcpcd.conf"
|
|
netifaces = "etc/network/interfaces"
|
|
)
|
|
|
|
testCases := []struct {
|
|
rootFsys fs.FS
|
|
name string
|
|
wantHas assert.BoolAssertionFunc
|
|
wantErrMsg string
|
|
}{{
|
|
rootFsys: fstest.MapFS{
|
|
dhcpcd: &fstest.MapFile{
|
|
Data: []byte(`#comment` + nl +
|
|
`# comment` + nl +
|
|
`interface eth0` + nl +
|
|
`static ip_address=192.168.0.1/24` + nl +
|
|
`# interface ` + ifaceName + nl +
|
|
`static ip_address=192.168.1.1/24` + nl +
|
|
`# comment` + nl,
|
|
),
|
|
},
|
|
},
|
|
name: "dhcpcd_has_not",
|
|
wantHas: assert.False,
|
|
wantErrMsg: `no information about static ip`,
|
|
}, {
|
|
rootFsys: fstest.MapFS{
|
|
dhcpcd: &fstest.MapFile{
|
|
Data: []byte(`#comment` + nl +
|
|
`# comment` + nl +
|
|
`interface ` + ifaceName + nl +
|
|
`static ip_address=192.168.0.1/24` + nl +
|
|
`# interface ` + ifaceName + nl +
|
|
`static ip_address=192.168.1.1/24` + nl +
|
|
`# comment` + nl,
|
|
),
|
|
},
|
|
},
|
|
name: "dhcpcd_has",
|
|
wantHas: assert.True,
|
|
wantErrMsg: ``,
|
|
}, {
|
|
rootFsys: fstest.MapFS{
|
|
netifaces: &fstest.MapFile{
|
|
Data: []byte(`allow-hotplug ` + ifaceName + nl +
|
|
`#iface enp0s3 inet static` + nl +
|
|
`# address 192.168.0.200` + nl +
|
|
`# netmask 255.255.255.0` + nl +
|
|
`# gateway 192.168.0.1` + nl +
|
|
`iface ` + ifaceName + ` inet dhcp` + nl,
|
|
),
|
|
},
|
|
},
|
|
name: "netifaces_has_not",
|
|
wantHas: assert.False,
|
|
wantErrMsg: `no information about static ip`,
|
|
}, {
|
|
rootFsys: fstest.MapFS{
|
|
netifaces: &fstest.MapFile{
|
|
Data: []byte(`allow-hotplug ` + ifaceName + nl +
|
|
`iface ` + ifaceName + ` inet static` + nl +
|
|
` address 192.168.0.200` + nl +
|
|
` netmask 255.255.255.0` + nl +
|
|
` gateway 192.168.0.1` + nl +
|
|
`#iface ` + ifaceName + ` inet dhcp` + nl,
|
|
),
|
|
},
|
|
},
|
|
name: "netifaces_has",
|
|
wantHas: assert.True,
|
|
wantErrMsg: ``,
|
|
}, {
|
|
rootFsys: fstest.MapFS{
|
|
netifaces: &fstest.MapFile{
|
|
Data: []byte(`source hello` + nl +
|
|
`#iface ` + ifaceName + ` inet static` + nl,
|
|
),
|
|
},
|
|
"hello": &fstest.MapFile{
|
|
Data: []byte(`iface ` + ifaceName + ` inet static` + nl),
|
|
},
|
|
},
|
|
name: "netifaces_another_file",
|
|
wantHas: assert.True,
|
|
wantErrMsg: ``,
|
|
}, {
|
|
rootFsys: fstest.MapFS{
|
|
netifaces: &fstest.MapFile{
|
|
Data: []byte(`source hello` + nl +
|
|
`iface ` + ifaceName + ` inet static` + nl,
|
|
),
|
|
},
|
|
},
|
|
name: "netifaces_ignore_another",
|
|
wantHas: assert.True,
|
|
wantErrMsg: ``,
|
|
}}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
substRootDirFS(t, tc.rootFsys)
|
|
|
|
has, err := IfaceHasStaticIP(ifaceName)
|
|
testutil.AssertErrorMsg(t, tc.wantErrMsg, err)
|
|
|
|
tc.wantHas(t, has)
|
|
})
|
|
}
|
|
}
|