117 lines
2.4 KiB
Go
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")
|
|
}
|
|
}
|