RestHealthCheckSummaryFAILTest.java example

Explorer
cougar-master
/*
 * Copyright 2013, The Sporting Exchange Limited
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

// Originally from UpdatedComponentTests/HealthCheck/Rest/Rest_HealthCheck_Summary_FAIL.xls;
package com.betfair.cougar.tests.updatedcomponenttests.healthcheck.rest;

import com.betfair.testing.utils.cougar.misc.XMLHelpers;
import com.betfair.testing.utils.JSONHelpers;
import com.betfair.testing.utils.cougar.assertions.AssertionUtils;
import com.betfair.testing.utils.cougar.beans.HttpCallBean;
import com.betfair.testing.utils.cougar.beans.HttpResponseBean;
import com.betfair.testing.utils.cougar.manager.CougarManager;
import org.json.JSONObject;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;
import org.w3c.dom.Document;

import javax.xml.parsers.DocumentBuilderFactory;
import java.io.ByteArrayInputStream;

/**
 * Ensure that when a Cougar container is running a service that has a component in a FAIL state, the heathcheck summary operation returns FAIL status
 */
public class RestHealthCheckSummaryFAILTest {
    @Test
    public void v3() throws Exception {
        // Set up the Http Call Bean to make the baseline service request
        CougarManager cougarManager1 = CougarManager.getInstance();
        HttpCallBean getNewHttpCallBean1 = cougarManager1.getNewHttpCallBean();
        try {
            getNewHttpCallBean1.setOperationName("setHealthStatusInfo");

            getNewHttpCallBean1.setServiceName("baseline", "cougarBaseline");

            getNewHttpCallBean1.setVersion("v2");
            // Set the component statuses to be set (With FAIL status)
            getNewHttpCallBean1.setRestPostQueryObjects(DocumentBuilderFactory.newInstance().newDocumentBuilder().parse(new ByteArrayInputStream("<message><initialiseHealthStatusObject>true</initialiseHealthStatusObject><DBConnectionStatusDetail>OK</DBConnectionStatusDetail><cacheAccessStatusDetail>FAIL</cacheAccessStatusDetail><serviceStatusDetail>OK</serviceStatusDetail></message>".getBytes())));
            // Make the REST call to the set the health statuses
            cougarManager1.makeRestCougarHTTPCall(getNewHttpCallBean1, com.betfair.testing.utils.cougar.enums.CougarMessageProtocolRequestTypeEnum.RESTXML);
            // Set up the Http Call Bean to make the healthcheck service request
            HttpCallBean getNewHttpCallBean7 = cougarManager1.getNewHttpCallBean();

            getNewHttpCallBean7.setOperationName("isHealthy", "summary");

            getNewHttpCallBean7.setServiceName("healthcheck");

            getNewHttpCallBean7.setVersion("v3");

            getNewHttpCallBean7.setNameSpaceServiceName("Health");
            // Make the 4 REST calls to the get the health status summary from the health service
            cougarManager1.makeRestCougarHTTPCalls(getNewHttpCallBean7);
            // Create expected response as XML document and JSON object (FAIL)
            XMLHelpers xMLHelpers2 = new XMLHelpers();
            Document expectedXML = xMLHelpers2.getXMLObjectFromString("<IsHealthyResponse><HealthSummaryResponse><healthy>FAIL</healthy></HealthSummaryResponse></IsHealthyResponse>");

            JSONHelpers jSONHelpers3 = new JSONHelpers();
            JSONObject expectedJSON = jSONHelpers3.createAsJSONObject(new JSONObject("{\"healthy\":\"FAIL\"}"));
            // Validate the response for each REST call
            HttpResponseBean response4 = getNewHttpCallBean7.getResponseObjectsByEnum(com.betfair.testing.utils.cougar.enums.CougarMessageProtocolResponseTypeEnum.RESTXMLXML);
            AssertionUtils.multiAssertEquals(expectedXML, response4.getResponseObject());
            AssertionUtils.multiAssertEquals((int) 200, response4.getHttpStatusCode());
            AssertionUtils.multiAssertEquals("OK", response4.getHttpStatusText());

            HttpResponseBean response5 = getNewHttpCallBean7.getResponseObjectsByEnum(com.betfair.testing.utils.cougar.enums.CougarMessageProtocolResponseTypeEnum.RESTJSONJSON);
            AssertionUtils.multiAssertEquals(expectedJSON, response5.getResponseObject());
            AssertionUtils.multiAssertEquals((int) 200, response5.getHttpStatusCode());
            AssertionUtils.multiAssertEquals("OK", response5.getHttpStatusText());

            HttpResponseBean response6 = getNewHttpCallBean7.getResponseObjectsByEnum(com.betfair.testing.utils.cougar.enums.CougarMessageProtocolResponseTypeEnum.RESTXMLJSON);
            AssertionUtils.multiAssertEquals(expectedJSON, response6.getResponseObject());
            AssertionUtils.multiAssertEquals((int) 200, response6.getHttpStatusCode());
            AssertionUtils.multiAssertEquals("OK", response6.getHttpStatusText());

            HttpResponseBean response7 = getNewHttpCallBean7.getResponseObjectsByEnum(com.betfair.testing.utils.cougar.enums.CougarMessageProtocolResponseTypeEnum.RESTJSONXML);
            AssertionUtils.multiAssertEquals(expectedXML, response7.getResponseObject());
            AssertionUtils.multiAssertEquals((int) 200, response7.getHttpStatusCode());
            AssertionUtils.multiAssertEquals("OK", response7.getHttpStatusText());
        }
        finally {
            getNewHttpCallBean1 = cougarManager1.getNewHttpCallBean();
            // set back to healthy once we're done...
            getNewHttpCallBean1.setOperationName("setHealthStatusInfo");

            getNewHttpCallBean1.setServiceName("baseline", "cougarBaseline");

            getNewHttpCallBean1.setVersion("v2");
            // Set the component statuses to be set (With FAIL status)
            getNewHttpCallBean1.setRestPostQueryObjects(DocumentBuilderFactory.newInstance().newDocumentBuilder().parse(new ByteArrayInputStream("<message><initialiseHealthStatusObject>true</initialiseHealthStatusObject><DBConnectionStatusDetail>OK</DBConnectionStatusDetail><cacheAccessStatusDetail>OK</cacheAccessStatusDetail><serviceStatusDetail>OK</serviceStatusDetail></message>".getBytes())));
            // Make the REST call to the set the health statuses
            cougarManager1.makeRestCougarHTTPCall(getNewHttpCallBean1, com.betfair.testing.utils.cougar.enums.CougarMessageProtocolRequestTypeEnum.RESTXML);
        }
    }

}