45 lines
760 B
Go
45 lines
760 B
Go
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")
|
|
}
|
|
}
|
|
}
|