From 50f1276c038a411e2e3dfd7d098e2c89650589f6 Mon Sep 17 00:00:00 2001
From: Prem Singh <94350992+singh-prem@users.noreply.github.com>
Date: Mon, 5 Aug 2024 20:49:12 +0530
Subject: [PATCH] adding fix to keep decap encap tunnels within specific limit
 (#3322)

Co-authored-by: Ram <rmachat@google.com>
---
 .../otg_tests/gribi_scaling/gribi_scaling_test.go      | 10 ++++++++++
 .../gribi/otg_tests/gribi_scaling/metadata.textproto   |  1 +
 2 files changed, 11 insertions(+)

diff --git a/feature/gribi/otg_tests/gribi_scaling/gribi_scaling_test.go b/feature/gribi/otg_tests/gribi_scaling/gribi_scaling_test.go
index ec4b9f81cc3..0d601a390aa 100644
--- a/feature/gribi/otg_tests/gribi_scaling/gribi_scaling_test.go
+++ b/feature/gribi/otg_tests/gribi_scaling/gribi_scaling_test.go
@@ -263,6 +263,7 @@ func incrementMAC(mac string, i int) (string, error) {
 
 func TestScaling(t *testing.T) {
 	dut := ondatra.DUT(t, "dut")
+	overrideScaleParams(dut)
 	ate := ondatra.ATE(t, "ate")
 
 	ctx := context.Background()
@@ -378,3 +379,12 @@ func checkTraffic(t *testing.T, ate *ondatra.ATEDevice, top gosnappi.Config) {
 		t.Errorf("FAIL- Got %v%% packet loss for %s ; expected < 1%%", lossPct, "flow")
 	}
 }
+
+// overrideScaleParams allows to override the default scale parameters based on the DUT vendor.
+func overrideScaleParams(dut *ondatra.DUTDevice) {
+	if deviations.OverrideDefaultNhScale(dut) {
+		if dut.Vendor() == ondatra.CISCO {
+			*fpargs.V4TunnelCount = 3328
+		}
+	}
+}
diff --git a/feature/gribi/otg_tests/gribi_scaling/metadata.textproto b/feature/gribi/otg_tests/gribi_scaling/metadata.textproto
index 7877fc7e26a..981e65d1ab9 100644
--- a/feature/gribi/otg_tests/gribi_scaling/metadata.textproto
+++ b/feature/gribi/otg_tests/gribi_scaling/metadata.textproto
@@ -12,6 +12,7 @@ platform_exceptions: {
   deviations: {
     ipv4_missing_enabled: true
     interface_ref_interface_id_format: true
+    override_default_nh_scale: true
   }
 }
 platform_exceptions: {