From 43e00fbf19fbd9576454a3b672ead48ab3b8333f Mon Sep 17 00:00:00 2001
From: Rodrigo Reis <rodrigo.gargravarr@gmail.com>
Date: Fri, 5 Jul 2024 17:10:53 +0200
Subject: [PATCH] Add Roman's test case.

Signed-off-by: Rodrigo Reis <rodrigo.gargravarr@gmail.com>
---
 pkg/core/traffic_test.go | 26 ++++++++++++++++++++++++++
 1 file changed, 26 insertions(+)

diff --git a/pkg/core/traffic_test.go b/pkg/core/traffic_test.go
index 4e908a72..a5e4ee53 100644
--- a/pkg/core/traffic_test.go
+++ b/pkg/core/traffic_test.go
@@ -1853,6 +1853,32 @@ func TestComputeTrafficSegments(t *testing.T) {
 			},
 			expectErr: false,
 		},
+		{
+			actualTrafficWeights: map[types.UID]float64{
+				"v1": 50.0,
+				"v2": 50.0,
+			},
+			ingressSegments: map[types.UID]string{
+				"v1": "TrafficSegment(0.0, 0.26)",
+				"v2": "TrafficSegment(0.24, 0.50)",
+				"v3": "TrafficSegment(0.50, 0.75)",
+				"v4": "TrafficSegment(0.75, 1.0)",
+			},
+			routeGroupSegments: map[types.UID]string{
+				"v1": "",
+				"v2": "",
+			},
+			expected: []types.UID{"v1", "v2"},
+			expectedLowerLimits: map[types.UID]float64{
+				"v2": 0.5,
+				"v1": 0.0,
+			},
+			expectedUpperLimits: map[types.UID]float64{
+				"v2": 1.0,
+				"v1": 0.5,
+			},
+			expectErr: false,
+		},
 	} {
 		stackContainers := map[types.UID]*StackContainer{}
 		for k, v := range tc.actualTrafficWeights {