From 97dc33a9381c39dc148ae51fad681c22ea9647cb Mon Sep 17 00:00:00 2001 From: Hakan Calim <hakan.calim@fau.de> Date: Fri, 26 Jan 2024 11:51:36 +0100 Subject: [PATCH] NAT-410: added a test for validate tt number --- test/utils/test_helpers.py | 24 +++++++++++++++++++++++- 1 file changed, 23 insertions(+), 1 deletion(-) diff --git a/test/utils/test_helpers.py b/test/utils/test_helpers.py index d5fb2282..a462d1d7 100644 --- a/test/utils/test_helpers.py +++ b/test/utils/test_helpers.py @@ -4,7 +4,7 @@ import pytest from gso.products.product_blocks.iptrunk import IptrunkInterfaceBlock from gso.products.product_blocks.router import RouterVendor -from gso.utils.helpers import available_interfaces_choices_including_current_members +from gso.utils.helpers import available_interfaces_choices_including_current_members, validate_tt_number @pytest.fixture() @@ -62,3 +62,25 @@ def test_nokia_router_with_interfaces_returns_choice(mock_router, mock_netbox_cl assert hasattr(result, "interface1") assert hasattr(result, "interface2") assert hasattr(result, "interface3") + + +def test_tt_number(faker): + """Test different TT numbers""" + + # Define different tt number + tt_number_valid1 = faker.tt_number() + tt_number_valid2 = faker.tt_number() + tt_number_valid3 = faker.tt_number() + + tt_number_not_valid1 = faker.sentence() + err_msg = ( + f"The given TT number: {tt_number_not_valid1} is not valid. " + f" A valid TT number starts with 'TT#' followed by 16 digits." + ) + + assert validate_tt_number(tt_number_valid1) == tt_number_valid1 + assert validate_tt_number(tt_number_valid2) == tt_number_valid2 + assert validate_tt_number(tt_number_valid3) == tt_number_valid3 + + with pytest.raises(ValueError, match=err_msg): + validate_tt_number(tt_number_not_valid1) -- GitLab