TestAbstractWrapper.java

/**
*    Copyright (c) 2011-2014, OpenIoT
*   
*    This file is part of OpenIoT.
*
*    OpenIoT is free software: you can redistribute it and/or modify
*    it under the terms of the GNU Lesser General Public License as published by
*    the Free Software Foundation, version 3 of the License.
*
*    OpenIoT is distributed in the hope that it will be useful,
*    but WITHOUT ANY WARRANTY; without even the implied warranty of
*    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
*    GNU Lesser General Public License for more details.
*
*    You should have received a copy of the GNU Lesser General Public License
*    along with OpenIoT.  If not, see <http://www.gnu.org/licenses/>.
*
*     Contact: OpenIoT mailto: info@openiot.eu
 * @author Mehdi Riahi
 * @author Ali Salehi
 * @author Timotee Maret
*/

package org.openiot.gsn.wrappers;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertTrue;

import org.openiot.gsn.Main;
import org.openiot.gsn.beans.AddressBean;
import org.openiot.gsn.beans.DataField;
import org.openiot.gsn.beans.InputStream;
import org.openiot.gsn.beans.StreamElement;
import org.openiot.gsn.beans.StreamSource;
import org.openiot.gsn.storage.StorageManager;
import org.openiot.gsn.storage.StorageManagerFactory;
import org.openiot.gsn.utils.GSNRuntimeException;
import org.openiot.gsn.utils.KeyValueImp;

import java.io.Serializable;
import java.sql.DriverManager;
import java.sql.ResultSet;
import java.sql.SQLException;

import javax.naming.OperationNotSupportedException;

import org.junit.AfterClass;
import org.junit.BeforeClass;
import org.junit.Test;

public class TestAbstractWrapper {

	@BeforeClass
	public static void setUpBeforeClass() throws Exception {
		DriverManager.registerDriver( new org.h2.Driver( ) );

		sm = StorageManagerFactory.getInstance( "org.h2.Driver","sa","" ,"jdbc:h2:mem:gsn_mem_db", Main.DEFAULT_MAX_DB_CONNECTIONS);
//		StorageManager.getInstance ( ).initialize ( "com.mysql.jdbc.Driver","root","" , "jdbc:mysql://localhost/org.openiot.gsn");
	}

	@AfterClass
	public static void tearDownAfterClass() throws Exception {
	}

	private static StorageManager sm;
	

	//@Before
	//public void setUp() throws Exception {
	//	sm = StorageManager.getInstance();
	//	}


	@Test (expected=OperationNotSupportedException.class)
	public void testSendToWrapper1() throws OperationNotSupportedException {
		SystemTime systemTimeWrapper = new SystemTime();
		systemTimeWrapper.sendToWrapper("bla");
	}
	/**
	 * Test method for {@link org.openiot.gsn.wrappers.AbstractWrapper#sendToWrapper(java.lang.Object)}.
	 * Test to see what is the behavior if the wrapper is disabled.
	 * @throws OperationNotSupportedException 
	 * @throws SQLException 
	 */
	@Test (expected=GSNRuntimeException.class)
	public void testSendToWrapper2() throws OperationNotSupportedException, SQLException {
		SystemTime systemTimeWrapper = new SystemTime();
		systemTimeWrapper.setActiveAddressBean(new AddressBean("system-time"));
		assertTrue(systemTimeWrapper.initialize());
		Thread thread = new Thread (systemTimeWrapper);
		thread.start();
		systemTimeWrapper.releaseResources();
		systemTimeWrapper.sendToWrapper("bla");
	}

	@Test
	public void testRemovingUselessData() throws SQLException, InterruptedException {
		SystemTime wrapper = new SystemTime();
		sm.executeCreateTable(wrapper.getDBAliasInStr(), new DataField[] {},true);
		wrapper.setActiveAddressBean(new AddressBean("system-time",new KeyValueImp(SystemTime.CLOCK_PERIOD_KEY,"100")));
		assertTrue(wrapper.initialize());
		Thread thread = new Thread(wrapper);
		InputStream is = new InputStream();
		StreamSource  ss = new StreamSource().setAlias("my-stream").setAddressing(new AddressBean[] {new AddressBean("system-time")}).setSqlQuery("select * from wrapper where TIMED <0").setRawHistorySize("2").setInputStream(is);   
		ss.setSamplingRate(1);
		ss.setWrapper(wrapper );
		assertTrue(ss.validate());
		assertEquals(wrapper.getTimerClockPeriod(), 100);
		thread.start();
		Thread.sleep(1000);
		
		ResultSet rs =sm.executeQueryWithResultSet(new StringBuilder("select count(*) from ").append(wrapper.getDBAliasInStr()), sm.getConnection());
		assertTrue(rs.next());
		//    System.out.println(rs.getInt(1));
		assertTrue(rs.getInt(1)<=(AbstractWrapper.GARBAGE_COLLECT_AFTER_SPECIFIED_NO_OF_ELEMENTS*2));
		wrapper.releaseResources();
	}
	
	@Test
	public void testOutOfOrderData() throws SQLException, InterruptedException{
		MockWrapper wrapper = new MockWrapper();
		sm.executeCreateTable(wrapper.getDBAliasInStr(), wrapper.getOutputFormat(),true);
		AddressBean addressBean = new AddressBean("test");
		wrapper.setActiveAddressBean(addressBean);
		assertTrue(wrapper.initialize());
		InputStream is = new InputStream();
		
		StreamSource  ss = new StreamSource().setAlias("my-stream").setAddressing(new AddressBean[] {addressBean}).setSqlQuery("select * from wrapper").setRawHistorySize("1").setInputStream(is);   
		ss.setSamplingRate(1);
		ss.setWrapper(wrapper );
		assertTrue(ss.validate());

		StreamElement se = new StreamElement(wrapper.getOutputFormat(), new Serializable[]{1000}, 1000L);
		assertTrue(wrapper.insertIntoWrapperTable(se));
		se = new StreamElement(wrapper.getOutputFormat(), new Serializable[]{2000}, 2000L);
		assertTrue(wrapper.insertIntoWrapperTable(se));
		se = new StreamElement(wrapper.getOutputFormat(), new Serializable[]{1500}, 1500L);
		assertFalse(wrapper.insertIntoWrapperTable(se));
		
		wrapper.releaseResources();
	}

}