server: fix advertising multiple local withdrawals with same prefix

a bug introduced by 332766189685028c4f9852e4285fb1a9025223cc

Signed-off-by: ISHIDA Wataru <ishida.wataru@lab.ntt.co.jp>

3 files changed, 48 insertions, 6 deletions

diff --git a/test/lib/base.py b/test/lib/base.py
index 4f2e8a69..38cac97a 100644
--- a/test/lib/base.py
+++ b/test/lib/base.py
@@ -158,6 +158,7 @@ class Container(object):
         self.ip6_addrs = []
         self.is_running = False
         self.eths = []
+        self.tcpdump_running = False
 
         if self.docker_name() in get_containers():
             self.remove()
@@ -230,12 +231,21 @@ class Container(object):
             return int(local(cmd, capture=True))
         return -1
 
-    def start_tcpdump(self, interface=None, filename=None):
+    def start_tcpdump(self, interface=None, filename=None, expr='tcp port 179'):
+        if self.tcpdump_running:
+            raise Exception('tcpdump already running')
+        self.tcpdump_running = True
         if not interface:
             interface = "eth0"
         if not filename:
-            filename = "{0}/{1}.dump".format(self.shared_volumes[0][1], interface)
-        self.local("tcpdump -i {0} -w {1}".format(interface, filename), detach=True)
+            filename = '{0}.dump'.format(interface)
+        self.local("tcpdump -i {0} -w {1}/{2} {3}".format(interface, self.shared_volumes[0][1], filename, expr), detach=True)
+        return '{0}/{1}'.format(self.shared_volumes[0][0], filename)
+
+    def stop_tcpdump(self):
+        self.local("pkill tcpdump")
+        self.tcpdump_running = False
+
 
 class BGPContainer(Container):
 
diff --git a/test/pip-requires.txt b/test/pip-requires.txt
index 708597d8..e7e580ab 100644
--- a/test/pip-requires.txt
+++ b/test/pip-requires.txt
@@ -1,10 +1,8 @@
 nose
 toml
 pyyaml
-ciscoconfparse
-ecdsa
-pycrypto>=2.1
 fabric
 netaddr
 nsenter
 docker-py
+ryu
diff --git a/test/scenario_test/bgp_router_test.py b/test/scenario_test/bgp_router_test.py
index a528c36d..4f7e981d 100644
--- a/test/scenario_test/bgp_router_test.py
+++ b/test/scenario_test/bgp_router_test.py
@@ -25,6 +25,9 @@ import time
 import nose
 from noseplugin import OptionParser, parser_option
 from itertools import chain
+import ryu.lib.pcaplib as pcap
+from ryu.lib.packet.packet import Packet
+from ryu.lib.packet.bgp import BGPMessage
 
 
 class GoBGPTestBase(unittest.TestCase):
@@ -354,6 +357,37 @@ class GoBGPTestBase(unittest.TestCase):
 
         g1.wait_for(expected_state=BGP_FSM_ESTABLISHED, peer=e1)
 
+    def test_20_check_withdrawal_2(self):
+        g1 = self.gobgp
+        g2 = self.quaggas['g2']
+
+        dumpfile = g2.start_tcpdump()
+
+        g1.add_route('10.40.0.0/24')
+
+        time.sleep(1)
+
+        paths = g2.get_global_rib('10.40.0.0/24')
+        self.assertTrue(len(paths) == 1)
+
+        g1.local('gobgp global rib del 10.40.0.0/24')
+
+        time.sleep(1)
+
+        paths = g2.get_global_rib('10.40.0.0/24')
+        self.assertTrue(len(paths) == 0)
+
+        g2.stop_tcpdump()
+        time.sleep(1)
+
+        cnt = 0
+        for pkt in pcap.Reader(open(dumpfile)):
+            last = Packet(pkt[1]).protocols[-1]
+            if type(last) == str:
+                pkt = BGPMessage.parser(last)[0]
+                cnt += len(pkt.withdrawn_routes)
+
+        self.assertTrue(cnt == 1)
 
 
 if __name__ == '__main__':