ryu/build/lib.linux-armv7l-2.7/ryu/tests/unit/packet/test_gre.py

   1 # Copyright (C) 2016 Nippon Telegraph and Telephone Corporation.
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at
   6 #
   7 #    http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
  12 # implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15
  16 from __future__ import print_function
  17
  18 import logging
  19 import os
  20 import sys
  21
  22 import unittest
  23 from nose.tools import eq_
  24 from nose.tools import ok_
  25
  26 from ryu.lib import pcaplib
  27 from ryu.lib.packet import gre
  28 from ryu.lib.packet import packet
  29 from ryu.utils import binary_str
  30 from ryu.lib.packet.ether_types import ETH_TYPE_IP, ETH_TYPE_TEB
  31
  32 LOG = logging.getLogger(__name__)
  33
  34 GENEVE_DATA_DIR = os.path.join(
  35     os.path.dirname(sys.modules[__name__].__file__),
  36     '../../packet_data/pcap/')
  37
  38
  39 class Test_gre(unittest.TestCase):
  40     """
  41     Test case gre for ryu.lib.packet.gre.
  42     """
  43
  44     version = 0
  45     gre_proto = ETH_TYPE_IP
  46     nvgre_proto = ETH_TYPE_TEB
  47     checksum = 0x440d
  48     seq_number = 10
  49     key = 256100
  50     vsid = 1000
  51     flow_id = 100
  52
  53     gre = gre.gre(version=version, protocol=gre_proto, checksum=checksum,
  54                   key=key, seq_number=seq_number)
  55
  56     def test_key_setter(self):
  57         self.gre.key = self.key
  58         eq_(self.gre._key, self.key)
  59         eq_(self.gre._vsid, self.vsid)
  60         eq_(self.gre._flow_id, self.flow_id)
  61
  62     def test_key_setter_none(self):
  63         self.gre.key = None
  64         eq_(self.gre._key, None)
  65         eq_(self.gre._vsid, None)
  66         eq_(self.gre._flow_id, None)
  67
  68         self.gre.key = self.key
  69
  70     def test_vsid_setter(self):
  71         self.gre.vsid = self.vsid
  72         eq_(self.gre._key, self.key)
  73         eq_(self.gre._vsid, self.vsid)
  74         eq_(self.gre._flow_id, self.flow_id)
  75
  76     def test_flowid_setter(self):
  77         self.gre.flow_id = self.flow_id
  78         eq_(self.gre._key, self.key)
  79         eq_(self.gre._vsid, self.vsid)
  80         eq_(self.gre._flow_id, self.flow_id)
  81
  82     def test_nvgre_init(self):
  83         nvgre = gre.nvgre(version=self.version, vsid=self.vsid,
  84                           flow_id=self.flow_id)
  85
  86         eq_(nvgre.version, self.version)
  87         eq_(nvgre.protocol, self.nvgre_proto)
  88         eq_(nvgre.checksum, None)
  89         eq_(nvgre.seq_number, None)
  90         eq_(nvgre._key, self.key)
  91         eq_(nvgre._vsid, self.vsid)
  92         eq_(nvgre._flow_id, self.flow_id)
  93
  94     def test_parser(self):
  95         files = [
  96             'gre_full_options',
  97             'gre_no_option',
  98             'gre_nvgre_option',
  99         ]
 100
 101         for f in files:
 102             # print('*** testing %s ...' % f)
 103             for _, buf in pcaplib.Reader(
 104                     open(GENEVE_DATA_DIR + f + '.pcap', 'rb')):
 105                 # Checks if message can be parsed as expected.
 106                 pkt = packet.Packet(buf)
 107                 gre_pkt = pkt.get_protocol(gre.gre)
 108                 ok_(isinstance(gre_pkt, gre.gre),
 109                     'Failed to parse Gre message: %s' % pkt)
 110
 111                 # Checks if message can be serialized as expected.
 112                 pkt.serialize()
 113
 114                 eq_(buf, pkt.data,
 115                     "b'%s' != b'%s'" % (binary_str(buf), binary_str(pkt.data)))