org.springframework.security.web.jackson2.WebAuthenticationDetailsMixinTest.java Source code

Introduction

Here is the source code for org.springframework.security.web.jackson2.WebAuthenticationDetailsMixinTest.java

Source

/*
 * Copyright 2015-2016 the original author or authors.
 *
 * 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.
 */

package org.springframework.security.web.jackson2;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.json.JSONException;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.runners.MockitoJUnitRunner;
import org.skyscreamer.jsonassert.JSONAssert;
import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockHttpSession;
import org.springframework.security.web.authentication.WebAuthenticationDetails;

import java.io.IOException;

import static org.assertj.core.api.Assertions.assertThat;

/**
 * @author Jitendra Singh
 */
@RunWith(MockitoJUnitRunner.class)
public class WebAuthenticationDetailsMixinTest {

    ObjectMapper mapper;

    @Before
    public void setup() {
        this.mapper = new ObjectMapper().addMixIn(WebAuthenticationDetails.class,
                WebAuthenticationDetailsMixin.class);
    }

    @Test
    public void buildWebAuthenticationDetailsUsingDifferentConstructors() throws IOException {
        MockHttpServletRequest request = new MockHttpServletRequest();
        request.setRemoteAddr("localhost");
        request.setSession(new MockHttpSession(null, "1"));

        WebAuthenticationDetails details = new WebAuthenticationDetails(request);
        String jsonString = "{\"@class\": \"org.springframework.security.web.authentication.WebAuthenticationDetails\","
                + "\"sessionId\": \"1\", \"remoteAddress\": \"/localhost\"}";
        WebAuthenticationDetails authenticationDetails = this.mapper.readValue(jsonString,
                WebAuthenticationDetails.class);
        assertThat(details.equals(authenticationDetails));
    }

    @Test
    public void webAuthenticationDetailsSerializeTest() throws JsonProcessingException, JSONException {
        MockHttpServletRequest request = new MockHttpServletRequest();
        request.setRemoteAddr("/home");
        request.setSession(new MockHttpSession(null, "1"));
        WebAuthenticationDetails details = new WebAuthenticationDetails(request);
        String expectedJson = "{\"@class\": \"org.springframework.security.web.authentication.WebAuthenticationDetails\","
                + "\"sessionId\": \"1\", \"remoteAddress\": \"/home\"}";
        String actualJson = this.mapper.writeValueAsString(details);
        JSONAssert.assertEquals(expectedJson, actualJson, true);
    }

    @Test
    public void webAuthenticationDetailsDeserializeTest() throws IOException, JSONException {
        String actualJson = "{\"@class\": \"org.springframework.security.web.authentication.WebAuthenticationDetails\","
                + "\"sessionId\": \"1\", \"remoteAddress\": \"/home\"}";
        WebAuthenticationDetails details = this.mapper.readValue(actualJson, WebAuthenticationDetails.class);
        assertThat(details).isNotNull();
        assertThat(details.getRemoteAddress()).isEqualTo("/home");
        assertThat(details.getSessionId()).isEqualTo("1");
    }
}