GENIUS-157: CSIT Test suite for OF-Tunnels Phase-1

[integration/test.git] / csit / libraries / BgpRpcClient.py

diff --git a/csit/libraries/BgpRpcClient.py b/csit/libraries/BgpRpcClient.py
index c744fabdfbacaf06c41816e170e2fe3bf07dc4cb..fe60e2c3be422f06be598b13a40a94600a4b5d45 100644 (file)
--- a/csit/libraries/BgpRpcClient.py
+++ b/csit/libraries/BgpRpcClient.py
@@ -12,9 +12,9 @@ import xmlrpclib
 class BgpRpcClient(object):
     """The client for SimpleXMLRPCServer."""
 
-    def __init__(self, peer_addr):
+    def __init__(self, peer_addr, port=8000):
         """Setup destination point of the rpc server"""
-        self.proxy = xmlrpclib.ServerProxy("http://{}:8000".format(peer_addr))
+        self.proxy = xmlrpclib.ServerProxy("http://{}:{}".format(peer_addr, port))
 
     def exa_announce(self, full_exabgp_cmd):
         """The full command to be passed to exabgp."""
@@ -86,7 +86,7 @@ class BgpRpcClient(object):
 
     def play_send(self, hexstring):
         """Sends given hex data, already encoded bgp update message is expected."""
-        return self.proxy.send(hexstring)
+        return self.proxy.send(hexstring.rstrip())
 
     def play_get(self, what='update'):
         """Gets the last received (update) mesage as hex string."""