#ifndef SPACED_REPETITION_MODEL_TEST_H
#define SPACED_REPETITION_MODEL_TEST_H

#include <gtest/gtest.h>
#include <iostream>
#include <QtCore>

#include "../../../src/study/SpacedRepetitionModel.h"
#include "../../../src/dictionary/CardPack.h"
#include "../../mocks/Dictionary_mock.h"

using std::ostream;

class MockRandomGenerator;
class Field;

class TestSpacedRepetitionModel: public SpacedRepetitionModel
{
public:
    TestSpacedRepetitionModel(CardPack* pack, IRandomGenerator* random):
        SpacedRepetitionModel(pack, random) {}
    void testPickCard() { pickNextCardAndNotify(); }
};

class SRModelTest: public testing::Test
{
public:
    SRModelTest();

protected:
    void SetUp();
    void TearDown();
    QString createCard();
    QString addRecord();

protected:
    MockDictionary dict;
    int recordId;
    Field* field1;
    Field* field2;
    CardPack pack;
    MockRandomGenerator* randomGenerator;
    TestSpacedRepetitionModel model;
};

#endif