/*
 * Copyright 2007 JBoss Inc
 * 
 * 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.
 *
 * Created on Dec 14, 2007
 */
package org.drools.integrationtests;

import static org.mockito.Matchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;

import java.io.IOException;
import java.util.concurrent.TimeUnit;

import org.drools.CommonTestMethodBase;
import org.drools.KnowledgeBase;
import org.drools.KnowledgeBaseConfiguration;
import org.drools.KnowledgeBaseFactory;
import org.drools.StockTick;
import org.drools.StockTickInterface;
import org.drools.compiler.DroolsParserException;
import org.drools.conf.EventProcessingOption;
import org.drools.event.rule.AfterActivationFiredEvent;
import org.drools.event.rule.AgendaEventListener;
import org.drools.runtime.KnowledgeSessionConfiguration;
import org.drools.runtime.StatefulKnowledgeSession;
import org.drools.runtime.conf.ClockTypeOption;
import org.drools.time.SessionClock;
import org.drools.time.impl.PseudoClockScheduler;
import org.junit.Test;

/**
 * Tests related to the pseudo session clock
 */
public class PseudoClockEventsTest extends CommonTestMethodBase {

	int evalFirePseudoClockStockCount = 15;
	private static final String evalFirePseudoClockDeclaration = "package org.drools\n" +
			"\n" +
			"declare StockTick\n" +
			"    @role( event )\n" +
			"end\n\n";
	private static final String evalFirePseudoClockRuleA = "rule A\n"
			+
			"when\n"
			+
			"	$a: StockTick( $priceA: price )\n"
			+
			"	$b: StockTick( $priceA < price )\n"
			+
			"then \n"
			+
			"    //System.out.println(\"Rule A fired by thread \" + Thread.currentThread().getName() + \": \" + $a + \", \" + $b);\n"
			+
			"end\n" +
			"";
	private static final String evalFirePseudoClockRuleB = "rule B\n" +
			"when\n" +
			"	$a: StockTick()\n" +
			"	not( StockTick( this after[1,10s] $a ) )\n" +
			"then \n" +
			"    //System.out.println(\"Rule B fired by thread \" + Thread.currentThread().getName());\n" +
			"end\n" +
			"";

	@Test(timeout = 6000)
	public void testEvenFirePseudoClockRuleA() throws Exception {

		AgendaEventListener ael = mock(AgendaEventListener.class);

		processStocks(evalFirePseudoClockStockCount, ael,
				evalFirePseudoClockDeclaration + evalFirePseudoClockRuleA);

		verify(ael,
				times(evalFirePseudoClockStockCount * (evalFirePseudoClockStockCount - 1) / 2)).afterActivationFired(
				any(AfterActivationFiredEvent.class));
	}

	@Test(timeout = 6000)
	public void testEvenFirePseudoClockRuleB() throws Exception {

		AgendaEventListener ael = mock(AgendaEventListener.class);

		processStocks(evalFirePseudoClockStockCount, ael,
				evalFirePseudoClockDeclaration + evalFirePseudoClockRuleB);

		verify(ael,
				times(evalFirePseudoClockStockCount - 1)).afterActivationFired(
				any(AfterActivationFiredEvent.class));
	}

	@Test(timeout = 60000)
	public void testEvenFirePseudoClockRulesAB() throws Exception {

		AgendaEventListener ael = mock(AgendaEventListener.class);

		processStocks(evalFirePseudoClockStockCount, ael,
				evalFirePseudoClockDeclaration + evalFirePseudoClockRuleA + evalFirePseudoClockRuleB);

		final int expectedActivationCount = evalFirePseudoClockStockCount * (evalFirePseudoClockStockCount - 1) / 2
				+ evalFirePseudoClockStockCount - 1;
		verify(ael,
				times(expectedActivationCount)).afterActivationFired(
				any(AfterActivationFiredEvent.class));
	}

	private int processStocks(int stockCount, AgendaEventListener agendaEventListener, String drlContentString)
			throws DroolsParserException, IOException, Exception {
		KnowledgeBaseConfiguration kconf = KnowledgeBaseFactory.newKnowledgeBaseConfiguration();
		kconf.setOption(EventProcessingOption.STREAM);
		KnowledgeBase kbase = loadKnowledgeBaseFromString(kconf, drlContentString);

		KnowledgeSessionConfiguration ksessionConfig = KnowledgeBaseFactory.newKnowledgeSessionConfiguration();
		ksessionConfig.setOption(ClockTypeOption.get("pseudo"));
		ksessionConfig.setProperty("keep.reference", "true");
		final StatefulKnowledgeSession ksession = kbase.newStatefulKnowledgeSession(ksessionConfig,
				null);
		ksession.addEventListener(agendaEventListener);

		PseudoClockScheduler clock = (PseudoClockScheduler) ksession.<SessionClock> getSessionClock();

		Runnable fireUntilHaltRunnable = new Runnable() {
			public void run() {
				ksession.fireUntilHalt();
			}
		};
		Thread fireUntilHaltThread = new Thread(fireUntilHaltRunnable);
		fireUntilHaltThread.start();

		for (int stIndex = 1; stIndex <= stockCount; stIndex++) {
			clock.advanceTime(20, TimeUnit.SECONDS);
			final StockTickInterface st = new StockTick(stIndex,
					"RHT",
					100 * stIndex,
					1000);
			ksession.insert(st);
			Thread.sleep(1);
		}

		Thread.sleep(2000);
		ksession.halt();

		return stockCount;
	}

}