mvpls_test.go

package main

import (
	"fmt"
	"io/ioutil"
	"os"
	"path/filepath"
	"regexp"
	"testing"
)

func TestMoveFile(t *testing.T) {
	testDir := createTestDir()
	defer os.RemoveAll(testDir)

	oldFile := filepath.Join(testDir, "file11.png")
	newFile := filepath.Join(testDir, "file11.new")
	MoveFile(oldFile, newFile)
	if _, err := os.Stat(newFile); err != nil && os.IsNotExist(err) {
		t.Error("File Not Moved")
	}

	if _, err := os.Stat(oldFile); err == nil && os.IsExist(err) {
		t.Error("Old file location still present")
	}

	oldFile = filepath.Join(testDir, "level2/file22.png")
	newFile = filepath.Join(testDir, "file22.png")
	MoveFile(oldFile, newFile)
	if _, err := os.Stat(newFile); err != nil && os.IsNotExist(err) {
		t.Error("file not moved to directory")
	}

	if _, err := os.Stat(oldFile); err == nil && os.IsExist(err) {
		t.Error("Old file location still present")
	}

	os.Remove("testingDir/beforeDirMove")

	oldFile = filepath.Join(testDir, "file11.jpg")
	newFile = filepath.Join(testDir, "level2/file11.png")

	MoveFile(oldFile, newFile)
	if _, err := os.Stat(newFile); err != nil && os.IsNotExist(err) {
		t.Error("file not moved to directory")
	}

	oldFile = filepath.Join(testDir, "level2")
	newFile = filepath.Join(testDir, "level2.moved")

	MoveFile(oldFile, newFile)
	if _, err := os.Stat(newFile); err != nil && os.IsNotExist(err) {
		t.Error("Test directory not moved")
	}

	if _, err := os.Stat(oldFile); err == nil && os.IsExist(err) {
		t.Error("Old test directory still present")
	}
}

func TestRemoveFile(t *testing.T) {
	testDir := createTestDir()
	defer os.RemoveAll(testDir)

	oldFile := filepath.Join(testDir, "file11.png")
	RemoveFile(oldFile, "")
	if _, err := os.Stat(oldFile); err != nil && os.IsExist(err) {
		t.Error("File Not Removed")
	}

	oldFile = filepath.Join(testDir, "level2/file22.png")
	RemoveFile(oldFile, "")
	if _, err := os.Stat(oldFile); err != nil && os.IsExist(err) {
		t.Error("file within directory not removed")
	}
}

func TestCopyFile(t *testing.T) {
	testDir := createTestDir()
	defer os.RemoveAll(testDir)

	oldFile := filepath.Join(testDir, "file11.png")
	newFile := filepath.Join(testDir, "file11.new")
	CopyFile(oldFile, newFile)
	if _, err := os.Stat(newFile); err != nil && os.IsNotExist(err) {
		t.Error("File Not Copied")
	}

	if _, err := os.Stat(oldFile); err == nil && os.IsExist(err) {
		t.Error("File original location not present")
	}

	oldFile = filepath.Join(testDir, "level2/file22.png")
	newFile = filepath.Join(testDir, "file22.png")
	CopyFile(oldFile, newFile)
	if _, err := os.Stat(newFile); err != nil && os.IsNotExist(err) {
		t.Error("file not copied to directory")
	}

	if _, err := os.Stat(oldFile); err == nil && os.IsExist(err) {
		t.Error("Original file not present")
	}

	os.Remove("testingDir/beforeDirMove")

	oldFile = filepath.Join(testDir, "file11.jpg")
	newFile = filepath.Join(testDir, "level2/file11.png")

	CopyFile(oldFile, newFile)
	if _, err := os.Stat(newFile); err != nil && os.IsNotExist(err) {
		t.Error("file not copied to directory")
	}

	if _, err := os.Stat(oldFile); err == nil && os.IsExist(err) {
		t.Error("Original file not present")
	}
}

func TestProbeDirectory(t *testing.T) {
	testDir := createTestDir()
	target := filepath.Join(testDir, "moved")
	movedDir := filepath.Join(testDir, "moved")
	defer os.RemoveAll(testDir)

	regex, _ := regexp.Compile(".*\\.png")
	ProbeDirectory(testDir, target, regex, MoveFile)

	path := testDir
	var full string
	for i := 1; i < 4; i++ {
		for j := 1; j < 3; j++ {
			full = filepath.Join(movedDir, fmt.Sprintf("file%d%d.png", i, j))

			if _, err := os.Stat(full); err != nil && os.IsNotExist(err) {
				t.Error(fmt.Sprintf("Test file %s is not present", full))
			}

			full = filepath.Join(path, fmt.Sprintf("file%d%d.jpg", i, j))
			if _, err := os.Stat(full); err != nil && os.IsNotExist(err) {
				t.Error(fmt.Sprintf("Test file %s should not have moved", full))
			}
		}
		path = filepath.Join(path, fmt.Sprintf("level%d", i+1))
	}
}

func createTestDir() (testDir string) {
	pdir := os.TempDir()
	testDir, _ = ioutil.TempDir(pdir, "*-mvplsTest")

	movedDir := filepath.Join(testDir, "moved")
	os.Mkdir(movedDir, 0777)
	testSubs := filepath.Join(testDir, "level2/level3/level4")
	os.MkdirAll(testSubs, 0777)

	subs := [...]string{"file11.jpg", "file12.jpg", "file11.png", "file12.png", "level2/file21.jpg",
		"level2/file22.jpg", "level2/file21.png", "level2/file22.png", "level2/level3/file31.jpg",
		"level2/level3/file32.jpg", "level2/level3/file31.png", "level2/level3/file32.png"}
	var sub string

	for i := 0; i < len(subs); i++ {
		sub = filepath.Join(testDir, subs[i])
		os.Create(sub)
	}
	return
}