util_test.go

package tdb

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

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

func TestRandBytes(t *testing.T) {
	assert.Len(t, randBytes(10), 10, "bytes length should be 10")
}

func BenchmarkRandBytes(b *testing.B) {
	for i := 0; i < b.N; i++ {
		randBytes(10)
	}
}

func TestFolderActions(t *testing.T) {
	demoFolder := "util_test_folder"
	demoFile := "util_test_file"

	exist, err := checkFolderExist(demoFolder)
	assert.NoError(t, err, "should have no error")
	assert.False(t, exist, "folder should not exist")

	var f *os.File
	f, err = os.Create(demoFile)
	if assert.NoError(t, err, "should have no error to create file") {
		defer func() {
			f.Close()
			err := os.Remove(demoFile)
			assert.NoError(t, err, "should have no error to delete file")
		}()
	}

	exist, err = checkFolderExist(demoFile)
	assert.Equal(t, ErrPathNotFolder, err, "should have error when path is not folder")
	assert.False(t, exist, "should return false if it is a file")

	err = createFolder(demoFolder)
	if assert.NoError(t, err, "should have no error to create folder") {
		defer func() {
			err := os.RemoveAll(demoFolder)
			assert.NoError(t, err, "should have no error to delete folder")
		}()
	}

	err = createFolder(demoFolder)
	assert.NoError(t, err, "should have no error if folder exist")

	exist, err = checkFolderExist(demoFolder)
	assert.NoError(t, err, "should have no error when path is a folder")
	assert.True(t, exist, "should return true if folder exist")
}

func TestGetFileName(t *testing.T) {
	fPath := filepath.Join("201702", "050.idx")
	assert.Equal(t, "050", getFileName(fPath), "file name wrong")
}