org.apache.juneau.rest.test.RestClientTest.java Source code

Introduction

Here is the source code for org.apache.juneau.rest.test.RestClientTest.java

Source

// ***************************************************************************************************************************
// * Licensed to the Apache Software Foundation (ASF) under one or more contributor license agreements.  See the NOTICE file *
// * distributed with this work for additional information regarding copyright ownership.  The ASF licenses this file        *
// * to you 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.                                              *
// ***************************************************************************************************************************
package org.apache.juneau.rest.test;

import static org.apache.juneau.rest.test.TestUtils.*;
import static org.junit.Assert.*;

import java.util.*;
import java.util.regex.*;

import org.apache.http.entity.*;
import org.apache.juneau.rest.client.*;
import org.junit.*;

public class RestClientTest extends RestTestcase {

    private static String URL = "/testRestClient";

    //====================================================================================================
    // successPattern()
    //====================================================================================================
    @Test
    public void testSuccessPattern() throws Exception {
        RestClient c = TestMicroservice.DEFAULT_CLIENT;
        String r;
        int rc;

        r = c.doPost(URL, new StringEntity("xxxSUCCESSxxx")).successPattern("SUCCESS").getResponseAsString();
        assertEquals("xxxSUCCESSxxx", r);
        rc = c.doPost(URL, new StringEntity("xxxSUCCESSxxx")).successPattern("SUCCESS").run();
        assertEquals(200, rc);

        try {
            r = c.doPost(URL, new StringEntity("xxxFAILURExxx")).successPattern("SUCCESS").getResponseAsString();
            fail();
        } catch (RestCallException e) {
            assertEquals("Success pattern not detected.", e.getLocalizedMessage());
        }
    }

    //====================================================================================================
    // failurePattern()
    //====================================================================================================
    @Test
    public void testFailurePattern() throws Exception {
        RestClient c = TestMicroservice.DEFAULT_CLIENT;
        String r;
        int rc;

        r = c.doPost(URL, new StringEntity("xxxSUCCESSxxx")).failurePattern("FAILURE").getResponseAsString();
        assertEquals("xxxSUCCESSxxx", r);
        rc = c.doPost(URL, new StringEntity("xxxSUCCESSxxx")).failurePattern("FAILURE").run();
        assertEquals(200, rc);

        try {
            r = c.doPost(URL, new StringEntity("xxxFAILURExxx")).failurePattern("FAILURE").getResponseAsString();
            fail();
        } catch (RestCallException e) {
            assertEquals("Failure pattern detected.", e.getLocalizedMessage());
        }

        try {
            r = c.doPost(URL, new StringEntity("xxxERRORxxx")).failurePattern("FAILURE|ERROR")
                    .getResponseAsString();
            fail();
        } catch (RestCallException e) {
            assertEquals("Failure pattern detected.", e.getLocalizedMessage());
        }
    }

    //====================================================================================================
    // captureResponse()/getCapturedResponse()
    //====================================================================================================
    @Test
    public void testCaptureResponse() throws Exception {
        RestClient c = TestMicroservice.DEFAULT_CLIENT;
        RestCall rc = c.doPost(URL, new StringEntity("xxx")).captureResponse();

        try {
            rc.getCapturedResponse();
            fail();
        } catch (IllegalStateException e) {
            assertEquals("This method cannot be called until the response has been consumed.",
                    e.getLocalizedMessage());
        }
        rc.run();
        assertEquals("xxx", rc.getCapturedResponse());
        assertEquals("xxx", rc.getCapturedResponse());

        rc = c.doPost(URL, new StringEntity("xxx")).captureResponse();
        assertEquals("xxx", rc.getResponseAsString());
        assertEquals("xxx", rc.getCapturedResponse());
        assertEquals("xxx", rc.getCapturedResponse());

        try {
            rc.getResponseAsString();
            fail();
        } catch (IllegalStateException e) {
            assertEquals("Method cannot be called.  Response has already been consumed.", e.getLocalizedMessage());
        }
    }

    //====================================================================================================
    // addResponsePattern()
    //====================================================================================================
    @Test
    public void testAddResponsePattern() throws Exception {
        RestClient c = TestMicroservice.DEFAULT_CLIENT;
        String r;

        final List<String> l = new ArrayList<String>();
        ResponsePattern p = new ResponsePattern("x=(\\d+),y=(\\S+)") {
            @Override
            public void onMatch(RestCall restCall, Matcher m) throws RestCallException {
                l.add(m.group(1) + '/' + m.group(2));
            }

            @Override
            public void onNoMatch(RestCall restCall) throws RestCallException {
                throw new RestCallException("Pattern not found!");
            }
        };

        r = c.doPost(URL, new StringEntity("x=1,y=2")).responsePattern(p).getResponseAsString();
        assertEquals("x=1,y=2", r);
        assertObjectEquals("['1/2']", l);

        l.clear();

        r = c.doPost(URL, new StringEntity("x=1,y=2\nx=3,y=4")).responsePattern(p).getResponseAsString();
        assertEquals("x=1,y=2\nx=3,y=4", r);
        assertObjectEquals("['1/2','3/4']", l);

        try {
            c.doPost(URL, new StringEntity("x=1")).responsePattern(p).run();
            fail();
        } catch (RestCallException e) {
            assertEquals("Pattern not found!", e.getLocalizedMessage());
            assertEquals(0, e.getResponseCode());
        }

        // Two patterns!
        ResponsePattern p1 = new ResponsePattern("x=(\\d+)") {
            @Override
            public void onMatch(RestCall restCall, Matcher m) throws RestCallException {
                l.add("x=" + m.group(1));
            }

            @Override
            public void onNoMatch(RestCall restCall) throws RestCallException {
                throw new RestCallException("Pattern x not found!");
            }
        };
        ResponsePattern p2 = new ResponsePattern("y=(\\S+)") {
            @Override
            public void onMatch(RestCall restCall, Matcher m) throws RestCallException {
                l.add("y=" + m.group(1));
            }

            @Override
            public void onNoMatch(RestCall restCall) throws RestCallException {
                throw new RestCallException("Pattern y not found!");
            }
        };

        l.clear();
        r = c.doPost(URL, new StringEntity("x=1,y=2\nx=3,y=4")).responsePattern(p1).responsePattern(p2)
                .getResponseAsString();
        assertEquals("x=1,y=2\nx=3,y=4", r);
        assertObjectEquals("['x=1','x=3','y=2','y=4']", l);

        try {
            c.doPost(URL, new StringEntity("x=1\nx=3")).responsePattern(p1).responsePattern(p2)
                    .getResponseAsString();
        } catch (RestCallException e) {
            assertEquals("Pattern y not found!", e.getLocalizedMessage());
            assertEquals(0, e.getResponseCode());
        }

        try {
            c.doPost(URL, new StringEntity("y=1\ny=3")).responsePattern(p1).responsePattern(p2)
                    .getResponseAsString();
        } catch (RestCallException e) {
            assertEquals("Pattern x not found!", e.getLocalizedMessage());
            assertEquals(0, e.getResponseCode());
        }
    }
}