deck_manager_test.go

package leaf

import (
	"io/ioutil"
	"os"
	"testing"
	"time"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/require"
)

func TestDeckManager(t *testing.T) {
	tmpfile, err := ioutil.TempFile("", "leaf.db")
	require.NoError(t, err)
	defer os.Remove(tmpfile.Name())

	db, err := OpenBoltStore(tmpfile.Name())
	require.NoError(t, err)

	dm, err := NewDeckManager("./fixtures", db, OutputFormatOrg)
	require.NoError(t, err)

	t.Run("ReviewDecks", func(t *testing.T) {
		decks, err := dm.ReviewDecks()
		require.NoError(t, err)
		require.Len(t, decks, 2)

		deck := decks[0]
		assert.Equal(t, "Hiragana", deck.Name)
		assert.Equal(t, 46, deck.CardsReady)
		assert.InDelta(t, time.Since(deck.NextReviewAt), 0, float64(time.Minute))
	})

	t.Run("ReviewSession", func(t *testing.T) {
		session, err := dm.ReviewSession("Hiragana")
		require.NoError(t, err)
		assert.Equal(t, 20, session.Total())
		assert.Equal(t, []string{"for-tests"}, session.Sides())

		question := session.Next()
		require.NoError(t, session.Again())
		err = db.RangeStats("Hiragana", SRSSupermemo2PlusCustom, func(card string, s *Stats) bool {
			if card != question {
				return true
			}

			sm := s.SRSAlgorithm.(*Supermemo2PlusCustom)
			assert.InDelta(t, 0.45, sm.Difficulty, 0.01)
			assert.InDelta(t, 0.2, sm.Interval, 0.01)
			return false
		})

		require.NoError(t, err)
	})

	t.Run("DeckStats", func(t *testing.T) {
		stats, err := dm.DeckStats("Hiragana")
		require.NoError(t, err)
		assert.Len(t, stats, 46)

		s := stats[0]
		assert.NotEmpty(t, s.Question)
		sm := s.SRSAlgorithm.(*Supermemo2PlusCustom)
		assert.InDelta(t, 0.3, sm.Difficulty, 0.01)
	})
}