org.owasp.webgoat.service.LabelServiceTest.java Source code

Introduction

Here is the source code for org.owasp.webgoat.service.LabelServiceTest.java

Source

package org.owasp.webgoat.service;

import org.assertj.core.util.Maps;
import org.hamcrest.CoreMatchers;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.owasp.webgoat.i18n.LabelProvider;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.test.autoconfigure.web.servlet.WebMvcTest;
import org.springframework.boot.test.mock.mockito.MockBean;
import org.springframework.security.test.context.support.WithMockUser;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.test.web.servlet.MockMvc;
import org.springframework.test.web.servlet.request.MockMvcRequestBuilders;

import java.util.Locale;

import static org.mockito.Mockito.when;
import static org.owasp.webgoat.service.LabelService.URL_LABELS_MVC;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;

/**
 * ************************************************************************************************
 * This file is part of WebGoat, an Open Web Application Security Project utility. For details,
 * please see http://www.owasp.org/
 * <p>
 * Copyright (c) 2002 - 20014 Bruce Mayhew
 * <p>
 * This program is free software; you can redistribute it and/or modify it under the terms of the
 * GNU General Public License as published by the Free Software Foundation; either version 2 of the
 * License, or (at your option) any later version.
 * <p>
 * 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
 * General Public License for more details.
 * <p>
 * You should have received a copy of the GNU General Public License along with this program; if
 * not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA
 * 02111-1307, USA.
 * <p>
 * Getting Source ==============
 * <p>
 * Source for this application is maintained at https://github.com/WebGoat/WebGoat, a repository for free software
 * projects.
 * <p>
 *
 * @author nbaars
 * @version $Id: $Id
 * @since November 29, 2016
 */
@WebMvcTest(value = { LabelService.class, LabelProvider.class })
@RunWith(SpringRunner.class)
public class LabelServiceTest {

    @Autowired
    public MockMvc mockMvc;
    @MockBean
    private LabelProvider labelProvider;

    @Test
    @WithMockUser(username = "guest", password = "guest")
    public void withoutLocale() throws Exception {
        when(labelProvider.getLabels(Locale.ENGLISH)).thenReturn(Maps.newHashMap("key", "value"));
        mockMvc.perform(MockMvcRequestBuilders.get(URL_LABELS_MVC)).andExpect(status().isOk())
                .andExpect(jsonPath("key", CoreMatchers.is("value")));
    }

    @Test
    @WithMockUser(username = "guest", password = "guest")
    public void withLocale() throws Exception {
        when(labelProvider.getLabels(Locale.GERMAN)).thenReturn(Maps.newHashMap("key", "value"));
        mockMvc.perform(MockMvcRequestBuilders.get(URL_LABELS_MVC).param("lang", "de")).andExpect(status().isOk())
                .andExpect(jsonPath("key", CoreMatchers.is("value")));
    }
}