From fdb240f5f99cb3301c2adaaa8acfde7c17835e46 Mon Sep 17 00:00:00 2001
From: Steve Yen <steve.yen@gmail.com>
Date: Mon, 5 Feb 2018 10:02:44 -0800
Subject: [PATCH] more zap merge-planner CalcBudget tests at larger sizes

Helps provide a sense of how # of segments grows as # of documents
grows.  Ex: 1B docs => budget of 54 segments.
---
 index/scorch/mergeplan/merge_plan_test.go | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/index/scorch/mergeplan/merge_plan_test.go b/index/scorch/mergeplan/merge_plan_test.go
index 4db8eb1e..9a2c779c 100644
--- a/index/scorch/mergeplan/merge_plan_test.go
+++ b/index/scorch/mergeplan/merge_plan_test.go
@@ -188,6 +188,18 @@ func TestCalcBudget(t *testing.T) {
 			},
 			7,
 		},
+		{1000, 2000, DefaultMergePlanOptions,
+			1},
+		{5000, 2000, DefaultMergePlanOptions,
+			3},
+		{10000, 2000, DefaultMergePlanOptions,
+			5},
+		{30000, 2000, DefaultMergePlanOptions,
+			11},
+		{1000000, 2000, DefaultMergePlanOptions,
+			24},
+		{1000000000, 2000, DefaultMergePlanOptions,
+			54},
 	}
 
 	for testi, test := range tests {