test_uninstall.py 文件源码

def test_remove_report_crd_success2(caplog):
    fake_http_resp = FakeHTTPResponse(500, "{\"message\":\"fake message\"}",
                                      "{\"reason\":\"NotFound\"}")
    fake_api_exception = K8sApiException(http_resp=fake_http_resp)

    fake_crd_report = MagicMock()
    fake_crd_report.remove.side_effect = fake_api_exception

    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=fake_crd_report)):

        uninstall.remove_report_crd("cmk-nodereport", ["cmk-nr"])
        caplog_tuple = caplog.record_tuples
        assert \
            caplog_tuple[-2][2] == "\"cmk-nodereport\" for node \"{}\" does "\
                                   "not exist.".format(os.getenv("NODE_NAME"))
        assert \
            caplog_tuple[-1][2] == "\"cmk-nodereport\" for node \"{}\" " \
                                   "removed.".format(os.getenv("NODE_NAME"))