Skip to content

Commit e256458

Browse files
committed
Code review fixes
1 parent 2736865 commit e256458

1 file changed

Lines changed: 28 additions & 16 deletions

File tree

test/integration/models/linode/test_linode.py

Lines changed: 28 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -234,16 +234,15 @@ def get_status(linode: Instance, status: str):
234234
def instance_type_condition(linode: Instance, type: str):
235235
return type in str(linode.type)
236236

237-
def test_get_linodes_verify_alerts(test_linode_client):
237+
238+
def test_get_linodes_verify_alerts(test_linode_client, create_linode):
238239
linodes_list = test_linode_client.linode.instances().lists[0]
239240
assert len(linodes_list) > 0
240241
assert linodes_list[0].alerts.cpu >= 0
241242
assert linodes_list[0].alerts.io >= 0
242243
assert linodes_list[0].alerts.network_in >= 0
243244
assert linodes_list[0].alerts.network_out >= 0
244245
assert linodes_list[0].alerts.transfer_quota >= 0
245-
assert isinstance(linodes_list[0].alerts.system_alerts, list)
246-
assert isinstance(linodes_list[0].alerts.user_alerts, list)
247246

248247

249248
def test_get_linode(test_linode_client, linode_with_volume_firewall):
@@ -379,42 +378,55 @@ def test_linode_alerts_workflow(test_linode_client, create_linode):
379378
assert isinstance(linode.alerts.system_alerts, list)
380379
assert isinstance(linode.alerts.user_alerts, list)
381380

382-
linode.alerts={
381+
linode.alerts = {
383382
"cpu": 50,
384383
"io": 6000,
385384
"network_in": 20,
386385
"network_out": 20,
387-
"transfer": 80,
386+
"transfer_quota": 40,
388387
}
389388
linode.save()
390389

391390
wait_for_condition(10, 100, get_status, linode, "running")
392-
new_linode = retry_sending_request(5, linode.clone, parent_linode_id)
393-
assert new_linode.alerts.cpu == 50
394-
assert new_linode.alerts.io == 6000
395-
assert new_linode.alerts.network_in == 20
396-
assert new_linode.alerts.network_out == 20
391+
new_linode = retry_sending_request(
392+
5,
393+
linode.clone,
394+
region=linode.region.id,
395+
instance_type=linode.type.id,
396+
label=get_test_label(),
397+
)
398+
assert new_linode.alerts.cpu == 90
399+
assert new_linode.alerts.io == 10000
400+
assert new_linode.alerts.network_in == 10
401+
assert new_linode.alerts.network_out == 10
397402
assert new_linode.alerts.transfer_quota == 80
398403
assert isinstance(new_linode.alerts.system_alerts, list)
399404
assert isinstance(new_linode.alerts.user_alerts, list)
400405

406+
if new_linode is not None:
407+
new_linode.delete()
408+
401409

402-
def test_try_to_update_linode_alerts_legacy_and_aclp_at_the_same_time(create_linode):
410+
def test_try_to_update_linode_alerts_legacy_and_aclp_at_the_same_time(
411+
create_linode,
412+
):
403413
linode = create_linode
404414

405-
linode.alerts={
415+
linode.alerts = {
406416
"cpu": 50,
407417
"io": 6000,
408418
"network_in": 20,
409419
"network_out": 20,
410-
"transfer": 50,
411-
"system_alerts": [1,436],
420+
"transfer_quota": 50,
421+
"system_alerts": [1, 436],
412422
"user_alerts": [555],
413423
}
414424

415-
with pytest.raises(RuntimeError) as err:
425+
with pytest.raises(ApiError) as err:
416426
linode.save()
417-
assert "Cannot set both legacy and ACLP alerts simultaneously" in str(err.value)
427+
assert "Cannot set both legacy and ACLP alerts simultaneously" in str(
428+
err.value
429+
)
418430
assert "[400] alerts" in str(err.value)
419431

420432

0 commit comments

Comments
 (0)