D-Net

package eu.dnetlib.enabling.database;

import static org.junit.Assert.assertEquals;

import static org.junit.Assert.assertFalse;

import static org.junit.Assert.assertTrue;

import static org.mockito.Mockito.verify;

import static org.mockito.Mockito.when;

import java.util.Arrays;

import java.util.Date;

import java.util.HashMap;

import java.util.List;

import java.util.Map;

import org.junit.Before;

import org.junit.Ignore;

import org.junit.Test;

import org.junit.runner.RunWith;

import org.mockito.Mock;

import org.mockito.runners.MockitoJUnitRunner;

import eu.dnetlib.enabling.database.objects.DnetDatabase;

import eu.dnetlib.enabling.database.resultset.SQLResultSetListenerFactory;

import eu.dnetlib.enabling.database.utils.DatabaseUtils;

import eu.dnetlib.enabling.database.utils.GenericRow;

import eu.dnetlib.enabling.resultset.ResultSetFactory;

import eu.dnetlib.enabling.resultset.ResultSetListener;

@RunWith(MockitoJUnitRunner.class)

public class DatabaseServiceCoreTest {

	private static final String DB_NAME = "cinema";

	private static final String TABLE_NAME = "films";

	private static final String TABLE_WHERE = "year=1980";

	private static final Date TABLE_FROM = new Date(100000);

	private static final Date TABLE_UNTIL = new Date(200000);

	private static final String SQL = "SELECT * from films";

	// Class under test

	private DatabaseServiceCore core;

	@Mock

	private DatabaseUtils dbUtils;

	@Mock

	private ResultSetFactory resultSetFactory;

	@Mock

	private SQLResultSetListenerFactory resultSetListenerFactory;

	@Mock

	private ResultSetListener listener_where;

	@Mock

	private ResultSetListener listener_from_until;

	@Mock

	private ResultSetListener listener_sql;

	@Mock

	private GenericRow ROW;

	private final Map<String, Object> ROW_FIELDS = new HashMap<String, Object>();

	@Before

	public void setUp() throws Exception {

		core = new DatabaseServiceCore();

		core.setDbUtils(dbUtils);

		core.setResultSetFactory(resultSetFactory);

		core.setResultSetListenerFactory(resultSetListenerFactory);

		ROW_FIELDS.put("name", "Pippo");

		ROW_FIELDS.put("age", 36);

		when(dbUtils.listCommonDBTables(DB_NAME)).thenReturn(Arrays.asList("T1", "T2", "T3"));;

		when(resultSetListenerFactory.createCondTableListener(DB_NAME, TABLE_NAME, TABLE_WHERE)).thenReturn(listener_where);

		when(resultSetListenerFactory.createLoggedTableListener(DB_NAME, TABLE_NAME, TABLE_FROM, TABLE_UNTIL)).thenReturn(listener_from_until);

		when(resultSetListenerFactory.createSQLListener(DB_NAME, SQL)).thenReturn(listener_sql);

		//when(ROW.getTable()).thenReturn(TABLE_NAME);

		//when(ROW.getFields()).thenReturn(ROW_FIELDS);

	}

	@Ignore

	@Test

	public void testListDatabases() throws Exception {

		List<DnetDatabase> list = core.listDatabases();

		verify(dbUtils).listAllDatabases();

		assertEquals("List size invalid", 5, list.size());

		for (DnetDatabase db : list) {

			if (db.getDbName().startsWith("M")) {

				assertTrue("Db should be managed", db.isManaged());

			} else {

				assertFalse("Db should be not managed", db.isManaged());

			}

		}

	}

	@Test

	public void testChangeDatabaseStatus_manage() throws Exception {

		core.changeDatabaseStatus(DB_NAME, true);

		verify(dbUtils).listCommonDBTables(DB_NAME);

		verify(dbUtils).setManaged(DB_NAME, true);

	}

	@Test

	public void testChangeDatabaseStatus_unmanage() throws Exception {

		core.changeDatabaseStatus(DB_NAME, false);

		verify(dbUtils).listCommonDBTables(DB_NAME);

		verify(dbUtils).setManaged(DB_NAME, false);

	}

	@Test

	public void testGenerateResultSet_1() throws Exception {

		core.generateResultSet(DB_NAME, SQL);

		verify(resultSetFactory).createResultSet(listener_sql);

	}

	@Test

	public void testGenerateResultSet_2() throws Exception {

		core.generateResultSet(DB_NAME, TABLE_NAME, TABLE_WHERE);

		verify(resultSetFactory).createResultSet(listener_where);

	}

	@Test

	public void testGenerateResultSet_3() throws Exception {

		core.generateResultSet(DB_NAME, TABLE_NAME, TABLE_FROM, TABLE_UNTIL);

		verify(resultSetFactory).createResultSet(listener_from_until);

	}

}