001/*
002 * Licensed to DuraSpace under one or more contributor license agreements.
003 * See the NOTICE file distributed with this work for additional information
004 * regarding copyright ownership.
005 *
006 * DuraSpace licenses this file to you under the Apache License,
007 * Version 2.0 (the "License"); you may not use this file except in
008 * compliance with the License.  You may obtain a copy of the License at
009 *
010 *     http://www.apache.org/licenses/LICENSE-2.0
011 *
012 * Unless required by applicable law or agreed to in writing, software
013 * distributed under the License is distributed on an "AS IS" BASIS,
014 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
015 * See the License for the specific language governing permissions and
016 * limitations under the License.
017 */
018package org.fcrepo.http.commons.exceptionhandlers;
019
020import org.fcrepo.kernel.api.exception.TombstoneException;
021import org.fcrepo.kernel.api.models.Tombstone;
022import org.junit.Before;
023import org.junit.Test;
024import org.mockito.Mock;
025
026import javax.ws.rs.core.Link;
027import javax.ws.rs.core.Response;
028import javax.ws.rs.ext.ExceptionMapper;
029
030import static javax.ws.rs.core.Response.Status.GONE;
031import static org.junit.Assert.assertEquals;
032import static org.mockito.MockitoAnnotations.initMocks;
033
034/**
035 * @author cabeer
036 */
037public class TombstoneExceptionMapperTest {
038
039    private ExceptionMapper<TombstoneException> testObj;
040
041    @Mock
042    public Tombstone mockTombstone;
043
044    @Before
045    public void setUp() {
046        initMocks(this);
047
048        testObj = new TombstoneExceptionMapper();
049    }
050
051    @Test
052    public void testUrilessException() {
053        final Response response = testObj.toResponse(new TombstoneException(mockTombstone));
054        assertEquals(GONE.getStatusCode(), response.getStatus());
055    }
056
057    @Test
058    public void testExceptionWithUri() {
059        final Response response = testObj.toResponse(new TombstoneException(mockTombstone, "some:uri"));
060        assertEquals(GONE.getStatusCode(), response.getStatus());
061        final Link link = Link.valueOf(response.getHeaderString("Link"));
062        assertEquals("some:uri", link.getUri().toString());
063        assertEquals("hasTombstone", link.getRel());
064    }
065
066}