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.HttpHeaders.LINK;
031import static javax.ws.rs.core.Response.Status.GONE;
032import static org.junit.Assert.assertEquals;
033import static org.mockito.MockitoAnnotations.initMocks;
034
035/**
036 * @author cabeer
037 */
038public class TombstoneExceptionMapperTest {
039
040    private ExceptionMapper<TombstoneException> testObj;
041
042    @Mock
043    public Tombstone mockTombstone;
044
045    @Before
046    public void setUp() {
047        initMocks(this);
048
049        testObj = new TombstoneExceptionMapper();
050    }
051
052    @Test
053    public void testUrilessException() {
054        final Response response = testObj.toResponse(new TombstoneException(mockTombstone));
055        assertEquals(GONE.getStatusCode(), response.getStatus());
056    }
057
058    @Test
059    public void testExceptionWithUri() {
060        final Response response = testObj.toResponse(new TombstoneException(mockTombstone, "some:uri"));
061        assertEquals(GONE.getStatusCode(), response.getStatus());
062        final Link link = Link.valueOf(response.getHeaderString(LINK));
063        assertEquals("some:uri", link.getUri().toString());
064        assertEquals("hasTombstone", link.getRel());
065    }
066
067}