Skip to content

Commit 22f56da

Browse files
committed
Merge branch 'morosi-feat' into 'master'
Don't make AutoVersion.get_version return None See merge request it/e3-aws!65
2 parents f2a3f2a + c9c43c5 commit 22f56da

File tree

2 files changed

+18
-23
lines changed

2 files changed

+18
-23
lines changed

src/e3/aws/troposphere/awslambda/__init__.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -746,18 +746,18 @@ def __init__(
746746
self.latest.provisioned_concurrency_config = provisioned_concurrency_config
747747
self.latest.code_sha256 = code_sha256
748748

749-
def get_version(self, number: int) -> Version | None:
749+
def get_version(self, number: int) -> Version:
750750
"""Return a version.
751751
752752
:param number: version number
753+
:raises ValueError: if number is out of range
753754
"""
754-
number = number - self.min_version
755+
index = number - self.min_version
755756

756-
return (
757-
self.versions[number]
758-
if number >= 0 and number < len(self.versions)
759-
else None
760-
)
757+
if index >= 0 and index < len(self.versions):
758+
return self.versions[index]
759+
760+
raise ValueError(f"version {number} if out of range")
761761

762762
@property
763763
def previous(self) -> Version:

tests/tests_e3_aws/troposphere/awslambda/awslambda_test.py

Lines changed: 11 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -794,14 +794,13 @@ def test_autoversion_default(stack: Stack, simple_lambda_function: PyFunction) -
794794
stack.add(auto_version)
795795
print(stack.export()["Resources"])
796796
assert stack.export()["Resources"] == EXPECTED_AUTOVERSION_DEFAULT_TEMPLATE
797-
assert (
798-
version := auto_version.get_version(1)
799-
) and version.name == "mypylambdaVersion1"
800-
assert (
801-
version := auto_version.get_version(2)
802-
) and version.name == "mypylambdaVersion2"
803-
assert (version := auto_version.previous) and version.name == "mypylambdaVersion1"
804-
assert (version := auto_version.latest) and version.name == "mypylambdaVersion2"
797+
assert auto_version.get_version(1).name == "mypylambdaVersion1"
798+
assert auto_version.get_version(2).name == "mypylambdaVersion2"
799+
assert auto_version.previous.name == "mypylambdaVersion1"
800+
assert auto_version.latest.name == "mypylambdaVersion2"
801+
802+
with pytest.raises(ValueError):
803+
auto_version.get_version(3)
805804

806805

807806
def test_autoversion_single(stack: Stack, simple_lambda_function: PyFunction) -> None:
@@ -832,14 +831,10 @@ def test_autoversion(stack: Stack, simple_lambda_function: PyFunction) -> None:
832831
stack.add(auto_version)
833832
print(stack.export()["Resources"])
834833
assert stack.export()["Resources"] == EXPECTED_AUTOVERSION_TEMPLATE
835-
assert (
836-
version := auto_version.get_version(2)
837-
) and version.name == "mypylambdaVersion2"
838-
assert (
839-
version := auto_version.get_version(3)
840-
) and version.name == "mypylambdaVersion3"
841-
assert (version := auto_version.previous) and version.name == "mypylambdaVersion2"
842-
assert (version := auto_version.latest) and version.name == "mypylambdaVersion3"
834+
assert auto_version.get_version(2).name == "mypylambdaVersion2"
835+
assert auto_version.get_version(3).name == "mypylambdaVersion3"
836+
assert auto_version.previous.name == "mypylambdaVersion2"
837+
assert auto_version.latest.name == "mypylambdaVersion3"
843838

844839

845840
def test_bluegreenaliases_default(

0 commit comments

Comments
 (0)