pathin_test.go

package pathin

import (
	"errors"
	"strconv"
	"testing"

	"github.com/stretchr/testify/assert"
)

type bucketInfo struct {
	bucketId int
	userId   int
}

func TestNewFS(t *testing.T) {
	newFs(t)
}

func BenchmarkNewFS(b *testing.B) {
	b.ReportAllocs()

	for i := 0; i < b.N; i++ {
		newFs(b)
	}
}

func newFs(tb testing.TB) {
	myFs := New("bucket-name")
	assert.Equal(tb, myFs.Name(), "bucket-name")

	inBucketDest := myFs.AddDestGroup("companyBucket", groupHandler)
	inBucketDest.AddDest("cad-files", RawHandler)

	inUserDest := inBucketDest.AddDestGroup("userBucket", userHandler)
	inUserDest.AddDest("profile-pictures", RawHandler)

	path, err := myFs.GetPath("profile-pictures", &bucketInfo{bucketId: 974, userId: 941})
	assert.NoError(tb, err)

	assert.Equal(tb, "/buckets/bucket_974/users/941/profile-pictures", path)
}

func groupHandler(typeName string, values interface{}) (string, error) {
	info, ok := values.(*bucketInfo)
	if ok == false || info == nil {
		return "", errors.New("No readable data")
	}

	if info.bucketId > 0 {
		return "/buckets/bucket_" + strconv.Itoa(info.bucketId), nil
	}

	return "", errors.New("No bucket Id defined")
}

func userHandler(typeName string, values interface{}) (string, error) {
	info, ok := values.(*bucketInfo)
	if ok == false || info == nil {
		return "", errors.New("No readable data")
	}

	if info.userId > 0 {
		return "users/" + strconv.Itoa(info.userId), nil
	}

	return "", errors.New("No user Id defined")
}