from typing import Dict


class SingleTestData(object):
    def __init__(self, name, testruns, testrail_case_id, geth_paths):
        self.testrail_case_id = testrail_case_id
        self.name = name
        self.testruns = testruns
        self.geth_paths = geth_paths

    class TestRunData(object):
        def __init__(self, steps, jobs, error, first_commands: Dict[str, int]):
            self.steps = steps
            self.jobs = jobs
            self.error = error
            self.first_commands = first_commands

    def create_new_testrun(self):
        self.testruns.append(SingleTestData.TestRunData(list(), dict(), None, dict()))


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, list())
            self.tests.append(test)
            self.current_test = test