fumiama/water
1
0
Fork 0
mirror of https://github.com/fumiama/water.git synced 2026-06-30 08:50:26 +08:00
Code Activity

tear down device at end of test

Browse Source
This commit is contained in:
Song Gao
2019-03-31 15:33:00 -07:00
parent af2921498b
commit 240a3d7b51
2 changed files with 20 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
ipv4_darwin_test.go
View File

@@ -21,6 +21,15 @@ func setupIfce(t *testing.T, self net.IP, remote net.IP, dev string) {
} }
} }
func teardownIfce(t *testing.T, ifce *Interface) {
if err := ifce.Close(); err != nil {
t.Fatal(err)
}
if err := exec.Command("ifconfig", ifce.Name(), "down").Run(); err != nil {
t.Fatal(err)
}
}
func TestP2PTUN(t *testing.T) { func TestP2PTUN(t *testing.T) {
var ( var (
self = net.IPv4(10, 0, 42, 1) self = net.IPv4(10, 0, 42, 1)
@@ -31,6 +40,7 @@ func TestP2PTUN(t *testing.T) {
if err != nil { if err != nil {
t.Fatalf("creating TUN error: %v\n", err) t.Fatalf("creating TUN error: %v\n", err)
} }
defer teardownIfce(t, ifce)
dataCh := make(chan []byte) dataCh := make(chan []byte)
errCh := make(chan error) errCh := make(chan error)

10
ipv4_linux_test.go
View File

@@ -24,6 +24,15 @@ func setupIfce(t *testing.T, ipNet net.IPNet, dev string) {
} }
} }
func teardownIfce(t *testing.T, ifce *Interface) {
if err := ifce.Close(); err != nil {
t.Fatal(err)
}
if err := exec.Command("ip", "link", "set", ifce.Name(), "down").Run(); err != nil {
t.Fatal(err)
}
}
func TestBroadcastTAP(t *testing.T) { func TestBroadcastTAP(t *testing.T) {
var ( var (
self = net.IPv4(10, 0, 42, 1) self = net.IPv4(10, 0, 42, 1)
@@ -35,6 +44,7 @@ func TestBroadcastTAP(t *testing.T) {
if err != nil { if err != nil {
t.Fatalf("creating TAP error: %v\n", err) t.Fatalf("creating TAP error: %v\n", err)
} }
defer teardownIfce(t, ifce)
setupIfce(t, net.IPNet{IP: self, Mask: mask}, ifce.Name()) setupIfce(t, net.IPNet{IP: self, Mask: mask}, ifce.Name())
startBroadcast(t, brd) startBroadcast(t, brd)