Fix tests

m3db · Nov 1, 2017 · a58effe · a58effe
1 parent b1e2205
commit a58effe
Showing 1 changed file with 25 additions and 0 deletions.
diff --git a/rules/ruleset_test.go b/rules/ruleset_test.go
@@ -175,6 +175,15 @@ func TestActiveRuleSetForwardMappingPoliciesForNonRollupID(t *testing.T) {
 						policy.NewPolicy(policy.NewStoragePolicy(time.Minute, xtime.Minute, time.Hour), policy.DefaultAggregationID),
 					},
 				),
+				policy.NewStagedPolicies(
+					35000,
+					false,
+					[]policy.Policy{
+						policy.NewPolicy(policy.NewStoragePolicy(10*time.Second, xtime.Second, 2*time.Hour), policy.DefaultAggregationID),
+						policy.NewPolicy(policy.NewStoragePolicy(30*time.Second, xtime.Second, 6*time.Hour), policy.DefaultAggregationID),
+						policy.NewPolicy(policy.NewStoragePolicy(time.Minute, xtime.Minute, time.Hour), policy.DefaultAggregationID),
+					},
+				),
 			},
 		},
 		{
@@ -387,6 +396,15 @@ func TestActiveRuleSetReverseMappingPoliciesForNonRollupID(t *testing.T) {
 						policy.NewPolicy(policy.NewStoragePolicy(time.Minute, xtime.Minute, time.Hour), policy.DefaultAggregationID),
 					},
 				),
+				policy.NewStagedPolicies(
+					35000,
+					false,
+					[]policy.Policy{
+						policy.NewPolicy(policy.NewStoragePolicy(10*time.Second, xtime.Second, 2*time.Hour), policy.DefaultAggregationID),
+						policy.NewPolicy(policy.NewStoragePolicy(30*time.Second, xtime.Second, 6*time.Hour), policy.DefaultAggregationID),
+						policy.NewPolicy(policy.NewStoragePolicy(time.Minute, xtime.Minute, time.Hour), policy.DefaultAggregationID),
+					},
+				),
 			},
 		},
 		{
@@ -729,6 +747,13 @@ func TestActiveRuleSetRollupResults(t *testing.T) {
 								policy.NewPolicy(policy.NewStoragePolicy(10*time.Second, xtime.Second, 24*time.Hour), policy.DefaultAggregationID),
 							},
 						),
+						policy.NewStagedPolicies(
+							30000,
+							false,
+							[]policy.Policy{
+								policy.NewPolicy(policy.NewStoragePolicy(10*time.Second, xtime.Second, 24*time.Hour), policy.DefaultAggregationID),
+							},
+						),
 						policy.NewStagedPolicies(
 							34000,
 							true,