import os

from bbs import BugSearcher, CaseRunner, util, cmd


class CaseRunnerImpl(CaseRunner):
    """
    Extends CaseRunner， run and check should be implemented
    """
    def __init__(self):
        super(CaseRunnerImpl, self).__init__()
        self.log_file = 'log.log'

    def run(self, commit_id=""):
        """
        Prepare work and running case，including setting environment parameter and everything needed
        Args:
            commit_id (string): The commit id string.
        """
        os.environ['XXX'] = 'XXX'
        self.log_file = 'log_{}.log'.format(commit_id)
        cmd.execute_cmd_quiet('python -m pytest -s xxx > {} 2>&1'.format(self.log_file))

    def check(self):
        """
        Check whether the case pass
        returns:
            bool, whether the case passed.
        """
        return cmd.execute_cmd_quiet('grep "1 passed" {}'.format(self.log_file))
        pass


def search():
    case_runner = CaseRunnerImpl()
    project_path = "/XXX/mindspore/"
    hardware_info = [util.kAscendPlatformName, util.kArmArchName, util.kLinuxOsName]
    latest_pass_commit = "XXX"
    earliest_fail_commit = "XXX"
    bug_searcher = BugSearcher(case_runner, project_path, hardware_info, latest_pass_commit,
                               earliest_fail_commit=earliest_fail_commit)
    bug_searcher.search_bug()


if __name__ == '__main__':
    search()