package limitedFs

import (
	"fmt"
	"testing"

	"github.com/spf13/afero"
)

func TestLimitedCreate(t *testing.T) {
	t.Parallel()

	memfs := afero.NewMemMapFs()
	lfs := NewLimitedFs(memfs, 10)

	for i := 0; i < 11; i++ {
		_, err := lfs.Create(fmt.Sprintf("file.%d", i))
		if err != nil {
			t.Fatalf("unexpected error: %v", err)
		}
	}
}

func TestLimitedWrite(t *testing.T) {
	t.Parallel()

	memfs := afero.NewMemMapFs()
	lfs := NewLimitedFs(memfs, 10)

	file, err := lfs.Create("file")
	if err != nil {
		t.Fatalf("unexpected error: %v", err)
	}

	for i := 0; i < 11; i++ {
		_, err = file.Write([]byte("1"))
		if i < 10 && err != nil {
			t.Fatalf("unexpected error: %v", err)
		}
		if i == 10 && err == nil {
			t.Fatal("expected error, got nil")
		}
	}
}