storj/pkg/ranger/file_test.go
2018-09-11 16:57:12 +03:00

117 lines
2.4 KiB
Go

// Copyright (C) 2018 Storj Labs, Inc.
// See LICENSE for copying information.
package ranger
import (
"bytes"
"context"
"io/ioutil"
"os"
"path/filepath"
"strconv"
"testing"
)
func TestFileRanger(t *testing.T) {
tempdir, err := ioutil.TempDir("", "storj-fileranger")
if err != nil {
t.Fatal(err)
}
defer func() {
err := os.RemoveAll(tempdir)
if err != nil {
_ = err // TODO: figure out what is holding the folder open
// t.Fatal(err)
}
}()
for i, example := range []struct {
data string
size, offset, length int64
substr string
fail bool
}{
{"", 0, 0, 0, "", false},
{"abcdef", 6, 0, 0, "", false},
{"abcdef", 6, 3, 0, "", false},
{"abcdef", 6, 0, 6, "abcdef", false},
{"abcdef", 6, 0, 5, "abcde", false},
{"abcdef", 6, 0, 4, "abcd", false},
{"abcdef", 6, 1, 4, "bcde", false},
{"abcdef", 6, 2, 4, "cdef", false},
{"abcdefg", 7, 1, 4, "bcde", false},
{"abcdef", 6, 0, 7, "", true},
{"abcdef", 6, -1, 7, "abcde", true},
{"abcdef", 6, 0, -1, "abcde", true},
} {
fh, err := os.Create(filepath.Join(tempdir, "test"+strconv.Itoa(i)))
if err != nil {
t.Fatalf("failed making tempfile")
}
_, err = fh.Write([]byte(example.data))
if err != nil {
t.Fatalf("failed writing data")
}
name := fh.Name()
err = fh.Close()
if err != nil {
t.Fatalf("failed closing data")
}
rr, err := FileRanger(name)
if err != nil {
t.Fatalf("failed opening tempfile")
}
if rr.Size() != example.size {
t.Fatalf("invalid size: %v != %v", rr.Size(), example.size)
}
r, err := rr.Range(context.Background(), example.offset, example.length)
if example.fail {
if err == nil {
t.Fatalf("expected error")
}
return
}
if err != nil {
t.Fatalf("unexpected err: %v", err)
}
data, err := ioutil.ReadAll(r)
if err != nil {
t.Fatal(err)
}
if !bytes.Equal(data, []byte(example.substr)) {
t.Fatalf("invalid subrange: %#v != %#v", string(data), example.substr)
}
if err := rr.Close(); err != nil {
t.Fatalf("unable to close file %q: %v", name, err)
}
}
}
func TestFileRangerOpenFileError(t *testing.T) {
rr, err := FileRanger("")
if rr != nil {
t.Fatal("Ranger expected to be nil")
}
if err == nil {
t.Fatal("Error expected")
}
}
func TestFileRangerHandlerFileStatError(t *testing.T) {
rr, err := FileHandleRanger(nil)
if rr != nil {
t.Fatal("Ranger expected to be nil")
}
if err == nil {
t.Fatal("Error expected")
}
}