123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192 |
- package directory
- import (
- "io/ioutil"
- "os"
- "path/filepath"
- "reflect"
- "sort"
- "testing"
- )
- // Size of an empty directory should be 0
- func TestSizeEmpty(t *testing.T) {
- var dir string
- var err error
- if dir, err = ioutil.TempDir(os.TempDir(), "testSizeEmptyDirectory"); err != nil {
- t.Fatalf("failed to create directory: %s", err)
- }
- var size int64
- if size, _ = Size(dir); size != 0 {
- t.Fatalf("empty directory has size: %d", size)
- }
- }
- // Size of a directory with one empty file should be 0
- func TestSizeEmptyFile(t *testing.T) {
- var dir string
- var err error
- if dir, err = ioutil.TempDir(os.TempDir(), "testSizeEmptyFile"); err != nil {
- t.Fatalf("failed to create directory: %s", err)
- }
- var file *os.File
- if file, err = ioutil.TempFile(dir, "file"); err != nil {
- t.Fatalf("failed to create file: %s", err)
- }
- var size int64
- if size, _ = Size(file.Name()); size != 0 {
- t.Fatalf("directory with one file has size: %d", size)
- }
- }
- // Size of a directory with one 5-byte file should be 5
- func TestSizeNonemptyFile(t *testing.T) {
- var dir string
- var err error
- if dir, err = ioutil.TempDir(os.TempDir(), "testSizeNonemptyFile"); err != nil {
- t.Fatalf("failed to create directory: %s", err)
- }
- var file *os.File
- if file, err = ioutil.TempFile(dir, "file"); err != nil {
- t.Fatalf("failed to create file: %s", err)
- }
- d := []byte{97, 98, 99, 100, 101}
- file.Write(d)
- var size int64
- if size, _ = Size(file.Name()); size != 5 {
- t.Fatalf("directory with one 5-byte file has size: %d", size)
- }
- }
- // Size of a directory with one empty directory should be 0
- func TestSizeNestedDirectoryEmpty(t *testing.T) {
- var dir string
- var err error
- if dir, err = ioutil.TempDir(os.TempDir(), "testSizeNestedDirectoryEmpty"); err != nil {
- t.Fatalf("failed to create directory: %s", err)
- }
- if dir, err = ioutil.TempDir(dir, "nested"); err != nil {
- t.Fatalf("failed to create nested directory: %s", err)
- }
- var size int64
- if size, _ = Size(dir); size != 0 {
- t.Fatalf("directory with one empty directory has size: %d", size)
- }
- }
- // Test directory with 1 file and 1 empty directory
- func TestSizeFileAndNestedDirectoryEmpty(t *testing.T) {
- var dir string
- var err error
- if dir, err = ioutil.TempDir(os.TempDir(), "testSizeFileAndNestedDirectoryEmpty"); err != nil {
- t.Fatalf("failed to create directory: %s", err)
- }
- if dir, err = ioutil.TempDir(dir, "nested"); err != nil {
- t.Fatalf("failed to create nested directory: %s", err)
- }
- var file *os.File
- if file, err = ioutil.TempFile(dir, "file"); err != nil {
- t.Fatalf("failed to create file: %s", err)
- }
- d := []byte{100, 111, 99, 107, 101, 114}
- file.Write(d)
- var size int64
- if size, _ = Size(dir); size != 6 {
- t.Fatalf("directory with 6-byte file and empty directory has size: %d", size)
- }
- }
- // Test directory with 1 file and 1 non-empty directory
- func TestSizeFileAndNestedDirectoryNonempty(t *testing.T) {
- var dir, dirNested string
- var err error
- if dir, err = ioutil.TempDir(os.TempDir(), "TestSizeFileAndNestedDirectoryNonempty"); err != nil {
- t.Fatalf("failed to create directory: %s", err)
- }
- if dirNested, err = ioutil.TempDir(dir, "nested"); err != nil {
- t.Fatalf("failed to create nested directory: %s", err)
- }
- var file *os.File
- if file, err = ioutil.TempFile(dir, "file"); err != nil {
- t.Fatalf("failed to create file: %s", err)
- }
- data := []byte{100, 111, 99, 107, 101, 114}
- file.Write(data)
- var nestedFile *os.File
- if nestedFile, err = ioutil.TempFile(dirNested, "file"); err != nil {
- t.Fatalf("failed to create file in nested directory: %s", err)
- }
- nestedData := []byte{100, 111, 99, 107, 101, 114}
- nestedFile.Write(nestedData)
- var size int64
- if size, _ = Size(dir); size != 12 {
- t.Fatalf("directory with 6-byte file and nested directory with 6-byte file has size: %d", size)
- }
- }
- // Test migration of directory to a subdir underneath itself
- func TestMoveToSubdir(t *testing.T) {
- var outerDir, subDir string
- var err error
- if outerDir, err = ioutil.TempDir(os.TempDir(), "TestMoveToSubdir"); err != nil {
- t.Fatalf("failed to create directory: %v", err)
- }
- if subDir, err = ioutil.TempDir(outerDir, "testSub"); err != nil {
- t.Fatalf("failed to create subdirectory: %v", err)
- }
- // write 4 temp files in the outer dir to get moved
- filesList := []string{"a", "b", "c", "d"}
- for _, fName := range filesList {
- if file, err := os.Create(filepath.Join(outerDir, fName)); err != nil {
- t.Fatalf("couldn't create temp file %q: %v", fName, err)
- } else {
- file.WriteString(fName)
- file.Close()
- }
- }
- if err = MoveToSubdir(outerDir, filepath.Base(subDir)); err != nil {
- t.Fatalf("Error during migration of content to subdirectory: %v", err)
- }
- // validate that the files were moved to the subdirectory
- infos, err := ioutil.ReadDir(subDir)
- if err != nil {
- t.Fatal(err)
- }
- if len(infos) != 4 {
- t.Fatalf("Should be four files in the subdir after the migration: actual length: %d", len(infos))
- }
- var results []string
- for _, info := range infos {
- results = append(results, info.Name())
- }
- sort.Sort(sort.StringSlice(results))
- if !reflect.DeepEqual(filesList, results) {
- t.Fatalf("Results after migration do not equal list of files: expected: %v, got: %v", filesList, results)
- }
- }
- // Test a non-existing directory
- func TestSizeNonExistingDirectory(t *testing.T) {
- if _, err := Size("/thisdirectoryshouldnotexist/TestSizeNonExistingDirectory"); err == nil {
- t.Fatalf("error is expected")
- }
- }
|