/*
 * Copyright 2008-2017 by Emeric Vernat
 *
 *     This file is part of Java Melody.
 *
 * 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
 *
 *     http://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 net.bull.javamelody;

import static org.easymock.EasyMock.createNiceMock;
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.replay;
import static org.easymock.EasyMock.verify;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;

import org.apache.log4j.Logger;
import org.bson.Document;
import org.bson.codecs.configuration.CodecRegistry;
import org.junit.Before;
import org.junit.Test;

import com.mongodb.client.MongoCollection;
import com.mongodb.client.MongoDatabase;

/**
 * Test unitaire de la classe MongoWrapper.
 * @author Emeric Vernat
 */
public class TestMongoWrapper {
	/** Test. */
	@Before
	public void setUp() {
		Utils.initialize();
	}

	/** Test. */
	@SuppressWarnings("unchecked")
	@Test
	public void testCreateDatabaseProxy() {
		try {
			Class.forName("com.mongodb.ReadPreference");
		} catch (final ClassNotFoundException e) {
			Logger.getRootLogger().info(e.toString());
			// si mongodb-driver-core n'est pas disponible dans le classpath (test depuis Ant),
			// on ne peut pas exécuter ce test
			return;
		}
		final MongoDatabase database = createNiceMock(MongoDatabase.class);
		final MongoCollection<Document> collection = createNiceMock(MongoCollection.class);
		final CodecRegistry codecRegistry = createNiceMock(CodecRegistry.class);
		expect(database.withCodecRegistry(codecRegistry)).andReturn(database).anyTimes();
		expect(database.getCollection("collection")).andReturn(collection).anyTimes();
		expect(database.getName()).andReturn("db").anyTimes();
		expect(collection.withCodecRegistry(codecRegistry)).andReturn(collection).anyTimes();

		replay(database);
		replay(collection);
		replay(codecRegistry);
		final MongoDatabase databaseProxy = MongoWrapper.createDatabaseProxy(database);
		assertNotNull("createDatabaseProxy", databaseProxy);
		assertNotNull("databaseProxy", databaseProxy.withCodecRegistry(codecRegistry));
		assertNotNull("getCollection", databaseProxy.getCollection("collection"));
		assertEquals("getName", "db", databaseProxy.getName());

		assertNotNull("collectionProxy",
				databaseProxy.getCollection("collection").withCodecRegistry(codecRegistry));
		databaseProxy.getCollection("collection").getCodecRegistry();
		databaseProxy.getCollection("collection").find();
		verify(database);
		verify(collection);
		verify(codecRegistry);
	}
}