From fc5f2511b4fe1962038cbd217ecbd746479095da Mon Sep 17 00:00:00 2001
From: Alp Kose <alperenkose@gmail.com>
Date: Wed, 6 Mar 2024 22:02:38 +0300
Subject: [PATCH] test: fix test snapshots with different level

---
 tests/snapshots.py | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/tests/snapshots.py b/tests/snapshots.py
index 4a711c0..d9ad910 100644
--- a/tests/snapshots.py
+++ b/tests/snapshots.py
@@ -376,8 +376,25 @@
             "nexthop-thirdparty": "yes",
             "nexthop-peer": "no",
             "config": {"remove-private-as": "no"},
-            "peer-capability": None,
-            "prefix-counter": None,
+            "peer-capability": {
+                "list": [
+                    {"capability": "Multiprotocol Extensions(1)", "value": "IPv4 Unicast"},
+                    {"capability": "Route Refresh(2)", "value": "yes"},
+                    {"capability": "4-Byte AS Number(65)", "value": "64512"},
+                    {"capability": "Route Refresh (Cisco)(128)", "value": "yes"},
+                ]
+            },
+            "prefix-counter": {
+                "entry": {
+                    "@afi-safi": "bgpAfiIpv4-unicast",
+                    "incoming-total": "2",
+                    "incoming-accepted": "2",
+                    "incoming-rejected": "0",
+                    "policy-rejected": "0",
+                    "outgoing-total": "0",
+                    "outgoing-advertised": "0",
+                }
+            },
         }
     },
     "session_stats": {