|
2 | 2 | # pylint:disable=unused-argument |
3 | 3 | # pylint:disable=redefined-outer-name |
4 | 4 | import pytest |
5 | | -from aws_library.ec2 import AWSTagKey, AWSTagValue, EC2Tags |
| 5 | +from aws_library.ec2 import AWSTagValue, EC2Tags |
6 | 6 | from faker import Faker |
7 | 7 | from fastapi import FastAPI |
8 | 8 | from models_library.docker import DockerGenericTag |
@@ -65,8 +65,8 @@ def test_get_deactivated_buffer_ec2_tags_dynamic( |
65 | 65 | | DEACTIVATED_BUFFER_MACHINE_EC2_TAGS |
66 | 66 | ) |
67 | 67 | assert "Name" in expected_tags |
68 | | - expected_tags[AWSTagKey("Name")] = TypeAdapter(AWSTagValue).validate_python( |
69 | | - str(expected_tags[AWSTagKey("Name")]) + "-buffer" |
| 68 | + expected_tags["Name"] = TypeAdapter(AWSTagValue).validate_python( |
| 69 | + str(expected_tags["Name"]) + "-buffer" |
70 | 70 | ) |
71 | 71 | assert expected_tags == deactivated_buffer_tags |
72 | 72 |
|
@@ -107,8 +107,8 @@ def test_get_deactivated_buffer_ec2_tags_computational( |
107 | 107 | | DEACTIVATED_BUFFER_MACHINE_EC2_TAGS |
108 | 108 | ) |
109 | 109 | assert "Name" in expected_tags |
110 | | - expected_tags[AWSTagKey("Name")] = TypeAdapter(AWSTagValue).validate_python( |
111 | | - str(expected_tags[AWSTagKey("Name")]) + "-buffer" |
| 110 | + expected_tags["Name"] = TypeAdapter(AWSTagValue).validate_python( |
| 111 | + str(expected_tags["Name"]) + "-buffer" |
112 | 112 | ) |
113 | 113 | assert expected_tags == deactivated_buffer_tags |
114 | 114 |
|
|
0 commit comments