From e9b34ee03738ca6c02d39083d98faa26ac858d07 Mon Sep 17 00:00:00 2001
From: Emmy Ni <emmyni220@gmail.com>
Date: Fri, 21 Oct 2022 10:23:31 -0400
Subject: [PATCH] fix tests

---
 api4/cloud_test.go | 42 +++++++++++++++++++++++++++---------------
 1 file changed, 27 insertions(+), 15 deletions(-)

diff --git a/api4/cloud_test.go b/api4/cloud_test.go
index 999f1077e4..8dfa7747b8 100644
--- a/api4/cloud_test.go
+++ b/api4/cloud_test.go
@@ -456,6 +456,7 @@ func TestGetCloudProducts(t *testing.T) {
 			Family:            "family",
 			RecurringInterval: "recurring_interval",
 			BillingScheme:     "billing_scheme",
+			CrossSellsTo:      "",
 		},
 		{
 			ID:                "prod_test2",
@@ -467,6 +468,7 @@ func TestGetCloudProducts(t *testing.T) {
 			Family:            "family2",
 			RecurringInterval: "recurring_interval2",
 			BillingScheme:     "billing_scheme2",
+			CrossSellsTo:      "prod_test3",
 		},
 		{
 			ID:                "prod_test3",
@@ -478,27 +480,34 @@ func TestGetCloudProducts(t *testing.T) {
 			Family:            "family3",
 			RecurringInterval: "recurring_interval3",
 			BillingScheme:     "billing_scheme3",
+			CrossSellsTo:      "prod_test2",
 		},
 	}
 
 	sanitizedProducts := []*model.Product{
 		{
-			ID:           "prod_test1",
-			Name:         "name",
-			PricePerSeat: 10,
-			SKU:          "sku",
+			ID:                "prod_test1",
+			Name:              "name",
+			PricePerSeat:      10,
+			SKU:               "sku",
+			RecurringInterval: "recurring_interval",
+			CrossSellsTo:      "",
 		},
 		{
-			ID:           "prod_test2",
-			Name:         "name2",
-			PricePerSeat: 100,
-			SKU:          "sku2",
+			ID:                "prod_test2",
+			Name:              "name2",
+			PricePerSeat:      100,
+			SKU:               "sku2",
+			RecurringInterval: "recurring_interval2",
+			CrossSellsTo:      "prod_test3",
 		},
 		{
-			ID:           "prod_test3",
-			Name:         "name3",
-			PricePerSeat: 1000,
-			SKU:          "sku3",
+			ID:                "prod_test3",
+			Name:              "name3",
+			PricePerSeat:      1000,
+			SKU:               "sku3",
+			RecurringInterval: "recurring_interval3",
+			CrossSellsTo:      "prod_test2",
 		},
 	}
 	t.Run("get products for admins", func(t *testing.T) {
@@ -554,8 +563,9 @@ func TestGetCloudProducts(t *testing.T) {
 		require.Equal(t, returnedProducts[0].Description, "")
 		require.Equal(t, returnedProducts[0].PriceID, "")
 		require.Equal(t, returnedProducts[0].Family, model.SubscriptionFamily(""))
-		require.Equal(t, returnedProducts[0].RecurringInterval, model.RecurringInterval(""))
+		require.Equal(t, returnedProducts[0].RecurringInterval, model.RecurringInterval("recurring_interval"))
 		require.Equal(t, returnedProducts[0].BillingScheme, model.BillingScheme(""))
+		require.Equal(t, returnedProducts[0].CrossSellsTo, "")
 
 		require.Equal(t, returnedProducts[1].ID, "prod_test2")
 		require.Equal(t, returnedProducts[1].Name, "name2")
@@ -564,8 +574,9 @@ func TestGetCloudProducts(t *testing.T) {
 		require.Equal(t, returnedProducts[1].Description, "")
 		require.Equal(t, returnedProducts[1].PriceID, "")
 		require.Equal(t, returnedProducts[1].Family, model.SubscriptionFamily(""))
-		require.Equal(t, returnedProducts[1].RecurringInterval, model.RecurringInterval(""))
+		require.Equal(t, returnedProducts[1].RecurringInterval, model.RecurringInterval("recurring_interval2"))
 		require.Equal(t, returnedProducts[1].BillingScheme, model.BillingScheme(""))
+		require.Equal(t, returnedProducts[1].CrossSellsTo, "prod_test3")
 
 		require.Equal(t, returnedProducts[2].ID, "prod_test3")
 		require.Equal(t, returnedProducts[2].Name, "name3")
@@ -574,7 +585,8 @@ func TestGetCloudProducts(t *testing.T) {
 		require.Equal(t, returnedProducts[2].Description, "")
 		require.Equal(t, returnedProducts[2].PriceID, "")
 		require.Equal(t, returnedProducts[2].Family, model.SubscriptionFamily(""))
-		require.Equal(t, returnedProducts[2].RecurringInterval, model.RecurringInterval(""))
+		require.Equal(t, returnedProducts[2].RecurringInterval, model.RecurringInterval("recurring_interval3"))
 		require.Equal(t, returnedProducts[2].BillingScheme, model.BillingScheme(""))
+		require.Equal(t, returnedProducts[2].CrossSellsTo, "prod_test2")
 	})
 }