From 10609aed4a6477f6edeef3500c6397532996d76e Mon Sep 17 00:00:00 2001
From: Hakan Calim <hakan.calim@fau.de>
Date: Thu, 26 Oct 2023 11:10:29 +0200
Subject: [PATCH] NAT-328: added random generator for site names in faker
 provider

---
 test/conftest.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/test/conftest.py b/test/conftest.py
index 25c830dc..8d08f7b5 100644
--- a/test/conftest.py
+++ b/test/conftest.py
@@ -59,6 +59,19 @@ class FakerProvider(BaseProvider):
     def geant_sid(self) -> str:
         return self.generator.numerify("SID-#####")
 
+    def site_name(self) -> str:
+        # define pointer to random letter method
+        def get_upper():
+            return self.generator.random_letter().upper()
+
+        random_int = self.generator.random_int(min=0, max=1)
+        if random_int == 0:
+            # Returns ABC0...ABC9
+            return get_upper() + get_upper() + get_upper() + str(self.generator.random_int(min=0, max=9))
+        elif random_int == 1:
+            # Returns ABC
+            return get_upper() + get_upper() + get_upper()
+
     def network_interface(self) -> str:
         return self.generator.numerify("ge-@#/@#/@#")
 
-- 
GitLab