Add missing Volume Pricing Information (#165)

Signed-off-by: Lukas Kämmerling <[email protected]>
hetznercloud · Mar 10, 2021 · 34e196b · 34e196b
1 parent 7accec4
commit 34e196b
Show file tree

Hide file tree

Showing 4 changed files with 40 additions and 1 deletion.
diff --git a/hcloud/pricing.go b/hcloud/pricing.go
@@ -14,6 +14,7 @@ type Pricing struct {
 	ServerBackup      ServerBackupPricing
 	ServerTypes       []ServerTypePricing
 	LoadBalancerTypes []LoadBalancerTypePricing
+	Volume            VolumePricing
 }
 
 // Price represents a price. Net amount, gross amount, as well as VAT rate are
@@ -41,6 +42,11 @@ type TrafficPricing struct {
 	PerTB Price
 }
 
+// VolumePricing provides pricing information for a Volume.
+type VolumePricing struct {
+	PerGBMonthly Price
+}
+
 // ServerBackupPricing provides pricing information for server backups.
 type ServerBackupPricing struct {
 	Percentage string

diff --git a/hcloud/schema.go b/hcloud/schema.go
@@ -628,6 +628,14 @@ func PricingFromSchema(s schema.Pricing) Pricing {
 		ServerBackup: ServerBackupPricing{
 			Percentage: s.ServerBackup.Percentage,
 		},
+		Volume: VolumePricing{
+			PerGBMonthly: Price{
+				Currency: s.Currency,
+				VATRate:  s.VATRate,
+				Net:      s.Volume.PricePerGBPerMonth.Net,
+				Gross:    s.Volume.PricePerGBPerMonth.Gross,
+			},
+		},
 	}
 	for _, serverType := range s.ServerTypes {
 		var pricings []ServerTypeLocationPricing

diff --git a/hcloud/schema/pricing.go b/hcloud/schema/pricing.go
@@ -10,6 +10,7 @@ type Pricing struct {
 	ServerBackup      PricingServerBackup       `json:"server_backup"`
 	ServerTypes       []PricingServerType       `json:"server_types"`
 	LoadBalancerTypes []PricingLoadBalancerType `json:"load_balancer_types"`
+	Volume            PricingVolume             `json:"volume"`
 }
 
 // Price defines the schema of a single price with net and gross amount.
@@ -33,6 +34,11 @@ type PricingTraffic struct {
 	PricePerTB Price `json:"price_per_tb"`
 }
 
+// PricingVolume defines the schema of pricing information for a Volume.
+type PricingVolume struct {
+	PricePerGBPerMonth Price `json:"price_per_gb_month"`
+}
+
 // PricingServerBackup defines the schema of pricing information for server backups.
 type PricingServerBackup struct {
 	Percentage string `json:"percentage"`

diff --git a/hcloud/schema_test.go b/hcloud/schema_test.go
@@ -1782,7 +1782,13 @@ func TestPricingFromSchema(t *testing.T) {
 					}
 				]
 			}
-		]
+		],
+		"volume": {
+			"price_per_gb_month": {
+				"net": "1",
+				"gross": "1.19"
+			}
+		}
 	}`)
 
 	var s schema.Pricing
@@ -1817,6 +1823,19 @@ func TestPricingFromSchema(t *testing.T) {
 		t.Errorf("unexpected FloatingIP.Monthly.Gross: %v", pricing.FloatingIP.Monthly.Gross)
 	}
 
+	if pricing.Volume.PerGBMonthly.Currency != "EUR" {
+		t.Errorf("unexpected Traffic.PerTB.Currency: %v", pricing.Volume.PerGBMonthly.Currency)
+	}
+	if pricing.Volume.PerGBMonthly.VATRate != "19.00" {
+		t.Errorf("unexpected Traffic.PerTB.VATRate: %v", pricing.Volume.PerGBMonthly.VATRate)
+	}
+	if pricing.Volume.PerGBMonthly.Net != "1" {
+		t.Errorf("unexpected Traffic.PerTB.Net: %v", pricing.Volume.PerGBMonthly.Net)
+	}
+	if pricing.Volume.PerGBMonthly.Gross != "1.19" {
+		t.Errorf("unexpected Traffic.PerTB.Gross: %v", pricing.Volume.PerGBMonthly.Gross)
+	}
+
 	if pricing.Traffic.PerTB.Currency != "EUR" {
 		t.Errorf("unexpected Traffic.PerTB.Currency: %v", pricing.Traffic.PerTB.Currency)
 	}