/*
 * rules-test.c
 *
 * Unit tests for the lambda project.
 *
 *  Created on: 28.05.2015
 *      Author: dode@luniks.net
 *
 */

#include "avrjunit.h"
#include "rules.h"
#include "alert.h"
#include "interrupts.h"

/* Module rules */

extern uint16_t age;
extern int8_t dir;
extern uint8_t airgate;
extern Rule rules[];

static bool testAirgate50(void) {

	Measurement meas = {0, 0, 0};

	resetRules();
	dir = -1;
	reason(meas);
	assertFalse(rules[0].fired);

	resetRules();
	dir = 0;
	reason(meas);
	assertFalse(rules[0].fired);

	resetRules();
	dir = 1;
	reason(meas);
	assertFalse(rules[0].fired);

	meas.tempI = 500;

	resetRules();
	dir = -1;
	reason(meas);
	assertFalse(rules[0].fired);

	resetRules();
	dir = 0;
	reason(meas);
	assertFalse(rules[0].fired);

	resetRules();
	dir = 1;
	reason(meas);
	assertTrue(rules[0].fired);
	assertTrue(50 == airgate);

	cancelAlert(false);

	return true;
}

static bool testAirgate25(void) {

	Measurement meas = {999, 0, 0};

	resetRules();
	dir = 1;
	reason(meas);
	assertFalse(rules[1].fired);

	resetRules();
	dir = 0;
	reason(meas);
	assertFalse(rules[1].fired);

	resetRules();
	dir = -1;
	reason(meas);
	assertFalse(rules[1].fired);

	meas.tempI = 799;

	resetRules();
	dir = 1;
	reason(meas);
	assertFalse(rules[1].fired);

	resetRules();
	dir = 0;
	reason(meas);
	assertFalse(rules[1].fired);

	resetRules();
	dir = -1;
	reason(meas);
	assertTrue(rules[1].fired);
	assertTrue(25 == airgate);

	cancelAlert(false);

	return true;
}

static bool testAirgateClose(void) {

	Measurement meas = {999, 0, 0};

	resetRules();
	dir = 1;
	reason(meas);
	assertFalse(rules[2].fired);

	resetRules();
	dir = 0;
	reason(meas);
	assertFalse(rules[2].fired);

	resetRules();
	dir = -1;
	reason(meas);
	assertFalse(rules[2].fired);

	meas.tempI = 399;

	resetRules();
	dir = 1;
	reason(meas);
	assertFalse(rules[2].fired);

	resetRules();
	dir = 0;
	reason(meas);
	assertFalse(rules[2].fired);

	resetRules();
	dir = -1;
	reason(meas);
	assertTrue(rules[2].fired);
	assertTrue(0 == airgate);

	cancelAlert(false);

	return true;
}

static bool testTooRich(void) {

	Measurement meas = {0, 0, 0};
	dir = 0;

	setHeatingState(HEATING_READY);

	meas.tempI = 100;
	meas.lambda = 2000;
	resetRules();
	reason(meas);
	assertFalse(rules[3].fired);

	meas.tempI = 101;
	meas.lambda = 2000;
	resetRules();
	reason(meas);
	assertFalse(rules[3].fired);

	meas.tempI = 100;
	meas.lambda = 1199;
	resetRules();
	reason(meas);
	assertFalse(rules[3].fired);

	meas.tempI = 101;
	meas.lambda = 1199;
	resetRules();
	airgate = 100;
	reason(meas);
	assertFalse(rules[3].fired);
	assertTrue(100 == airgate);

	meas.tempI = 101;
	meas.lambda = 1199;
	resetRules();
	airgate = 50;
	reason(meas);
	assertTrue(rules[3].fired);
	assertTrue(100 == airgate);

	meas.lambda = 1300;
	age = 0;
	reason(meas);
	assertFalse(rules[3].fired);

	cancelAlert(false);

	return true;
}

static bool testTooLean(void) {

	Measurement meas = {0, 0, 0};
	dir = 0;

	setHeatingState(HEATING_READY);

	meas.tempI = 500;
	meas.lambda = 1601;
	resetRules();
	reason(meas);
	assertFalse(rules[4].fired);

	meas.tempI = 501;
	meas.lambda = 1300;
	resetRules();
	reason(meas);
	assertFalse(rules[4].fired);

	meas.tempI = 501;
	meas.lambda = 1601;
	resetRules();
	airgate = 50;
	reason(meas);
	assertFalse(rules[4].fired);
	assertTrue(50 == airgate);

	meas.tempI = 501;
	meas.lambda = 1601;
	resetRules();
	airgate = 100;
	reason(meas);
	assertTrue(rules[4].fired);
	assertTrue(50 == airgate);

	meas.lambda = 1500;
	age = 0;
	reason(meas);
	assertFalse(rules[4].fired);

	cancelAlert(false);

	return true;
}

static bool testFireOut(void) {

	resetRules();
	Measurement meas = {0, 0, 0};
	dir = 1;

	meas.tempI = 50;
	age = 0;
	reason(meas);
	assertFalse(rules[5].fired);

	meas.tempI = 100;
	age = 0;
	reason(meas);
	assertFalse(rules[5].fired);

	meas.tempI = 125;
	age = 0;
	reason(meas);
	assertFalse(rules[5].fired);

	meas.tempI = 99;
	age = 0;
	reason(meas);
	assertTrue(rules[5].fired);

	meas.tempI = 125;
	age = 0;
	reason(meas);
	assertFalse(rules[5].fired);

	cancelAlert(false);

	return true;
}

static bool testHeatingReady(void) {

	resetRules();
	resetTime();
	Measurement meas = {0, 0, 0};
	dir = 1;

	setHeatingOn(false);
	meas.current = 1300;
	reason(meas);
	assertTrue(HEATING_OFF == getHeatingState());

	setHeatingOn(true);
	meas.current = 5000;
	reason(meas);
	assertTrue(HEATING_UP == getHeatingState());

	setHeatingOn(true);
	meas.current = 1300;
	reason(meas);
	assertTrue(HEATING_READY == getHeatingState());

	cancelAlert(false);

	return true;
}

static bool testHeatingFaultNoconn(void) {

	resetRules();
	resetTime();
	Measurement meas = {0, 0, 0};
	dir = 1;

	setHeatingOn(true);
	meas.current = 0;
	reason(meas);
	assertTrue(HEATING_FAULT == getHeatingState());

	cancelAlert(false);

	return true;
}

static bool testHeatingFaultShort(void) {

	resetRules();
	resetTime();
	Measurement meas = {0, 0, 0};
	dir = 1;

	setHeatingOn(true);
	meas.current = 8000;
	reason(meas);
	assertTrue(HEATING_FAULT == getHeatingState());

	cancelAlert(false);

	return true;
}

static bool testHeatingFaultNoheat(void) {

	resetRules();
	resetTime();
	Measurement meas = {0, 0, 0};
	dir = 1;

	setHeatingOn(true);

	// more than 3 mins
	addInts(INTS_PER_SEC * 181);

	meas.current = 5000;
	reason(meas);
	assertTrue(HEATING_FAULT == getHeatingState());

	cancelAlert(false);

	return true;
}

static bool testHeatingTimeout(void) {

	resetRules();
	resetTime();
	Measurement meas = {0, 0, 0};
	dir = 1;

	setHeatingOn(true);

	// more than 3 hours below 400°C
	addInts(INTS_PER_SEC * 10800UL);

	meas.tempI = 300;
	meas.current = 1300;
	reason(meas);
	assertTrue(HEATING_OFF == getHeatingState());

	cancelAlert(false);

	return true;
}

static bool testReasonDirBurnUp(void) {

	resetRules();
	Measurement meas = {0, 0, 2000};

	age = 0;
	reason(meas);
	assertTrue(dir == 0);

	meas.tempI = 9;
	meas.lambda = 2000;
	age = 180;
	reason(meas);
	assertTrue(dir == 0);

	meas.tempI = 100;
	meas.lambda = 2000;
	age = 180;
	reason(meas);
	assertTrue(dir == 1);

	meas.tempI = 790;
	meas.lambda = 2000;
	age = 180;
	reason(meas);
	assertTrue(dir == 1);

	meas.tempI = 790;
	meas.lambda = 2000;
	age = 180;
	reason(meas);
	assertTrue(dir == 0);

	meas.tempI = 800;
	meas.lambda = 2000;
	age = 180;
	reason(meas);
	assertTrue(dir == 1);

	meas.tempI = 850;
	meas.lambda = 2000;
	age = 180;
	reason(meas);
	assertTrue(dir == 0);

	meas.tempI = 900;
	meas.lambda = 1999;
	age = 180;
	reason(meas);
	assertTrue(dir == 0);

	cancelAlert(false);

	return true;
}

static bool testReasonDirBurnDown(void) {

	resetRules();
	Measurement meas = {999, 0, 1999};

	age = 0;
	reason(meas);
	assertTrue(dir == 0);

	meas.tempI = 900;
	meas.lambda = 1999;
	age = 180;
	reason(meas);
	assertTrue(dir == 0);

	meas.tempI = 800;
	meas.lambda = 1999;
	age = 180;
	reason(meas);
	assertTrue(dir == 0);

	meas.tempI = 750;
	meas.lambda = 1999;
	age = 180;
	reason(meas);
	assertTrue(dir == 0);

	meas.tempI = 700;
	meas.lambda = 2000;
	age = 180;
	reason(meas);
	assertTrue(dir == -1);

	meas.tempI = 700;
	meas.lambda = 2000;
	age = 180;
	reason(meas);
	assertTrue(dir == 0);

	meas.tempI = 100;
	meas.lambda = 2000;
	age = 180;
	reason(meas);
	assertTrue(dir == -1);

	cancelAlert(false);

	return true;
}

/* Test "class" */
static const char class[] PROGMEM = "rules";

/* Test names */
static const char testAirgate50_P[] PROGMEM = "testAirgate50";
static const char testAirgate25_P[] PROGMEM = "testAirgate25";
static const char testAirgateClose_P[] PROGMEM = "testAirgateClose";
static const char testTooRich_P[] PROGMEM = "testTooRich";
static const char testTooLean_P[] PROGMEM = "testTooLean";
static const char testFireOut_P[] PROGMEM = "testFireOut";
static const char testHeatingReady_P[] PROGMEM = "testHeatingReady";
static const char testHeatingFaultNoconn_P[] PROGMEM = "testHeatingFaultNoconn";
static const char testHeatingFaultShort_P[] PROGMEM = "testHeatingFaultShort";
static const char testHeatingFaultNoheat_P[] PROGMEM = "testHeatingFaultNoheat";
static const char testHeatingTimeout_P[] PROGMEM = "testHeatingTimeout";
static const char testReasonDirBurnUp_P[] PROGMEM = "testReasonDirBurnUp";
static const char testReasonDirBurnDown_P[] PROGMEM = "testReasonDirBurnDown";

/* Tests */
static TestCase const tests[] = {
		{class, testAirgate50_P, testAirgate50},
		{class, testAirgate25_P, testAirgate25},
		{class, testAirgateClose_P, testAirgateClose},
		{class, testTooRich_P, testTooRich},
		{class, testTooLean_P, testTooLean},
		{class, testFireOut_P, testFireOut},
		{class, testHeatingReady_P, testHeatingReady},
		{class, testHeatingFaultNoconn_P, testHeatingFaultNoconn},
		{class, testHeatingFaultShort_P, testHeatingFaultShort},
		{class, testHeatingFaultNoheat_P, testHeatingFaultNoheat},
		{class, testHeatingTimeout_P, testHeatingTimeout},
		{class, testReasonDirBurnUp_P, testReasonDirBurnUp},
		{class, testReasonDirBurnDown_P, testReasonDirBurnDown}
};

TestClass rulesClass = {tests, sizeof(tests) / sizeof(tests[0])};