test_uninstall.py 文件源码

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

    fake_tpr_report = MagicMock()
    fake_tpr_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(third_party.ThirdPartyResourceType, 'create',
                         MagicMock(return_value=fake_tpr_report)):

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