cli: use client library

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

1 files changed, 1 insertions, 1 deletions

diff --git a/test/lib/gobgp.py b/test/lib/gobgp.py
index 9d2b4237..8c0d9cdd 100644
--- a/test/lib/gobgp.py
+++ b/test/lib/gobgp.py
@@ -211,7 +211,7 @@ class GoBGPContainer(BGPContainer):
         return json.loads(self.local(cmd, capture=True))
 
     def get_neighbor_state(self, peer):
-        return self.get_neighbor(peer)['info']['bgp_state']
+        return self.get_neighbor(peer)['state']['session-state']
 
     def clear_policy(self):
         self.policies = {}