com.github.stephanarts.cas.ticket.registry.provider.GetMethodTest.java Source code

Introduction

Here is the source code for com.github.stephanarts.cas.ticket.registry.provider.GetMethodTest.java

Source

/*
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published
 * by the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.github.stephanarts.cas.ticket.registry.provider;

import java.util.HashMap;

import org.junit.Test;
import org.junit.Ignore;
import org.junit.Assert;
import org.junit.runner.RunWith;
import org.junit.runners.JUnit4;

import java.io.ObjectOutputStream;

import org.json.JSONObject;

import org.jasig.cas.ticket.Ticket;
import org.jasig.cas.ticket.ServiceTicket;

import com.github.stephanarts.cas.ticket.registry.provider.GetMethod;
import com.github.stephanarts.cas.ticket.registry.support.IMethod;
import com.github.stephanarts.cas.ticket.registry.support.JSONRPCException;

import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import static org.mockito.Mockito.withSettings;
import static org.mockito.Matchers.*;

import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

/**
 * Unit test for GetMethod.
 */
@RunWith(PowerMockRunner.class)
@PrepareForTest(GetMethod.class)
public class GetMethodTest {
    @Test
    public void testValidInput() throws Exception {
        final HashMap<Integer, Ticket> map = new HashMap<Integer, Ticket>();
        final JSONObject params = new JSONObject();
        final IMethod method = new GetMethod(map);
        final JSONObject result;

        final String ticketId = "ST-1234567890ABCDEFGHIJKL-crud";
        final ServiceTicket ticket = mock(ServiceTicket.class, withSettings().serializable());
        when(ticket.getId()).thenReturn(ticketId);

        map.put(ticketId.hashCode(), ticket);

        params.put("ticket-id", ticketId);

        try {
            result = method.execute(params);
        } catch (final JSONRPCException e) {
            Assert.fail(e.getMessage());
        } catch (final Exception e) {
            throw new Exception(e);
        }
    }

    @Test
    public void testInvalidParameters1() throws Exception {
        final HashMap<Integer, Ticket> map = new HashMap<Integer, Ticket>();
        final JSONObject params = new JSONObject();
        final IMethod method = new GetMethod(map);

        final String ticketId = "ST-1234567890ABCDEFGHIJKL-crud";

        params.put("ticket-id", ticketId);
        params.put("invalid-param", "MUST_FAIL");

        try {
            method.execute(params);
        } catch (final JSONRPCException e) {
            Assert.assertEquals(-32602, e.getCode());
            Assert.assertTrue(e.getMessage().equals("Invalid Params"));
            return;
        }

        Assert.fail("No Exception Thrown");
    }

    @Test
    public void testInvalidParameters2() throws Exception {
        final HashMap<Integer, Ticket> map = new HashMap<Integer, Ticket>();
        final JSONObject params = new JSONObject();
        final IMethod method = new GetMethod(map);

        params.put("invalid-param", "MUST_FAIL");

        try {
            method.execute(params);
        } catch (final JSONRPCException e) {
            Assert.assertEquals(-32602, e.getCode());
            Assert.assertTrue(e.getMessage().equals("Invalid Params"));
            return;
        }

        Assert.fail("No Exception Thrown");
    }

    @Test
    public void testMissingTicket() throws Exception {
        final HashMap<Integer, Ticket> map = new HashMap<Integer, Ticket>();
        final JSONObject params = new JSONObject();
        final IMethod method = new GetMethod(map);

        final String ticketId = "ST-1234567890ABCDEFGHIJKL-crud";

        params.put("ticket-id", ticketId);

        try {
            method.execute(params);
        } catch (final JSONRPCException e) {
            Assert.assertEquals(-32503, e.getCode());
            Assert.assertTrue(e.getMessage().equals("Missing Ticket"));
            return;
        }

        Assert.fail("No Exception Thrown");
    }

    @Test
    public void testSerializationError() throws Exception {
        final HashMap<Integer, Ticket> map = new HashMap<Integer, Ticket>();
        final JSONObject params = new JSONObject();
        final IMethod method = new GetMethod(map);
        final JSONObject result;

        final String ticketId = "ST-1234567890ABCDEFGHIJKL-crud";
        final ServiceTicket ticket = mock(ServiceTicket.class, withSettings().serializable());

        PowerMockito.whenNew(ObjectOutputStream.class).withAnyArguments().thenThrow(new Exception("broken"));

        map.put(ticketId.hashCode(), ticket);

        params.put("ticket-id", ticketId);

        try {
            result = method.execute(params);
        } catch (final JSONRPCException e) {
            Assert.assertEquals(-32500, e.getCode());
            Assert.assertTrue(e.getMessage().equals("Error extracting Ticket"));
            return;
        } catch (final Exception e) {
            throw new Exception(e);
        }

        Assert.fail("No Exception Thrown");
    }

}