/*
 * MarkovAutomatonParserTest.cpp
 *
 *  Created on: 25.02.2014
 *      Author: Manuel Sascha Weiand
 */

#include "gtest/gtest.h"
#include "storm-config.h"

#include "src/parser/MarkovAutomatonParser.h"
#include "src/exceptions/FileIoException.h"

TEST(MarkovAutomatonParserTest, NonExistingFile) {

	// No matter what happens, please do NOT create a file with the name "nonExistingFile.not"!
	ASSERT_THROW(storm::parser::MarkovAutomatonParser::parseMarkovAutomaton(STORM_CPP_TESTS_BASE_PATH "/nonExistingFile.not", STORM_CPP_TESTS_BASE_PATH "/nonExistingFile.not", STORM_CPP_TESTS_BASE_PATH "/nonExistingFile.not"), storm::exceptions::FileIoException);
}

TEST(MarkovAutomatonParserTest, BasicParsing) {

	// Get the parsing result.
	storm::models::MarkovAutomaton<double> result = storm::parser::MarkovAutomatonParser::parseMarkovAutomaton(STORM_CPP_TESTS_BASE_PATH "/functional/parser/tra_files/ma_general.tra", STORM_CPP_TESTS_BASE_PATH "/functional/parser/lab_files/ma_general.lab", STORM_CPP_TESTS_BASE_PATH "/functional/parser/rew_files/ma_general.state.rew");

	// Test sizes and counts.
	ASSERT_EQ(result.getNumberOfStates(), 6);
	ASSERT_EQ(result.getNumberOfChoices(), 7);
	ASSERT_EQ(result.getNumberOfTransitions(), 12);

	// Test the exit rates. These have to be 0 for all non-Markovian states.
	std::vector<double> rates = result.getExitRates();
	ASSERT_EQ(result.getExitRate(0), 2);
	ASSERT_FALSE(result.isMarkovianState(1));
	ASSERT_EQ(result.getExitRate(1), 0);
	ASSERT_EQ(result.getExitRate(2), 15);
	ASSERT_FALSE(result.isMarkovianState(3));
	ASSERT_EQ(result.getExitRate(3), 0);
	ASSERT_FALSE(result.isMarkovianState(4));
	ASSERT_EQ(result.getExitRate(4), 0);
	ASSERT_FALSE(result.isMarkovianState(5));
	ASSERT_EQ(result.getExitRate(5), 0);

	// Test the labeling.
	ASSERT_EQ(result.getStateLabeling().getNumberOfAtomicPropositions(), 3);
	ASSERT_EQ(result.getInitialStates().getNumberOfSetBits(), 1);
	ASSERT_EQ(result.getLabelsForState(4).size(), 0);
	ASSERT_EQ(result.getStateLabeling().getLabeledStates("goal").getNumberOfSetBits(), 1);

	// Test the state rewards.
	ASSERT_TRUE(result.hasStateRewards());
	double rewardSum = 0;
	for(uint_fast64_t i = 0; i < result.getStateRewardVector().size(); i++) {
		rewardSum += result.getStateRewardVector()[i];
	}
	ASSERT_EQ(rewardSum, 1015.765099984);
	ASSERT_EQ(result.getStateRewardVector()[0], 0);

	// Test the transition rewards.
	ASSERT_FALSE(result.hasTransitionRewards());
}

TEST(MarkovAutomatonParserTest, MismatchedFiles) {

	// Test file combinations that do not match, i.e. differing number of states, transitions, etc.

	// The labeling file contains a label for a non existent state.
	ASSERT_THROW(storm::parser::MarkovAutomatonParser::parseMarkovAutomaton(STORM_CPP_TESTS_BASE_PATH "/functional/parser/tra_files/ma_general.tra", STORM_CPP_TESTS_BASE_PATH "/functional/parser/lab_files/ma_mismatched.lab"), storm::exceptions::OutOfRangeException);

	// The state reward file contains a reward for a non existent state.
	ASSERT_THROW(storm::parser::MarkovAutomatonParser::parseMarkovAutomaton(STORM_CPP_TESTS_BASE_PATH "/functional/parser/tra_files/ma_general.tra", STORM_CPP_TESTS_BASE_PATH "/functional/parser/lab_files/ma_general.lab", STORM_CPP_TESTS_BASE_PATH "/functional/parser/rew_files/ma_mismatched.state.rew"), storm::exceptions::OutOfRangeException);
}