1 files changed, 16 insertions, 1 deletions
diff --git a/test/scenario_test/quagga_access.py b/test/scenario_test/quagga_access.py
index f29cb301..634dc439 100644
--- a/test/scenario_test/quagga_access.py
+++ b/test/scenario_test/quagga_access.py
@@ -142,4 +142,19 @@ def lookup_prefix(tn, prefix, af):
             path['Next Hop'] = nexthop
             paths.append(path)
 
-    return paths
-\ No newline at end of file
+    return paths
+
+def check_community(tn, addr, community, af=IPv4):
+    if af == IPv4:
+        tn.write("show ip bgp community " + community + "\n")
+    elif af == IPv6:
+        tn.write("show bgp ipv6 community " + community + "\n")
+    else:
+        print "invalid af: ", af
+        return
+    result = tn.read_until("bgpd#")
+    for line in result.split("\n"):
+        if addr in line:
+            return True
+
+    return False
+\ No newline at end of file