osmo-gsm-tester/src/osmo_ms_driver/test_support.py

52 lines
1.6 KiB
Python
Raw Normal View History

# osmo_ms_driver: Test helpers and base classes
#
# Copyright (C) 2018 by Holger Hans Peter Freyther
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
from osmo_gsm_tester import log
def imsi_ki_gen():
"""
Generate IMSIs and KIs to be used by test.
"""
n = 1010000000000
while True:
yield ("%.15d" % n, "00 00 00 00 00 00 00 00 00 00 00 00 00 00 00 00")
n += 1
class Results(log.Origin):
"""
A base class to collect results from tests.
"""
def __init__(self, name):
super().__init__(log.C_RUN, name)
self._time_of_registration = None
self._time_of_launch = None
def set_start_time(self, time):
assert self._time_of_registration is None
self._time_of_registration = time
def set_launch_time(self, time):
assert self._time_of_launch is None
self._time_of_launch = time
def start_time(self):
return self._time_of_registration or 0
def launch_time(self):
return self._time_of_launch or 0