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