Adds helper functions for integration tests.

2021-05-17 22:49:29 +03:00 · 2021-05-17 22:49:29 +03:00 · f3e06dcd92
commit f3e06dcd92
parent c0f578db75
1 changed files with 3 additions and 6 deletions
--- a/apps/cic-ussd/tests/helpers/accounts.py
+++ b/apps/cic-ussd/tests/helpers/accounts.py
@ -1,4 +1,5 @@
 # standard imports
+import random
 import uuid

 # external imports
@ -21,9 +22,5 @@ def session_id() -> str:
    return uuid.uuid4().hex


-def given_name() -> str:
-    return fake.first_name()
-
-
-def family_name() -> str:
-    return fake.last_name()
+def pin_number() -> int:
+    return random.randint(1000, 9999)