/*
 * Hibernate OGM, Domain model persistence for NoSQL datastores
 *
 * License: GNU Lesser General Public License (LGPL), version 2.1 or later
 * See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
 */
package org.hibernate.ogm.datastore.mongodb.test.options.readpreference;

import static org.fest.assertions.Assertions.assertThat;

import java.lang.annotation.ElementType;

import org.hibernate.ogm.datastore.mongodb.MongoDB;
import org.hibernate.ogm.datastore.mongodb.options.ReadPreferenceType;
import org.hibernate.ogm.datastore.mongodb.options.impl.ReadPreferenceOption;
import org.hibernate.ogm.datastore.mongodb.options.navigation.MongoDBGlobalContext;
import org.hibernate.ogm.options.container.impl.OptionsContainer;
import org.hibernate.ogm.options.navigation.impl.AppendableConfigurationContext;
import org.hibernate.ogm.options.navigation.impl.ConfigurationContextImpl;
import org.hibernate.ogm.options.navigation.source.impl.OptionValueSource;
import org.hibernate.ogm.options.navigation.source.impl.ProgrammaticOptionValueSource;
import org.junit.Before;
import org.junit.Test;

import com.mongodb.ReadPreference;

/**
 * Test for setting the {@link ReadPreferenceOption} programmatically.
 *
 * @author Gunnar Morling
 */
public class ReadPreferenceConfiguredProgrammaticallyTest {

	private MongoDBGlobalContext mongoOptions;
	private AppendableConfigurationContext context;

	@Before
	public void setupBuilder() {
		context = new AppendableConfigurationContext();
		mongoOptions = new MongoDB().getConfigurationBuilder( new ConfigurationContextImpl( context ) );
	}

	@Test
	public void testReadPreferenceGivenOnGlobalLevel() throws Exception {
		mongoOptions.readPreference( ReadPreferenceType.SECONDARY );

		OptionsContainer options = getSource().getGlobalOptions();
		assertThat( options.getUnique( ReadPreferenceOption.class ) ).isEqualTo( ReadPreference.secondary() );
	}

	@Test
	public void testReadPreferenceGivenOnEntityLevel() throws Exception {
		mongoOptions
			.entity( MyEntity.class )
				.readPreference( ReadPreferenceType.SECONDARY_PREFERRED );

		OptionsContainer options = getSource().getEntityOptions( MyEntity.class );
		assertThat( options.getUnique( ReadPreferenceOption.class ) ).isEqualTo( ReadPreference.secondaryPreferred() );
	}

	@Test
	public void testReadPreferenceGivenOnPropertyLevel() throws Exception {
		mongoOptions
			.entity( MyEntity.class )
				.property( "content", ElementType.FIELD )
				.readPreference( ReadPreferenceType.NEAREST );

		OptionsContainer options = getSource().getPropertyOptions( MyEntity.class, "content" );
		assertThat( options.getUnique( ReadPreferenceOption.class ) ).isEqualTo( ReadPreference.nearest() );
	}

	private OptionValueSource getSource() {
		return new ProgrammaticOptionValueSource( context );
	}

	@SuppressWarnings("unused")
	private static final class MyEntity {
		String content;
	}
}