Update subscription_updater_test.go (#1281)

* Update subscription_updater_test.go

* Update subscription_updater_test.go

Author: shwetamurali

subscription_updater_test.go

@@ -30,6 +30,7 @@ type updateFeatureSetOptions struct {
 	Type                          string     `jsonapi:"primary,subscription"`
 	RunsCeiling                   *int       `jsonapi:"attr,runs-ceiling,omitempty"`
 	ContractStartAt               *time.Time `jsonapi:"attr,contract-start-at,iso8601,omitempty"`
+	EndAt                         *time.Time `jsonapi:"attr,end-at,iso8601,omitempty"`
 	ContractUserLimit             *int       `jsonapi:"attr,contract-user-limit,omitempty"`
 	ContractApplyLimit            *int       `jsonapi:"attr,contract-apply-limit,omitempty"`
 	ContractManagedResourcesLimit *int       `jsonapi:"attr,contract-managed-resources-limit,omitempty"`
@@ -55,11 +56,13 @@ func (b *organizationSubscriptionUpdater) WithBusinessPlan() *organizationSubscr
 
 	ceiling := 10
 	start := time.Now()
+	end := time.Now().AddDate(1, 0, 0) // 1 year from now
 	userLimit := 1000
 	applyLimit := 5000
 
 	b.updateOpts.RunsCeiling = &ceiling
 	b.updateOpts.ContractStartAt = &start
+	b.updateOpts.EndAt = &end
 	b.updateOpts.ContractUserLimit = &userLimit
 	b.updateOpts.ContractApplyLimit = &applyLimit
 	return b
@@ -76,10 +79,12 @@ func (b *organizationSubscriptionUpdater) WithStandardEntitlementPlan() *organiz
 	b.planName = "Standard (entitlement)"
 
 	start := time.Now()
+	end := time.Now().AddDate(1, 0, 0) // 1 year from now
 	ceiling := 1
 	managedResourcesLimit := 1000
 
 	b.updateOpts.ContractStartAt = &start
+	b.updateOpts.EndAt = &end
 	b.updateOpts.RunsCeiling = &ceiling
 	b.updateOpts.ContractManagedResourcesLimit = &managedResourcesLimit
 	return b