[VOL-3461] Lowering flow chunk size because of lager flow size in TT workflow
Change-Id: I840619c3a153cf3106f333198fbca809c1c4cca9
diff --git a/internal/pkg/openflow/stats_test.go b/internal/pkg/openflow/stats_test.go
index e80d239..d618fe4 100644
--- a/internal/pkg/openflow/stats_test.go
+++ b/internal/pkg/openflow/stats_test.go
@@ -185,23 +185,29 @@
}
}
-func TestHandleFlowStatsRequest(t *testing.T) {
+func TestHandleFlowStatsRequestExactlyDivisible(t *testing.T) {
- generatedFlowsCount := 2000
+ generatedFlowsCount := ofcFlowsChunkSize * 5
- ofc := newTestOFConnection(2000, 0)
+ testCorrectChunkAndReplyMoreFlag(t, generatedFlowsCount)
+}
+func TestHandleFlowStatsRequestNotExactlyDivisible(t *testing.T) {
+
+ generatedFlowsCount := int(ofcFlowsChunkSize * 5.5)
+
+ testCorrectChunkAndReplyMoreFlag(t, generatedFlowsCount)
+}
+
+func testCorrectChunkAndReplyMoreFlag(t *testing.T, generatedFlowsCount int) {
+ ofc := newTestOFConnection(generatedFlowsCount, 0)
request := of13.NewFlowStatsRequest()
-
replies, err := ofc.handleFlowStatsRequest(context.Background(), request)
assert.Equal(t, err, nil)
-
// check that the correct number of messages is generated
assert.Equal(t, int(math.Ceil(float64(generatedFlowsCount)/ofcFlowsChunkSize)), len(replies))
-
n := 1
entriesCount := 0
-
for _, r := range replies {
json, _ := r.Flags.MarshalJSON()
@@ -226,7 +232,6 @@
entriesCount = entriesCount + len(r.GetEntries())
n++
}
-
// make sure all the generate item are included in the responses
assert.Equal(t, generatedFlowsCount, entriesCount)
}