class SingleTestData(object): def __init__(self, name, testruns, testrail_case_id): self.testrail_case_id = testrail_case_id self.name = name self.testruns = testruns class TestRunData(object): def __init__(self, steps, jobs, error): self.steps = steps self.jobs = jobs self.error = error def create_new_testrun(self): self.testruns.append(SingleTestData.TestRunData(list(), list(), None)) class TestSuiteData(object): def __init__(self): self.apk_name = None self.current_test = None self.tests = list() def set_current_test(self, test_name, testrail_case_id): existing_test = next((test for test in self.tests if test.name == test_name), None) if existing_test: self.current_test = existing_test else: test = SingleTestData(test_name, list(), testrail_case_id) self.tests.append(test) self.current_test = test