Add SwitchUserGrantedAuthority to Web Jackson Module

Closes gh-17041

Signed-off-by: John Niang <johnniang@foxmail.com>
This commit is contained in:
John Niang
2025-05-08 11:44:47 +08:00
committed by Josh Cummings
parent 8e2067bb3e
commit 9ba5c7b2ce
3 changed files with 46 additions and 16 deletions

View File

@@ -1,5 +1,5 @@
/*
* Copyright 2002-2024 the original author or authors.
* Copyright 2002-2025 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.
@@ -16,14 +16,20 @@
package org.springframework.security.web.jackson2;
import java.util.stream.Stream;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.params.ParameterizedTest;
import org.junit.jupiter.params.provider.Arguments;
import org.junit.jupiter.params.provider.MethodSource;
import org.skyscreamer.jsonassert.JSONAssert;
import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.authority.AuthorityUtils;
import org.springframework.security.jackson2.AbstractMixinTests;
import org.springframework.security.jackson2.CoreJackson2Module;
import org.springframework.security.jackson2.SecurityJackson2Modules;
import org.springframework.security.jackson2.SimpleGrantedAuthorityMixinTests;
import org.springframework.security.web.authentication.switchuser.SwitchUserGrantedAuthority;
@@ -33,7 +39,7 @@ import static org.assertj.core.api.Assertions.assertThat;
* @author Markus Heiden
* @since 6.3
*/
public class SwitchUserGrantedAuthorityMixInTests extends AbstractMixinTests {
public class SwitchUserGrantedAuthorityMixInTests {
// language=JSON
private static final String SWITCH_JSON = """
@@ -53,22 +59,42 @@ public class SwitchUserGrantedAuthorityMixInTests extends AbstractMixinTests {
private Authentication source;
static Stream<Arguments> mappers() {
ObjectMapper securityJackson2ModulesMapper = new ObjectMapper();
ClassLoader classLoader = SwitchUserGrantedAuthorityMixInTests.class.getClassLoader();
securityJackson2ModulesMapper.registerModules(SecurityJackson2Modules.getModules(classLoader));
ObjectMapper webJackson2ModuleMapper = new ObjectMapper();
webJackson2ModuleMapper.registerModule(new CoreJackson2Module());
webJackson2ModuleMapper.registerModule(new WebJackson2Module());
ObjectMapper webServletJackson2ModuleMapper = new ObjectMapper();
webServletJackson2ModuleMapper.registerModule(new CoreJackson2Module());
webServletJackson2ModuleMapper.registerModule(new WebServletJackson2Module());
return Stream.of(Arguments.of(securityJackson2ModulesMapper), Arguments.of(webJackson2ModuleMapper),
Arguments.of(webServletJackson2ModuleMapper));
}
@BeforeEach
public void setUp() {
this.source = new UsernamePasswordAuthenticationToken("principal", "credentials",
AuthorityUtils.createAuthorityList("ROLE_USER"));
}
@Test
public void serializeWhenPrincipalCredentialsAuthoritiesThenSuccess() throws Exception {
@ParameterizedTest
@MethodSource("mappers")
public void serializeWhenPrincipalCredentialsAuthoritiesThenSuccess(ObjectMapper mapper) throws Exception {
SwitchUserGrantedAuthority expected = new SwitchUserGrantedAuthority("switched", this.source);
String serializedJson = this.mapper.writeValueAsString(expected);
String serializedJson = mapper.writeValueAsString(expected);
JSONAssert.assertEquals(SWITCH_JSON, serializedJson, true);
}
@Test
public void deserializeWhenSourceIsUsernamePasswordAuthenticationTokenThenSuccess() throws Exception {
SwitchUserGrantedAuthority deserialized = this.mapper.readValue(SWITCH_JSON, SwitchUserGrantedAuthority.class);
@ParameterizedTest
@MethodSource("mappers")
public void deserializeWhenSourceIsUsernamePasswordAuthenticationTokenThenSuccess(ObjectMapper mapper)
throws Exception {
SwitchUserGrantedAuthority deserialized = mapper.readValue(SWITCH_JSON, SwitchUserGrantedAuthority.class);
assertThat(deserialized).isNotNull();
assertThat(deserialized.getAuthority()).isEqualTo("switched");
assertThat(deserialized.getSource()).isEqualTo(this.source);