api/materialize: make aggregates roundtrip (#198)

Fixes #197
tierrun · Dec 9, 2022 · 83726d7 · 83726d7
1 parent c8f5624
commit 83726d7
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 10 deletions.
diff --git a/api/materialize/views.go b/api/materialize/views.go
@@ -100,9 +100,11 @@ func ToPricingJSON(fs []control.Feature) ([]byte, error) {
 		}
 
 		p.Features[f.FeaturePlan.Name()] = apitypes.Feature{
-			Title: values.ZeroIf(f.Title, f.FeaturePlan.String()),
-			Base:  f.Base,
-			Tiers: tiers,
+			Title:     values.ZeroIf(f.Title, f.FeaturePlan.String()),
+			Base:      f.Base,
+			Mode:      values.ZeroIf(f.Mode, "graduated"),
+			Aggregate: values.ZeroIf(f.Aggregate, "sum"),
+			Tiers:     tiers,
 		}
 		m.Plans[f.Plan()] = p
 	}

diff --git a/api/materialize/views_test.go b/api/materialize/views_test.go
@@ -19,8 +19,9 @@ func TestPricingHuJSON(t *testing.T) {
 			"plan:example@1": {
 				"title": "Just an example plan to show off features",
 				"features": {
-					"feature:graduated": {
-						"mode": "graduated",
+					"feature:volume": {
+						"mode": "volume",
+						"aggregate": "perpetual",
 						"tiers": [
 							{ "upto": 10 },
 							{ "upto": 20, "price": 100 },
@@ -61,12 +62,12 @@ func TestPricingHuJSON(t *testing.T) {
 		},
 		{
 			PlanTitle:   "Just an example plan to show off features",
-			Title:       "feature:graduated@plan:example@1",
-			FeaturePlan: refs.MustParseFeaturePlan("feature:graduated@plan:example@1"),
+			Title:       "feature:volume@plan:example@1",
+			FeaturePlan: refs.MustParseFeaturePlan("feature:volume@plan:example@1"),
 			Currency:    "usd",
 			Interval:    "@monthly",
-			Mode:        "graduated",
-			Aggregate:   "sum",
+			Mode:        "volume",
+			Aggregate:   "perpetual",
 			Tiers: []control.Tier{
 				{Upto: 10, Price: 0, Base: 0},
 				{Upto: 20, Price: 100, Base: 0},
@@ -87,7 +88,9 @@ func TestPricingHuJSON(t *testing.T) {
 			"plan:example@1": {
 				"title": "Just an example plan to show off features",
 				"features": {
-					"feature:graduated": {
+					"feature:volume": {
+						"mode": "volume",
+						"aggregate": "perpetual",
 						"tiers": [
 							{ "upto": 10 },
 							{ "upto": 20, "price": 100 },