catarrh/pysim
1
0
Fork 0
mirror of https://gitea.osmocom.org/sim-card/pysim.git synced 2026-03-16 18:38:32 +03:00
Code Issues Packages Projects Releases Wiki Activity

param_source: allow plugging a random implementation (for testing)

Browse Source 
Change-Id: Idce2b18af70c17844d6f09f7704efc869456ac39
This commit is contained in:
Neels Hofmeyr
2025-08-05 03:19:30 +02:00
parent 3bee435c2a
commit b39f69ba89
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
pySim/esim/saip/param_source.py
View File

@@ -120,15 +120,18 @@ class DecimalRangeSource(InputExpandingParamSource):
last_value = int(last_str) if last_str is not None else "9" * len(first_str)
return cls(num_digits=len(first_str), first_value=first_value, last_value=last_value)
class RandomDigitSource(DecimalRangeSource):
class RandomSourceMixin:
random_impl = random # TODO secure random source?
class RandomDigitSource(DecimalRangeSource, RandomSourceMixin):
"""return a different sequence of random decimal digits each"""
name = "random decimal digits"
def get_next(self, csv_row:dict=None):
val = random.randint(*self.val_first_last) # TODO secure random source?
val = self.random_impl.randint(*self.val_first_last)
return self.val_to_digit(val)
class RandomHexDigitSource(InputExpandingParamSource):
class RandomHexDigitSource(InputExpandingParamSource, RandomSourceMixin):
"""return a different sequence of random hexadecimal digits each"""
name = "random hexadecimal digits"
@@ -143,7 +146,7 @@ class RandomHexDigitSource(InputExpandingParamSource):
self.num_digits = num_digits
def get_next(self, csv_row:dict=None):
val = random.randbytes(self.num_digits // 2) # TODO secure random source?
val = self.random_impl.randbytes(self.num_digits // 2)
return b2h(val)
@classmethod