#include "SRModel_test.h"
#include "../../common/printQtTypes.h"
#include "../../../src/dictionary/DicRecord.h"
#include "../../../src/dictionary/Card.h"
#include "../../../src/study/StudySettings.h"
#include "../../mocks/RandomGenerator_mock.h"

SRModelTest::SRModelTest():
    recordId(0),
    pack(&dict),
    randomGenerator(new MockRandomGenerator),
    model(&pack, randomGenerator)
{
    field1 = new Field("English", "Normal");
    field2 = new Field("Russian", "Normal");
    pack.addField(field1);
    pack.addField(field2);
    dict.addCardPack(&pack);
}

void SRModelTest::SetUp()
{
    QCoreApplication::setOrganizationName("freshmemory");
    QCoreApplication::setApplicationName("freshmemory");
    QSettings::setPath(QSettings::IniFormat, QSettings::UserScope, "");
    StudySettings::inst()->load();
}

void SRModelTest::TearDown()
{
    delete field1;
    delete field2;
}

TEST_F(SRModelTest, EmptyStats)
    {
    ASSERT_EQ(0, model.estimatedNewReviewedCardsToday());
    ASSERT_EQ(0, model.countTodayRemainingCards());
    }

TEST_F(SRModelTest, FirstStudy)
    {
    QString cardName = createCard();
    StudyRecord newStudy = pack.getStudyRecord(cardName);

    ASSERT_EQ(StudyRecord::Unknown, newStudy.grade);
    ASSERT_DOUBLE_EQ(StudySettings::inst()->initEasiness, newStudy.easiness);
    ASSERT_DOUBLE_EQ(0, newStudy.interval);
    }

QString SRModelTest::createCard()
{
    addRecord();
    pack.generateQuestions();
    return model.getCurCard()->getQuestion();
}

QString SRModelTest::addRecord()
{
    DicRecord* record = new DicRecord;
    QString recordName = QString::number(recordId);
    record->setField("English", recordName);
    record->setField("Russian", "Odin");
    recordId++;
    dict.addRecord(record);
    return recordName;
}