BUG-2571 : fixed flowspec model

Change-Id: Id38fcd8543a0a248a5cc0bb4058f6f7fa1f37784
Signed-off-by: Dana Kutenicsova <dkutenic@cisco.com>

diff --git a/bgp/flowspec/src/main/yang/bgp-flowspec.yang b/bgp/flowspec/src/main/yang/bgp-flowspec.yang
index d7f63fd948f29c611c79bc18e7656d6f298dda62..89f385a99faffac2cf2c808b24de5a78cf0b2c79 100644 (file)
--- a/bgp/flowspec/src/main/yang/bgp-flowspec.yang
+++ b/bgp/flowspec/src/main/yang/bgp-flowspec.yang
@@ -303,11 +303,10 @@ module bgp-flowspec {
                     type string;
                 }
                 key "route-key";
-                uses bgp-rib:route {
-                    augment bgp-msg:attributes {
-                        uses flowspec-destination;
-                    }
-                }
+
+                uses flowspec-destination;
+
+                uses bgp-rib:route;
             }
         }
     }
@@ -372,4 +371,22 @@ module bgp-flowspec {
             uses flowspec-routes;
         }
     }
+
+    augment "/bgp-rib:bgp-rib/bgp-rib:rib/bgp-rib:peer/bgp-rib:adj-rib-in/bgp-rib:tables/bgp-rib:routes" {
+        case flowspec-routes-case {
+            uses flowspec-routes;
+        }
+    }
+
+    augment "/bgp-rib:bgp-rib/bgp-rib:rib/bgp-rib:peer/bgp-rib:effective-rib-in/bgp-rib:tables/bgp-rib:routes" {
+        case flowspec-routes-case {
+            uses flowspec-routes;
+        }
+    }
+
+    augment "/bgp-rib:bgp-rib/bgp-rib:rib/bgp-rib:peer/bgp-rib:adj-rib-out/bgp-rib:tables/bgp-rib:routes" {
+        case flowspec-routes-case {
+            uses flowspec-routes;
+        }
+    }
 }