test_uninstall.py 文件源码

def test_remove_all_report_crd_success(caplog):
    mock = MagicMock()
    mock.remove.return_value = 0

    with patch('kubernetes.config.load_incluster_config',
               MagicMock(return_value=0)), \
            patch('kubernetes.client.ExtensionsV1beta1Api',
                  MagicMock(return_value=0)), \
            patch.object(custom_resource.CustomResourceDefinitionType,
                         'create',
                         MagicMock(return_value=mock)):
        uninstall.remove_all_report()
        caplog_tuple = caplog.record_tuples
        assert caplog_tuple[-1][2] == "\"cmk-reconcilereport\" for node " \
                                      "\"{}\" removed."\
            .format(os.getenv("NODE_NAME"))
        assert caplog_tuple[-3][2] == "\"cmk-nodereport\" for node \"{}\" " \
                                      "removed.".format(os.getenv("NODE_NAME"))