test_main.py 文件源码

def test_main__run_backup_ami(self, main_log_handler_mock, debug_log_handler_mock, mkdir_mock):
        """Test that _run_backup() runs correctly when ami is specified."""
        instanceid = self.setup_ec2()
        responses.add(responses.GET, "http://169.254.169.254/latest/meta-data/placement/availability-zone",
                      body="us-east-1a", status=200)
        responses.add(responses.GET, "http://169.254.169.254/latest/meta-data/instance-id", body=instanceid,
                      status=200)
        ec2rl_prediag_test = ec2rlcore.main.Main(debug=True, full_init=True)
        ec2rl_prediag_test.options.global_args["backup"] = "ami"
        with contextlib.redirect_stdout(self.output):
            self.assertTrue(ec2rl_prediag_test._run_backup())
        self.assertTrue(re.match(r"^\n-----------\[Backup\s{2}Creation\]-----------\n\nCreating AMI "
                                 r"ami-[a-z0-9]{8} for i-[a-z0-9]{8}\n$",
                                 self.output.getvalue(), re.M))
        self.assertTrue(main_log_handler_mock.called)
        self.assertTrue(debug_log_handler_mock.called)
        self.assertTrue(mkdir_mock.called)