Signed-off-by: IWASE Yusuke <iwase.yusu...@gmail.com>
---
ryu/tests/unit/lib/test_mrtlib.py | 61 +++++++++++++++++++++++++++++++++++++++
1 file changed, 61 insertions(+)
diff --git a/ryu/tests/unit/lib/test_mrtlib.py
b/ryu/tests/unit/lib/test_mrtlib.py
index e870f40..55c7cda 100644
--- a/ryu/tests/unit/lib/test_mrtlib.py
+++ b/ryu/tests/unit/lib/test_mrtlib.py
@@ -19,6 +19,7 @@ import bz2
import io
import logging
import os
+import struct
import sys
import unittest
@@ -609,6 +610,66 @@ class TestMrtlibMrtPeer(unittest.TestCase):
eq_(buf, output)
+class TestMrtlibMrtRibEntry(unittest.TestCase):
+ """
+ Test case for ryu.lib.mrtlib.MrtRibEntry.
+ """
+
+ def test_parse_add_path(self):
+ peer_index = 1
+ originated_time = 2
+ nexthop = '1.1.1.1'
+ bgp_attribute = bgp.BGPPathAttributeNextHop(nexthop)
+ path_id = 3
+ bgp_attr_buf = bgp_attribute.serialize()
+ attr_len = len(bgp_attr_buf)
+ buf = (
+ b'\x00\x01' # peer_index
+ b'\x00\x00\x00\x02' # originated_time
+ b'\x00\x00\x00\x03' # path_id
+ + struct.pack('!H', attr_len) # attr_len
+ + bgp_attribute.serialize() # bgp_attributes
+ )
+
+ rib, rest = mrtlib.MrtRibEntry.parse(buf, is_addpath=True)
+
+ eq_(peer_index, rib.peer_index)
+ eq_(originated_time, rib.originated_time)
+ eq_(path_id, rib.path_id)
+ eq_(attr_len, rib.attr_len)
+ eq_(1, len(rib.bgp_attributes))
+ eq_(nexthop, rib.bgp_attributes[0].value)
+ eq_(b'', rest)
+
+ def test_serialize_add_path(self):
+ peer_index = 1
+ originated_time = 2
+ nexthop = '1.1.1.1'
+ bgp_attribute = bgp.BGPPathAttributeNextHop(nexthop)
+ path_id = 3
+ bgp_attr_buf = bgp_attribute.serialize()
+ attr_len = len(bgp_attr_buf)
+ buf = (
+ b'\x00\x01' # peer_index
+ b'\x00\x00\x00\x02' # originated_time
+ b'\x00\x00\x00\x03' # path_id
+ + struct.pack('!H', attr_len) # attr_len
+ + bgp_attribute.serialize() # bgp_attributes
+ )
+
+ rib = mrtlib.MrtRibEntry(
+ peer_index=peer_index,
+ originated_time=originated_time,
+ bgp_attributes=[bgp_attribute],
+ # attr_len=attr_len,
+ path_id=path_id,
+ )
+
+ output = rib.serialize()
+
+ eq_(buf, output)
+
+
class TestMrtlibBgp4MpMrtRecord(unittest.TestCase):
"""
Test case for ryu.lib.mrtlib.Bgp4MpMrtRecord.
--
2.7.4
------------------------------------------------------------------------------
Check out the vibrant tech community on one of the world's most
engaging tech sites, Slashdot.org! http://sdm.link/slashdot
_______________________________________________
Ryu-devel mailing list
Ryu-devel@lists.sourceforge.net
https://lists.sourceforge.net/lists/listinfo/ryu-devel
