mirrors/ryu
1
0
Fork 0
mirror of https://github.com/faucetsdn/ryu.git synced 2026-01-22 00:52:04 +01:00
Code Issues Projects Releases Wiki Activity
ryu/ryu/tests/unit/ofproto/test_parser.py
Satoshi Kobayashi cfdba5c12d Fix ImportError(s) with Python3 
Signed-off-by: FUJITA Tomonori <fujita.tomonori@lab.ntt.co.jp>
2015-04-14 05:19:56 +09:00

263 lines
10 KiB
Python
Raw Blame History

#!/usr/bin/env python
#
# Copyright (C) 2013,2014,2015 Nippon Telegraph and Telephone Corporation.
# Copyright (C) 2013,2014,2015 YAMAMOTO Takashi <yamamoto at valinux co jp>
#
# 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 sys
import unittest
from nose.tools import eq_
from ryu.ofproto import ofproto_parser
from ryu.ofproto import ofproto_protocol
from ryu.ofproto import ofproto_v1_0
from ryu.ofproto import ofproto_v1_2
from ryu.ofproto import ofproto_v1_3
from ryu.ofproto import ofproto_v1_4
from ryu.ofproto import ofproto_v1_5
import json
# (has_parser, has_serializer)
implemented = {
1: {
ofproto_v1_0.OFPT_PACKET_OUT: (False, True),
ofproto_v1_0.OFPT_FEATURES_REQUEST: (False, True),
ofproto_v1_0.OFPT_FEATURES_REPLY: (True, False),
ofproto_v1_0.OFPT_PACKET_IN: (True, False),
ofproto_v1_0.OFPT_FLOW_MOD: (False, True),
},
3: {
ofproto_v1_2.OFPT_FEATURES_REQUEST: (False, True),
ofproto_v1_2.OFPT_FEATURES_REPLY: (True, False),
ofproto_v1_2.OFPT_GET_CONFIG_REQUEST: (False, True),
ofproto_v1_2.OFPT_GET_CONFIG_REPLY: (True, False),
ofproto_v1_2.OFPT_SET_CONFIG: (False, True),
ofproto_v1_2.OFPT_PACKET_IN: (True, False),
ofproto_v1_2.OFPT_FLOW_REMOVED: (True, False),
ofproto_v1_2.OFPT_PORT_STATUS: (True, False),
ofproto_v1_2.OFPT_PACKET_OUT: (False, True),
ofproto_v1_2.OFPT_FLOW_MOD: (False, True),
ofproto_v1_2.OFPT_GROUP_MOD: (False, True),
ofproto_v1_2.OFPT_PORT_MOD: (False, True),
ofproto_v1_2.OFPT_TABLE_MOD: (False, True),
ofproto_v1_2.OFPT_STATS_REQUEST: (False, True),
ofproto_v1_2.OFPT_STATS_REPLY: (True, False),
ofproto_v1_2.OFPT_BARRIER_REQUEST: (False, True),
ofproto_v1_2.OFPT_QUEUE_GET_CONFIG_REQUEST: (False, True),
ofproto_v1_2.OFPT_QUEUE_GET_CONFIG_REPLY: (True, False),
ofproto_v1_2.OFPT_ROLE_REQUEST: (False, True),
ofproto_v1_2.OFPT_ROLE_REPLY: (True, False),
},
4: {
ofproto_v1_3.OFPT_HELLO: (True, False),
ofproto_v1_3.OFPT_FEATURES_REQUEST: (False, True),
ofproto_v1_3.OFPT_FEATURES_REPLY: (True, False),
ofproto_v1_3.OFPT_GET_CONFIG_REQUEST: (False, True),
ofproto_v1_3.OFPT_GET_CONFIG_REPLY: (True, False),
ofproto_v1_3.OFPT_SET_CONFIG: (False, True),
ofproto_v1_3.OFPT_PACKET_IN: (True, False),
ofproto_v1_3.OFPT_FLOW_REMOVED: (True, False),
ofproto_v1_3.OFPT_PORT_STATUS: (True, False),
ofproto_v1_3.OFPT_PACKET_OUT: (False, True),
ofproto_v1_3.OFPT_FLOW_MOD: (False, True),
ofproto_v1_3.OFPT_GROUP_MOD: (False, True),
ofproto_v1_3.OFPT_PORT_MOD: (False, True),
ofproto_v1_3.OFPT_METER_MOD: (False, True),
ofproto_v1_3.OFPT_TABLE_MOD: (False, True),
ofproto_v1_3.OFPT_MULTIPART_REQUEST: (False, True),
ofproto_v1_3.OFPT_MULTIPART_REPLY: (True, False),
ofproto_v1_3.OFPT_BARRIER_REQUEST: (False, True),
ofproto_v1_3.OFPT_QUEUE_GET_CONFIG_REQUEST: (False, True),
ofproto_v1_3.OFPT_QUEUE_GET_CONFIG_REPLY: (True, False),
ofproto_v1_3.OFPT_ROLE_REQUEST: (False, True),
ofproto_v1_3.OFPT_ROLE_REPLY: (True, False),
ofproto_v1_3.OFPT_GET_ASYNC_REQUEST: (False, True),
ofproto_v1_3.OFPT_GET_ASYNC_REPLY: (True, False),
ofproto_v1_3.OFPT_SET_ASYNC: (False, True),
},
5: {
ofproto_v1_4.OFPT_HELLO: (True, False),
ofproto_v1_4.OFPT_FEATURES_REQUEST: (False, True),
ofproto_v1_4.OFPT_FEATURES_REPLY: (True, False),
ofproto_v1_4.OFPT_GET_CONFIG_REQUEST: (False, True),
ofproto_v1_4.OFPT_GET_CONFIG_REPLY: (True, False),
ofproto_v1_4.OFPT_SET_CONFIG: (False, True),
ofproto_v1_4.OFPT_PACKET_IN: (True, False),
ofproto_v1_4.OFPT_FLOW_REMOVED: (True, False),
ofproto_v1_4.OFPT_PORT_STATUS: (True, False),
ofproto_v1_4.OFPT_PACKET_OUT: (False, True),
ofproto_v1_4.OFPT_FLOW_MOD: (False, True),
ofproto_v1_4.OFPT_GROUP_MOD: (False, True),
ofproto_v1_4.OFPT_PORT_MOD: (False, True),
ofproto_v1_4.OFPT_METER_MOD: (False, True),
ofproto_v1_4.OFPT_TABLE_MOD: (False, True),
ofproto_v1_4.OFPT_MULTIPART_REQUEST: (False, True),
ofproto_v1_4.OFPT_MULTIPART_REPLY: (True, False),
ofproto_v1_4.OFPT_BARRIER_REQUEST: (False, True),
ofproto_v1_4.OFPT_ROLE_REQUEST: (False, True),
ofproto_v1_4.OFPT_ROLE_REPLY: (True, False),
ofproto_v1_4.OFPT_GET_ASYNC_REQUEST: (False, True),
ofproto_v1_4.OFPT_GET_ASYNC_REPLY: (True, False),
ofproto_v1_4.OFPT_SET_ASYNC: (False, True),
ofproto_v1_4.OFPT_ROLE_STATUS: (True, False),
ofproto_v1_4.OFPT_TABLE_STATUS: (True, False),
ofproto_v1_4.OFPT_REQUESTFORWARD: (False, True),
ofproto_v1_4.OFPT_BUNDLE_CONTROL: (False, True),
ofproto_v1_4.OFPT_BUNDLE_ADD_MESSAGE: (False, True),
},
6: {
ofproto_v1_5.OFPT_HELLO: (True, False),
ofproto_v1_5.OFPT_FEATURES_REQUEST: (False, True),
ofproto_v1_5.OFPT_FEATURES_REPLY: (True, False),
ofproto_v1_5.OFPT_GET_CONFIG_REQUEST: (False, True),
ofproto_v1_5.OFPT_GET_CONFIG_REPLY: (True, False),
ofproto_v1_5.OFPT_SET_CONFIG: (False, True),
ofproto_v1_5.OFPT_PACKET_IN: (True, False),
ofproto_v1_5.OFPT_FLOW_REMOVED: (True, False),
ofproto_v1_5.OFPT_PORT_STATUS: (True, False),
ofproto_v1_5.OFPT_PACKET_OUT: (False, True),
ofproto_v1_5.OFPT_FLOW_MOD: (False, True),
ofproto_v1_5.OFPT_GROUP_MOD: (False, True),
ofproto_v1_5.OFPT_PORT_MOD: (False, True),
ofproto_v1_5.OFPT_METER_MOD: (False, True),
ofproto_v1_5.OFPT_TABLE_MOD: (False, True),
ofproto_v1_5.OFPT_MULTIPART_REQUEST: (False, True),
ofproto_v1_5.OFPT_MULTIPART_REPLY: (True, False),
ofproto_v1_5.OFPT_BARRIER_REQUEST: (False, True),
ofproto_v1_5.OFPT_ROLE_REQUEST: (False, True),
ofproto_v1_5.OFPT_ROLE_REPLY: (True, False),
ofproto_v1_5.OFPT_GET_ASYNC_REQUEST: (False, True),
ofproto_v1_5.OFPT_GET_ASYNC_REPLY: (True, False),
ofproto_v1_5.OFPT_SET_ASYNC: (False, True),
ofproto_v1_5.OFPT_ROLE_STATUS: (True, False),
ofproto_v1_5.OFPT_TABLE_STATUS: (True, False),
ofproto_v1_5.OFPT_REQUESTFORWARD: (False, True),
ofproto_v1_5.OFPT_BUNDLE_CONTROL: (True, True),
ofproto_v1_5.OFPT_BUNDLE_ADD_MESSAGE: (False, True),
},
}
class Test_Parser(unittest.TestCase):
""" Test case for ryu.ofproto, especially json representation
"""
def __init__(self, methodName):
print('init %s' % methodName)
super(Test_Parser, self).__init__(methodName)
def setUp(self):
pass
def tearDown(self):
pass
@staticmethod
def _msg_to_jsondict(msg):
return msg.to_jsondict()
@staticmethod
def _jsondict_to_msg(dp, jsondict):
return ofproto_parser.ofp_msg_from_jsondict(dp, jsondict)
def _test_msg(self, name, wire_msg, json_str):
json_dict = json.loads(json_str)
# on-wire -> OFPxxx -> json
(version, msg_type, msg_len, xid) = ofproto_parser.header(wire_msg)
try:
has_parser, has_serializer = implemented[version][msg_type]
except KeyError:
has_parser = True
has_serializer = True
dp = ofproto_protocol.ProtocolDesc(version=version)
if has_parser:
msg = ofproto_parser.msg(dp, version, msg_type, msg_len, xid,
wire_msg)
json_dict2 = self._msg_to_jsondict(msg)
# XXXdebug code
open(('/tmp/%s.json' % name), 'wb').write(json.dumps(json_dict2))
eq_(json_dict, json_dict2)
# json -> OFPxxx -> json
msg2 = self._jsondict_to_msg(dp, json_dict)
if has_serializer:
msg2.serialize()
eq_(self._msg_to_jsondict(msg2), json_dict)
eq_(wire_msg, msg2.buf)
# check if "len" "length" fields can be omitted
def _remove(d, names):
f = lambda x: _remove(x, names)
if isinstance(d, list):
return map(f, d)
if isinstance(d, dict):
d2 = {}
for k, v in d.iteritems():
if k in names:
continue
d2[k] = f(v)
return d2
return d
json_dict3 = _remove(json_dict, ['len', 'length'])
msg3 = self._jsondict_to_msg(dp, json_dict3)
msg3.serialize()
eq_(wire_msg, msg3.buf)
msg2.serialize()
eq_(wire_msg, msg2.buf)
def _add_tests():
import os
import os.path
import fnmatch
import functools
this_dir = os.path.dirname(sys.modules[__name__].__file__)
packet_data_dir = os.path.join(this_dir, '../../packet_data')
json_dir = os.path.join(this_dir, 'json')
ofvers = [
'of10',
'of12',
'of13',
'of14',
'of15',
]
for ver in ofvers:
pdir = packet_data_dir + '/' + ver
jdir = json_dir + '/' + ver
for file in os.listdir(pdir):
if not fnmatch.fnmatch(file, '*.packet'):
continue
wire_msg = open(pdir + '/' + file, 'rb').read()
json_str = open(jdir + '/' + file + '.json', 'rb').read()
method_name = ('test_' + file).replace('-', '_').replace('.', '_')
def _run(self, name, wire_msg, json_str):
print('processing %s ...' % name)
self._test_msg(name, wire_msg, json_str)
print('adding %s ...' % method_name)
f = functools.partial(_run, name=method_name, wire_msg=wire_msg,
json_str=json_str)
f.func_name = method_name
f.__name__ = method_name
setattr(Test_Parser, method_name, f)
_add_tests()
Reference in New Issue View Git Blame Copy Permalink