spring.travel.site.auth.CookieEncoderTest.java Source code

Introduction

Here is the source code for spring.travel.site.auth.CookieEncoderTest.java

Source

/**
 * Copyright 2014 Andy Godwin
 *
 * 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 spring.travel.site.auth;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.mockito.runners.MockitoJUnitRunner;
import org.springframework.test.util.ReflectionTestUtils;

import java.util.LinkedHashMap;
import java.util.Map;

import static org.junit.Assert.assertEquals;
import static org.mockito.Matchers.anyString;
import static org.mockito.Mockito.when;

@RunWith(MockitoJUnitRunner.class)
public class CookieEncoderTest {

    @Mock
    private Signer mockSigner;

    @Mock
    private Verifier mockVerifier;

    private CookieEncoder encoder;

    @Before
    public void before() {
        encoder = new CookieEncoder();
        ReflectionTestUtils.setField(encoder, "signer", mockSigner);
    }

    @Test
    public void shouldEncodeAndSignMap() throws Exception {
        ReflectionTestUtils.setField(encoder, "cookieName", "TEST_SESSION");

        String signature = "9834793875298375295";
        Map<String, String> values = new LinkedHashMap<>();
        values.put("foo", "bar");
        values.put("moose", "antlers");

        when(mockSigner.sign(anyString())).thenReturn(signature);
        String cookie = encoder.encode(values);
        assertEquals("TEST_SESSION=" + signature + "-foo=bar&moose=antlers", cookie);
    }

    @Test
    public void shouldUrlEncodeKeysAndValues() throws Exception {
        ReflectionTestUtils.setField(encoder, "cookieName", "MOOSE_SESSION");

        String signature = "9834793875298375295";
        Map<String, String> values = new LinkedHashMap<>();
        values.put("mo ose", "ant@lers");

        when(mockSigner.sign(anyString())).thenReturn(signature);
        String cookie = encoder.encode(values);
        assertEquals("MOOSE_SESSION=" + signature + "-mo+ose=ant%40lers", cookie);
    }
}