1 files changed, 62 insertions, 0 deletions
diff --git a/vendor/github.com/google/gopacket/layers/dot1q_test.go b/vendor/github.com/google/gopacket/layers/dot1q_test.go
new file mode 100644
index 0000000..4a409c7
--- /dev/null
+++ b/vendor/github.com/google/gopacket/layers/dot1q_test.go
@@ -0,0 +1,62 @@
+// Copyright 2012 Google, Inc. All rights reserved.
+//
+// Use of this source code is governed by a BSD-style license
+// that can be found in the LICENSE file in the root of the source
+// tree.
+package layers
+
+import (
+	"fmt"
+	"reflect"
+	"testing"
+
+	"github.com/google/gopacket"
+)
+
+// test harness to ensure the dot1q layer can be encoded/decoded properly
+// return error if decoded data not match.
+func testEncodeDecodeDot1Q(dot1Q *Dot1Q) error {
+	buf := gopacket.NewSerializeBuffer()
+	opts := gopacket.SerializeOptions{}
+	expectedDot1Q := dot1Q
+
+	err := dot1Q.SerializeTo(buf, opts)
+	if err != nil {
+		return err
+	}
+
+	newDot1q := &Dot1Q{}
+	err = newDot1q.DecodeFromBytes(buf.Bytes(), gopacket.NilDecodeFeedback)
+	if err != nil {
+		return err
+	}
+	newDot1q.BaseLayer = BaseLayer{}
+
+	if !reflect.DeepEqual(expectedDot1Q, newDot1q) {
+		return fmt.Errorf("Expect %v actual %v", expectedDot1Q, newDot1q)
+	}
+	return nil
+
+}
+
+// Test to ensure what has been encode can be decoded
+func TestEncodeDecodeDot1Q(t *testing.T) {
+	dot1Qs := []*Dot1Q{
+		&Dot1Q{
+			Priority:       uint8(3),
+			VLANIdentifier: uint16(30),
+		},
+		&Dot1Q{
+			Priority:       uint8(0x07),
+			DropEligible:   true,
+			VLANIdentifier: uint16(0xFFF),
+		},
+	}
+
+	for i, curTest := range dot1Qs {
+		err := testEncodeDecodeDot1Q(curTest)
+		if err != nil {
+			t.Error("Error with item ", i, " with error message :", err)
+		}
+	}
+}