package engine

import (
	"bytes"
	"context"
	"errors"
	"io/ioutil"
	"os"
	"testing"

	"git.defalsify.org/festive/cache"
	"git.defalsify.org/festive/persist"
	"git.defalsify.org/festive/state"
)

func TestPersist(t *testing.T) {
	generateTestData(t)
	cfg := Config{
		OutputSize: 128,
		SessionId: "xyzzy",
		Root: "root",
	}
	rs := NewFsWrapper(dataDir, nil)

	persistDir, err := ioutil.TempDir("", "festive_engine_persist")
	if err != nil {
		t.Fatal(err)
	}

	st := state.NewState(3)
	ca := cache.NewCache().WithCacheSize(1024)
	pr := persist.NewFsPersister(persistDir).WithContent(&st, ca)

	w := bytes.NewBuffer(nil)
	ctx := context.TODO()

	
	err = RunPersisted(cfg, rs, pr, []byte{}, w, ctx)
	if err != nil {
		if !errors.Is(err, os.ErrNotExist) {
			t.Fatal(err)
		}
		st := state.NewState(cfg.FlagCount)
		ca := cache.NewCache()
		if cfg.CacheSize > 0 {
			ca = ca.WithCacheSize(cfg.CacheSize)
		}
		pr = persist.NewFsPersister(persistDir).WithContent(&st, ca)
		err = pr.Save(cfg.SessionId)
		if err != nil {
			t.Fatal(err)
		}
	}

	pr = persist.NewFsPersister(persistDir)
	inputs := []string{
		"",
		"1",
		"2",
		"00",
		}
	for _, v := range inputs {
		err = RunPersisted(cfg, rs, pr, []byte(v), w, ctx)
		if err != nil {
			t.Fatal(err)
		}
	}
}