test/lib/base: Avoid to del all routes without identifier specified

Currently, del_route() with "identifier=None" deletes all routes even if
some installed routes have the valid identifier.
This patch fixes to delete only route which exactly matches against the
given identifier.

Signed-off-by: IWASE Yusuke <iwase.yusuke0@gmail.com>

1 files changed, 1 insertions, 8 deletions

diff --git a/test/lib/base.py b/test/lib/base.py
index 161bb9eb..7718de93 100644
--- a/test/lib/base.py
+++ b/test/lib/base.py
@@ -403,14 +403,7 @@ class BGPContainer(Container):
     def del_route(self, route, identifier=None, reload_config=True):
         if route not in self.routes:
             return
-        if identifier:
-            new_paths = []
-            for path in self.routes[route]:
-                if path['identifier'] != identifier:
-                    new_paths.append(path)
-            self.routes[route] = new_paths
-        else:
-            self.routes[route] = []
+        self.routes[route] = [p for p in self.routes[route] if p['identifier'] != identifier]
         if self.is_running and reload_config:
             self.create_config()
             self.reload_config()