/*
 * Copyright 2004-present 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
 *
 *      https://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 java.util.stream.Stream;

import com.fasterxml.jackson.databind.ObjectMapper;
import org.junit.jupiter.api.BeforeEach;
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.CoreJackson2Module;
import org.springframework.security.jackson2.SecurityJackson2Modules;
import org.springframework.security.jackson2.SimpleGrantedAuthorityMixinTests;
import org.springframework.security.web.authentication.switchuser.SwitchUserGrantedAuthority;

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

/**
 * @author Markus Heiden
 * @since 6.3
 */
@SuppressWarnings("removal")
public class SwitchUserGrantedAuthorityMixInTests {

	// language=JSON
	private static final String SWITCH_JSON = """
			{
				"@class": "org.springframework.security.web.authentication.switchuser.SwitchUserGrantedAuthority",
				"role": "switched",
				"source": {
					"@class": "org.springframework.security.authentication.UsernamePasswordAuthenticationToken",
					"principal": "principal",
					"credentials": "credentials",
					"authenticated": true,
					"details": null,
					"authorities": %s
				}
			}
			""".formatted(SimpleGrantedAuthorityMixinTests.AUTHORITIES_ARRAYLIST_JSON);

	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"));
	}

	@ParameterizedTest
	@MethodSource("mappers")
	public void serializeWhenPrincipalCredentialsAuthoritiesThenSuccess(ObjectMapper mapper) throws Exception {
		SwitchUserGrantedAuthority expected = new SwitchUserGrantedAuthority("switched", this.source);
		String serializedJson = mapper.writeValueAsString(expected);
		JSONAssert.assertEquals(SWITCH_JSON, serializedJson, true);
	}

	@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);
	}

}
