Hulk did something

[bytom/vapor.git] / vendor / golang.org / x / net / ipv4 / icmp_test.go

diff --git a/vendor/golang.org/x/net/ipv4/icmp_test.go b/vendor/golang.org/x/net/ipv4/icmp_test.go
new file mode 100644 (file)
index 0000000..3324b54
--- /dev/null
+++ b/vendor/golang.org/x/net/ipv4/icmp_test.go
@@ -0,0 +1,95 @@
+// Copyright 2014 The Go Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style
+// license that can be found in the LICENSE file.
+
+package ipv4_test
+
+import (
+       "net"
+       "reflect"
+       "runtime"
+       "testing"
+
+       "golang.org/x/net/internal/nettest"
+       "golang.org/x/net/ipv4"
+)
+
+var icmpStringTests = []struct {
+       in  ipv4.ICMPType
+       out string
+}{
+       {ipv4.ICMPTypeDestinationUnreachable, "destination unreachable"},
+
+       {256, "<nil>"},
+}
+
+func TestICMPString(t *testing.T) {
+       for _, tt := range icmpStringTests {
+               s := tt.in.String()
+               if s != tt.out {
+                       t.Errorf("got %s; want %s", s, tt.out)
+               }
+       }
+}
+
+func TestICMPFilter(t *testing.T) {
+       switch runtime.GOOS {
+       case "linux":
+       default:
+               t.Skipf("not supported on %s", runtime.GOOS)
+       }
+
+       var f ipv4.ICMPFilter
+       for _, toggle := range []bool{false, true} {
+               f.SetAll(toggle)
+               for _, typ := range []ipv4.ICMPType{
+                       ipv4.ICMPTypeDestinationUnreachable,
+                       ipv4.ICMPTypeEchoReply,
+                       ipv4.ICMPTypeTimeExceeded,
+                       ipv4.ICMPTypeParameterProblem,
+               } {
+                       f.Accept(typ)
+                       if f.WillBlock(typ) {
+                               t.Errorf("ipv4.ICMPFilter.Set(%v, false) failed", typ)
+                       }
+                       f.Block(typ)
+                       if !f.WillBlock(typ) {
+                               t.Errorf("ipv4.ICMPFilter.Set(%v, true) failed", typ)
+                       }
+               }
+       }
+}
+
+func TestSetICMPFilter(t *testing.T) {
+       switch runtime.GOOS {
+       case "linux":
+       default:
+               t.Skipf("not supported on %s", runtime.GOOS)
+       }
+       if m, ok := nettest.SupportsRawIPSocket(); !ok {
+               t.Skip(m)
+       }
+
+       c, err := net.ListenPacket("ip4:icmp", "127.0.0.1")
+       if err != nil {
+               t.Fatal(err)
+       }
+       defer c.Close()
+
+       p := ipv4.NewPacketConn(c)
+
+       var f ipv4.ICMPFilter
+       f.SetAll(true)
+       f.Accept(ipv4.ICMPTypeEcho)
+       f.Accept(ipv4.ICMPTypeEchoReply)
+       if err := p.SetICMPFilter(&f); err != nil {
+               t.Fatal(err)
+       }
+       kf, err := p.ICMPFilter()
+       if err != nil {
+               t.Fatal(err)
+       }
+       if !reflect.DeepEqual(kf, &f) {
+               t.Fatalf("got %#v; want %#v", kf, f)
+       }
+}