/*
 * Copyright 2019-2023 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.data.mapping;

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

import org.junit.jupiter.api.Test;
import org.springframework.data.mapping.context.SampleMappingContext;
import org.springframework.data.mapping.context.SamplePersistentProperty;
import org.springframework.data.mapping.model.EntityInstantiators;
import org.springframework.data.mapping.model.InstantiationAwarePropertyAccessor;

/**
 * Unit tests for {@link InstantiationAwarePropertyAccessor}.
 *
 * @author Oliver Drotbohm
 * @author Mark Paluch
 */
class InstantiationAwarePersistentPropertyAccessorUnitTests {

	@Test // DATACMNS-1639
	void shouldCreateNewInstance() {

		var instantiators = new EntityInstantiators();
		var context = new SampleMappingContext();

		PersistentEntity<Object, SamplePersistentProperty> entity = context.getRequiredPersistentEntity(Sample.class);

		var sample = new Sample("Dave", "Matthews", 42);

		PersistentPropertyAccessor<Sample> wrapper = new InstantiationAwarePropertyAccessor<>(sample,
				entity::getPropertyAccessor, instantiators);

		wrapper.setProperty(entity.getRequiredPersistentProperty("firstname"), "Oliver August");

		assertThat(wrapper.getBean()).isEqualTo(new Sample("Oliver August", "Matthews", 42));
	}

	@Test // DATACMNS-1768
	void shouldSetMultipleProperties() {

		var instantiators = new EntityInstantiators();
		var context = new SampleMappingContext();

		PersistentEntity<Object, SamplePersistentProperty> entity = context.getRequiredPersistentEntity(Sample.class);

		var bean = new Sample("Dave", "Matthews", 42);

		PersistentPropertyAccessor<Sample> wrapper = new InstantiationAwarePropertyAccessor<>(bean,
				entity::getPropertyAccessor, instantiators);

		wrapper.setProperty(entity.getRequiredPersistentProperty("firstname"), "Oliver August");
		wrapper.setProperty(entity.getRequiredPersistentProperty("lastname"), "Heisenberg");

		assertThat(wrapper.getBean()).isEqualTo(new Sample("Oliver August", "Heisenberg", 42));
	}

	@Test // GH-2625
	void shouldSetPropertyOfRecordUsingCanonicalConstructor() {

		var instantiators = new EntityInstantiators();
		var context = new SampleMappingContext();

		PersistentEntity<Object, SamplePersistentProperty> entity = context
				.getRequiredPersistentEntity(WithSingleArgConstructor.class);

		var bean = new WithSingleArgConstructor(42L, "Dave");

		PersistentPropertyAccessor<WithSingleArgConstructor> wrapper = new InstantiationAwarePropertyAccessor<>(bean,
				entity::getPropertyAccessor, instantiators);

		wrapper.setProperty(entity.getRequiredPersistentProperty("name"), "Oliver August");
		wrapper.setProperty(entity.getRequiredPersistentProperty("id"), 41L);

		assertThat(wrapper.getBean()).isEqualTo(new WithSingleArgConstructor(41L, "Oliver August"));
	}

	record Sample(String firstname, String lastname, int age) {

	}

	public record WithSingleArgConstructor(Long id, String name) {

		public WithSingleArgConstructor(String name) {
			this(null, name);
		}
	}

}
