ensurePricing

models/resources/data.go

@@ -164,11 +164,28 @@ type PricedDataResource struct {
 	UsageStorageGB float64 `json:"storage_gb,omitempty" bson:"storage_gb,omitempty"`
 }
 
+func (r *PricedDataResource) ensurePricing() {
+	if r.SelectedPricing == nil {
+		r.SelectedPricing = &DataResourcePricingProfile{}
+	}
+}
+
+func (r *PricedDataResource) IsPurchasable() bool {
+	r.ensurePricing()
+	return r.SelectedPricing.IsPurchasable()
+}
+
+func (r *PricedDataResource) IsBooked() bool {
+	r.ensurePricing()
+	return r.SelectedPricing.IsBooked()
+}
+
 func (r *PricedDataResource) GetType() tools.DataType {
 	return tools.DATA_RESOURCE
 }
 
 func (r *PricedDataResource) GetPriceHT() (float64, error) {
+	r.ensurePricing()
 	if r.BookingConfiguration == nil {
 		r.BookingConfiguration = &BookingConfiguration{}
 	}
@@ -180,9 +197,6 @@ func (r *PricedDataResource) GetPriceHT() (float64, error) {
 		add := r.BookingConfiguration.UsageStart.Add(time.Duration(1 * time.Hour))
 		r.BookingConfiguration.UsageEnd = &add
 	}
-	if r.SelectedPricing == nil {
-		return 0, errors.New("pricing profile must be set on Priced Data" + r.ResourceID)
-	}
 	pricing := r.SelectedPricing
 	var err error
 	amountOfData := float64(1)