test_main.py 文件源码

def setUp(self, main_log_handler_mock, debug_log_handler_mock, mkdir_mock):
        sys.argv = ["test/modules/not_a_real_file", "run", "--abc=def"]
        os.chdir(self.callpath)
        os.environ["EC2RL_SUDO"] = "False"
        os.environ["EC2RL_DISTRO"] = "alami"
        os.environ["EC2RL_NET_DRIVER"] = "ixgbevf"
        os.environ["EC2RL_VIRT_TYPE"] = "hvm"
        with contextlib.redirect_stdout(StringIO()):
            self.ec2rl = ec2rlcore.main.Main(debug=True, full_init=True)
            self.ec2rl.full_init()

        self.assertTrue(debug_log_handler_mock.called)
        self.assertTrue(main_log_handler_mock.called)
        self.assertTrue(mkdir_mock.called)

        self.output = StringIO()