Skip to content

Commit b33c711

Browse files
committed
Fix image endpoint tests
1 parent 0f0697c commit b33c711

2 files changed

Lines changed: 6 additions & 6 deletions

File tree

tests/integration/images/test_images.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ def test_get_actions(self, bound_image):
1818
assert actions[0].command == "change_protection"
1919

2020
def test_update(self, bound_image):
21-
image = bound_image.update(description="My new Image description", type="the new image type", labels={})
21+
image = bound_image.update(description="My new Image description", type="snapshot", labels={})
2222

2323
assert image.id == 4711
2424
assert image.description == "My new Image description"
@@ -58,7 +58,7 @@ def test_get_actions(self, hetzner_client, image):
5858

5959
@pytest.mark.parametrize("image", [Image(id=1), BoundImage(mock.MagicMock(), dict(id=1))])
6060
def test_update(self, hetzner_client, image):
61-
image = hetzner_client.images.update(image, description="My new Image description", type="the new image type", labels={})
61+
image = hetzner_client.images.update(image, description="My new Image description", type="snapshot", labels={})
6262

6363
assert image.id == 4711
6464
assert image.description == "My new Image description"

tests/unit/images/test_client.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -55,8 +55,8 @@ def test_get_actions(self, hetzner_client, bound_image, response_get_actions):
5555

5656
def test_update(self, hetzner_client, bound_image, response_update_image):
5757
hetzner_client.request.return_value = response_update_image
58-
image = bound_image.update(description="My new Image description", type="the new image type", labels={})
59-
hetzner_client.request.assert_called_with(url="/images/14", method="PUT", json={"description": "My new Image description", "type": "the new image type", "labels": {}})
58+
image = bound_image.update(description="My new Image description", type="snapshot", labels={})
59+
hetzner_client.request.assert_called_with(url="/images/14", method="PUT", json={"description": "My new Image description", "type": "snapshot", "labels": {}})
6060

6161
assert image.id == 4711
6262
assert image.description == "My new Image description"
@@ -130,8 +130,8 @@ def test_get_actions(self, images_client, image, response_get_actions):
130130
@pytest.mark.parametrize("image", [Image(id=1), BoundImage(mock.MagicMock(), dict(id=1))])
131131
def test_update(self, images_client, image, response_update_image):
132132
images_client._client.request.return_value = response_update_image
133-
image = images_client.update(image, description="My new Image description", type="the new image type", labels={})
134-
images_client._client.request.assert_called_with(url="/images/1", method="PUT", json={"description": "My new Image description", "type": "the new image type", "labels": {}})
133+
image = images_client.update(image, description="My new Image description", type="snapshot", labels={})
134+
images_client._client.request.assert_called_with(url="/images/1", method="PUT", json={"description": "My new Image description", "type": "snapshot", "labels": {}})
135135

136136
assert image.id == 4711
137137
assert image.description == "My new Image description"

0 commit comments

Comments
 (0)