diff --git a/hcloud/servers/domain.py b/hcloud/servers/domain.py index 93392b39..80d845df 100644 --- a/hcloud/servers/domain.py +++ b/hcloud/servers/domain.py @@ -33,6 +33,8 @@ class Server(BaseDomain): Inbound Traffic for the current billing period in bytes :param included_traffic: int Free Traffic for the current billing period in bytes + :param primary_disk_size: int + Size of the primary Disk :param protection: dict Protection configuration for the server :param labels: dict @@ -79,7 +81,8 @@ class Server(BaseDomain): "labels", "volumes", "private_net", - "created" + "created", + "primary_disk_size" ) def __init__( @@ -103,6 +106,7 @@ def __init__( labels=None, volumes=None, private_net=None, + primary_disk_size=None, ): self.id = id self.name = name @@ -123,6 +127,7 @@ def __init__( self.labels = labels self.volumes = volumes self.private_net = private_net + self.primary_disk_size = primary_disk_size class CreateServerResponse(BaseDomain): diff --git a/tests/unit/servers/conftest.py b/tests/unit/servers/conftest.py index c967c979..cb526421 100644 --- a/tests/unit/servers/conftest.py +++ b/tests/unit/servers/conftest.py @@ -122,6 +122,7 @@ def response_simple_server(): "outgoing_traffic": 123456, "ingoing_traffic": 123456, "included_traffic": 654321, + "primary_disk_size": 20, "protection": {}, "labels": {}, "volumes": [] @@ -137,6 +138,7 @@ def response_create_simple_server(): "name": "my-server", "status": "running", "created": "2016-01-30T23:50+00:00", + "primary_disk_size": 20, "public_net": { "ipv4": { "ip": "1.2.3.4", @@ -661,6 +663,7 @@ def response_simple_servers(): "outgoing_traffic": 123456, "ingoing_traffic": 123456, "included_traffic": 654321, + "primary_disk_size": 20, "protection": {}, "labels": {}, "volumes": [] @@ -676,6 +679,7 @@ def response_full_server(): "name": "my-server", "status": "running", "created": "2016-01-30T23:50+00:00", + "primary_disk_size": 20, "public_net": { "ipv4": { "ip": "1.2.3.4", diff --git a/tests/unit/servers/test_client.py b/tests/unit/servers/test_client.py index 52a7b2b7..d605d188 100644 --- a/tests/unit/servers/test_client.py +++ b/tests/unit/servers/test_client.py @@ -35,6 +35,7 @@ def test_bound_server_init(self, response_full_server): assert bound_server.id == 42 assert bound_server.name == "my-server" + assert bound_server.primary_disk_size == 20 assert isinstance(bound_server.public_net, PublicNetwork) assert isinstance(bound_server.public_net.ipv4, IPv4Address) assert bound_server.public_net.ipv4.ip == "1.2.3.4"