service_test.go

// Copyright 2019 The Cockroach Authors.
//
// Use of this software is governed by the CockroachDB Software License
// included in the /LICENSE file.

package blobs

import (
"context"
"os"
"path/filepath"
"testing"

"github.com/cockroachdb/cockroach/pkg/blobs/blobspb"
"github.com/cockroachdb/cockroach/pkg/testutils"
"github.com/cockroachdb/errors/oserror"
)

funcTestBlobServiceList(t*testing.T) {
tmpDir, cleanupFn:=testutils.TempDir(t)
defercleanupFn()

fileContent:= []byte("a")
files:= []string{"/file/dir/a.csv", "/file/dir/b.csv", "/file/dir/c.csv"}
for_, file:=rangefiles {
writeTestFile(t, filepath.Join(tmpDir, file), fileContent)
 }

service, err:=NewBlobService(tmpDir)
iferr!=nil {
t.Fatal(err)
 }
ctx:=context.Background()

t.Run("list-correct-files", func(t*testing.T) {
resp, err:=service.List(ctx, &blobspb.GlobRequest{
Pattern: "file/dir/*.csv",
 })
iferr!=nil {
t.Fatal(err)
 }
resultList:=resp.Files
iflen(resultList) !=len(files) {
t.Fatal("result list does not have the correct number of files")
 }
fori, f:=rangeresultList {
iff!=files[i] {
t.Fatalf("result list is incorrect %s", resultList)
 }
 }
 })
t.Run("not-in-external-io-dir", func(t*testing.T) {
_, err:=service.List(ctx, &blobspb.GlobRequest{
Pattern: "file/../../*.csv",
 })
iferr==nil {
t.Fatal("expected error but was not caught")
 }
if!testutils.IsError(err, "outside of external-io-dir is not allowed") {
t.Fatal("incorrect error message: "+err.Error())
 }
 })
}

funcTestBlobServiceDelete(t*testing.T) {
tmpDir, cleanupFn:=testutils.TempDir(t)
defercleanupFn()

fileContent:= []byte("file_content")
filename:="path/to/file/content.txt"
writeTestFile(t, filepath.Join(tmpDir, filename), fileContent)

service, err:=NewBlobService(tmpDir)
iferr!=nil {
t.Fatal(err)
 }
ctx:=context.Background()

t.Run("delete-correct-file", func(t*testing.T) {
_, err:=service.Delete(ctx, &blobspb.DeleteRequest{
Filename: filename,
 })
iferr!=nil {
t.Fatal(err)
 }
if_, err:=os.Stat(filepath.Join(tmpDir, filename)); !oserror.IsNotExist(err) {
t.Fatalf("expected not exists err, got: %s", err)
 }
 })
t.Run("file-not-exist", func(t*testing.T) {
_, err:=service.Delete(ctx, &blobspb.DeleteRequest{
Filename: "file/does/not/exist",
 })
iferr==nil {
t.Fatal("expected error but was not caught")
 }
if!testutils.IsError(err, "no such file") {
t.Fatal("incorrect error message: "+err.Error())
 }
 })
t.Run("not-in-external-io-dir", func(t*testing.T) {
_, err:=service.Delete(ctx, &blobspb.DeleteRequest{
Filename: "file/../../content.txt",
 })
iferr==nil {
t.Fatal("expected error but was not caught")
 }
if!testutils.IsError(err, "outside of external-io-dir is not allowed") {
t.Fatal("incorrect error message: "+err.Error())
 }
 })
}

funcTestBlobServiceStat(t*testing.T) {
tmpDir, cleanupFn:=testutils.TempDir(t)
defercleanupFn()

fileContent:= []byte("file_content")
filename:="path/to/file/content.txt"
writeTestFile(t, filepath.Join(tmpDir, filename), fileContent)

service, err:=NewBlobService(tmpDir)
iferr!=nil {
t.Fatal(err)
 }
ctx:=context.Background()

t.Run("get-correct-file-size", func(t*testing.T) {
resp, err:=service.Stat(ctx, &blobspb.StatRequest{
Filename: filename,
 })
iferr!=nil {
t.Fatal(err)
 }
ifresp.Filesize!=int64(len(fileContent)) {
t.Fatalf("expected filesize: %d, got %d", len(fileContent), resp.Filesize)
 }
 })
t.Run("file-not-exist", func(t*testing.T) {
_, err:=service.Stat(ctx, &blobspb.StatRequest{
Filename: "file/does/not/exist",
 })
iferr==nil {
t.Fatal("expected error but was not caught")
 }
if!testutils.IsError(err, "no such file") {
t.Fatal("incorrect error message: "+err.Error())
 }
 })
t.Run("not-in-external-io-dir", func(t*testing.T) {
_, err:=service.Stat(ctx, &blobspb.StatRequest{
Filename: "file/../../content.txt",
 })
iferr==nil {
t.Fatal("expected error but was not caught")
 }
if!testutils.IsError(err, "outside of external-io-dir is not allowed") {
t.Fatal("incorrect error message: "+err.Error())
 }
 })
t.Run("stat-directory", func(t*testing.T) {
_, err:=service.Stat(ctx, &blobspb.StatRequest{
Filename: filepath.Dir(filename),
 })
iferr==nil {
t.Fatalf("expected error but was not caught")
 }
if!testutils.IsError(err, "expected a file") {
t.Fatal("incorrect error message: "+err.Error())
 }
 })
}