files_test.go

package files

import (
	"bytes"
	"github.com/stretchr/testify/assert"
	"os"
	"path"
	"testing"
)

var tempFolder = os.TempDir()
var fileData = []byte("aaaaaaaaaaaabbbbbbbbbbbbbbbbbbbaeeeeeeeeeeeeeeaaaaaa")
var fileData2 = []byte("bbbbbbbbbbbb66666666666666666666664bb")

func TestWriteRemove(t *testing.T) {
	fileName := "testfile"
	write(t, fileName, fileData)
	remove(t, fileName)
}

func TestWriteReadRemove(t *testing.T) {
	fileName := "testfile"
	write(t, fileName, fileData)
	readAndCheck(t, fileName, fileData)
	remove(t, fileName)
}

func TestRenameRemove(t *testing.T) {
	fileName0 := "file0"
	fileName1 := "file1"
	write(t, fileName0, fileData)
	rename(t, fileName0, fileName1)
	readAndCheck(t, fileName1, fileData)
	remove(t, fileName1)
	checkGone(t, fileName0)
}

func TestWriteAndBackup(t *testing.T) {
	fileName := "testfile"
	backupName := "testfile.bak"
	if FileExists(fileName) {
		remove(t, fileName)
	}
	if FileExists(backupName) {
		remove(t, backupName)
	}
	write(t, fileName, fileData)
	readAndCheck(t, fileName, fileData)
	WriteAndBackup(getName(fileName), fileData2)
	readAndCheck(t, backupName, fileData)
	remove(t, fileName)
	remove(t, backupName)
	checkGone(t, fileName)
}

// Helpers

func getName(name string) string {
	return path.Join(tempFolder, name)
}

func write(t *testing.T, fileName string, data []byte) {
	err := WriteFile(getName(fileName), data, FILE_RW)
	assert.NoError(t, err)
}

func readAndCheck(t *testing.T, fileName string, btsIn []byte) {
	bts, err := ReadFile(getName(fileName))
	assert.NoError(t, err)
	assert.True(t, bytes.Equal(bts, btsIn), "Failed to read file data. Written: %s, Read: %s\n", string(fileData), string(bts))
}

func remove(t *testing.T, fileName string) {
	err := os.Remove(getName(fileName))
	assert.NoError(t, err)
	checkGone(t, fileName)
}

func rename(t *testing.T, fileName0, fileName1 string) {
	assert.NoError(t, Rename(getName(fileName0), getName(fileName1)))
}

func checkGone(t *testing.T, fileName string) {
	name := getName(fileName)
	_, err := os.Stat(name)
	assert.True(t, os.IsNotExist(err), "File not removed: "+name)
}