1 files changed, 82 insertions, 0 deletions
diff --git a/tests/unit/packet/test_vxlan.py b/tests/unit/packet/test_vxlan.py
new file mode 100644
index 00000000..8cace581
--- /dev/null
+++ b/tests/unit/packet/test_vxlan.py
@@ -0,0 +1,82 @@
+# Copyright (C) 2016 Nippon Telegraph and Telephone Corporation.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#    http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+# implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import logging
+import unittest
+
+from nose.tools import eq_
+from nose.tools import raises
+
+from ryu.lib.packet import ethernet
+from ryu.lib.packet import vxlan
+
+
+LOG = logging.getLogger(__name__)
+
+
+class Test_vxlan(unittest.TestCase):
+    """
+    Test case for VXLAN (RFC 7348) header encoder/decoder class.
+    """
+
+    vni = 0x123456
+    buf = (
+        b'\x08\x00\x00\x00'  # flags = R|R|R|R|I|R|R|R (8 bits)
+        b'\x12\x34\x56\x00'  # vni = 0x123456 (24 bits)
+        b'test_payload'      # for test
+    )
+    pkt = vxlan.vxlan(vni)
+    jsondict = {
+        'vxlan': {
+            'vni': vni
+        }
+    }
+
+    def test_init(self):
+        eq_(self.vni, self.pkt.vni)
+
+    def test_parser(self):
+        parsed_pkt, next_proto_cls, rest_buf = vxlan.vxlan.parser(self.buf)
+        eq_(self.vni, parsed_pkt.vni)
+        eq_(ethernet.ethernet, next_proto_cls)
+        eq_(b'test_payload', rest_buf)
+
+    @raises(AssertionError)
+    def test_invalid_flags(self):
+        invalid_flags_bug = (
+            b'\x00\x00\x00\x00'  # all bits are set to zero
+            b'\x12\x34\x56\x00'  # vni = 0x123456 (24 bits)
+        )
+        vxlan.vxlan.parser(invalid_flags_bug)
+
+    def test_serialize(self):
+        serialized_buf = self.pkt.serialize(payload=None, prev=None)
+        eq_(self.buf[:vxlan.vxlan._MIN_LEN], serialized_buf)
+
+    def test_from_jsondict(self):
+        pkt_from_json = vxlan.vxlan.from_jsondict(
+            self.jsondict[vxlan.vxlan.__name__])
+        eq_(self.vni, pkt_from_json.vni)
+
+    def test_to_jsondict(self):
+        jsondict_from_pkt = self.pkt.to_jsondict()
+        eq_(self.jsondict, jsondict_from_pkt)
+
+    def test_vni_from_bin(self):
+        vni = vxlan.vni_from_bin(b'\x12\x34\x56')
+        eq_(self.vni, vni)
+
+    def test_vni_to_bin(self):
+        eq_(b'\x12\x34\x56', vxlan.vni_to_bin(self.vni))