package instruction

import (
	"fmt"
	"os"
	"testing"
)

func TestCopyInvalidSyntax(t *testing.T) {
	tests := []string{
		"one-arg",
		"src dst extra-arg",
	}

	for i, test := range tests {
		inst := new(copy)
		err := inst.Build(test)
		if err != ErrInvalidSyntax {
			t.Errorf("[%d] expected error <%s>, got <%s>", i, ErrInvalidSyntax, err)
		}
	}
}
func TestCopyBuildArgs(t *testing.T) {
	tests := []string{
		"source destination",
		"\tsource destination",
		" source destination",
		"source\t destination",
		"source  destination",
		"source \tdestination",
		"source  destination",
		"source  destination\t",
		"source\t\tdestination\t",
		"source \t\t destination\t",
		"source\t  \tdestination\t",
	}

	for i, test := range tests {
		inst := new(copy)
		err := inst.Build(test)
		if err != nil {
			t.Errorf("[%d] unexpected error <%s>", i, err)
			continue
		}

		if inst.Src != "source" {
			t.Errorf("[%d] expected 'source', got '%s'", i, inst.Src)
			continue
		}
		if inst.Dst != "destination" {
			t.Errorf("[%d] expected 'source', got '%s'", i, inst.Dst)
			continue
		}

	}
}

func TestCopySourceNotExist(t *testing.T) {
	defer os.RemoveAll("/tmp/destination")
	raw := "/tmp/source /tmp/destination"

	ctx, err := CreateContext("apt-get", "")
	if err != nil {
		t.Fatalf("cannot create context")
	}

	for i := 0; i < 2; i++ {
		inst := new(copy)
		err := inst.Build(raw)
		if err != nil {
			t.Fatalf("[%d] unexpected error <%s>", i, err)
		}

		if i == 0 {
			_, err = inst.Exec(*ctx)
		} else {
			_, err = inst.DryRun(*ctx)
		}
		if err == nil {
			t.Fatalf("[%d] expected error", i)
		}

		ce, ok := err.(*FileError)
		if !ok {
			t.Fatalf("[%d] expected error of type <*FileError>", i)
		}
		if ce.Reason != "cannot find file" || ce.File != "/tmp/source" {
			t.Fatalf("[%d] expected error <%s '%s'> got <%s '%s'>", i, "cannot find file", "/tmp/source", ce.Reason, ce.File)
		}
	}
}

func TestCopySourceIsDir(t *testing.T) {
	src, dst := "/tmp/sourcedir", "/tmp/destinationdir"
	raw := fmt.Sprintf("%s %s", src, dst)
	defer os.RemoveAll(src)
	defer os.RemoveAll(dst)

	ctx, err := CreateContext("apt-get", "")
	if err != nil {
		t.Fatalf("cannot create context")
	}

	for i := 0; i < 2; i++ {
		// 1. create directory
		os.RemoveAll(src)
		if err := os.MkdirAll(src, os.FileMode(0777)); err != nil {
			t.Fatalf("[%d] cannot create test directory | %s", i, err)
		}

		inst := new(copy)
		err := inst.Build(raw)
		if err != nil {
			t.Fatalf("[%d] unexpected error <%s>", i, err)
		}

		if i == 0 {
			_, err = inst.Exec(*ctx)
		} else {
			_, err = inst.DryRun(*ctx)
		}
		if err != nil {
			t.Fatalf("[%d] unexpected error <%s>", i, err)
		}
	}

}
func TestCopyInvalidDestination(t *testing.T) {
	src, dst := "/tmp/source", "/tmp/missing-directory/invalid-destination"
	raw := fmt.Sprintf("%s %s", src, dst)
	defer os.RemoveAll(src)
	defer os.RemoveAll(dst)

	ctx, err := CreateContext("apt-get", "")
	if err != nil {
		t.Fatalf("cannot create context")
	}

	for i := 0; i < 2; i++ {

		os.RemoveAll(src)
		// 1. create directory
		if err := os.MkdirAll(src, os.FileMode(0777)); err != nil {
			t.Fatalf("[%d] cannot create test directory | %s", i, err)
		}
		inst := new(copy)
		err := inst.Build(raw)
		if err != nil {
			t.Fatalf("[%d] unexpected error <%s>", i, err)
		}

		if i == 0 {
			_, err = inst.Exec(*ctx)
		} else {
			_, err = inst.DryRun(*ctx)
		}
		if err == nil {
			t.Fatalf("[%d] expected error", i)
		}

		ce, ok := err.(*FileError)
		if !ok {
			t.Fatalf("[%d] expected error of type <*FileError>", i)
		}
		if ce.Reason != "cannot copy to" || ce.File != dst {
			t.Fatalf("[%d] expected error <%s '%s'> got <%s '%s'>", i, "cannot copy to", dst, ce.Reason, ce.File)
		}

	}
}
func TestCopySourceIsFile(t *testing.T) {
	src, dst := "/tmp/source", "/tmp/destination-file"
	raw := fmt.Sprintf("%s %s", src, dst)
	defer os.RemoveAll(src)
	defer os.RemoveAll(dst)

	ctx, err := CreateContext("apt-get", "")
	if err != nil {
		t.Fatalf("cannot create context")
	}

	for i := 0; i < 2; i++ {

		os.RemoveAll(src)

		// 1. create directory
		fd, err := os.Create(src)
		if err != nil {
			t.Fatalf("[%d] cannot create test file | %s", i, err)
		}
		fd.Close()

		inst := new(copy)
		err = inst.Build(raw)
		if err != nil {
			t.Fatalf("[%d] unexpected error <%s>", i, err)
		}

		if i == 0 {
			_, err = inst.Exec(*ctx)
		} else {
			_, err = inst.DryRun(*ctx)
		}
		if err != nil {
			t.Fatalf("[%d] unexpected error <%s>", i, err)
		}

	}
}

func TestCopySourceIsFileDestinationExists(t *testing.T) {
	src, dst := "/tmp/source", "/tmp/destination-file"
	raw := fmt.Sprintf("%s %s", src, dst)
	defer os.RemoveAll(src)
	defer os.RemoveAll(dst)

	ctx, err := CreateContext("apt-get", "")
	if err != nil {
		t.Fatalf("cannot create context")
	}

	for i := 0; i < 2; i++ {

		os.RemoveAll(src)
		os.RemoveAll(dst)

		// 1. create source
		fd, err := os.Create(src)
		if err != nil {
			t.Fatalf("[%d] cannot create test file | %s", i, err)
		}
		fd.Close()
		// 1. create destination
		fd, err = os.Create(src)
		if err != nil {
			t.Fatalf("[%d] cannot create test file | %s", i, err)
		}
		fd.Close()

		inst := new(copy)
		err = inst.Build(raw)
		if err != nil {
			t.Fatalf("[%d] unexpected error <%s>", i, err)
		}

		if i == 0 {
			_, err = inst.Exec(*ctx)
		} else {
			_, err = inst.DryRun(*ctx)
		}
		if err != nil {
			t.Fatalf("[%d] unexpected error <%s>", i, err)
		}

	}
}
func TestCopyCannotWriteDestination(t *testing.T) {
	src, dst := "/tmp/source", "/tmp/destination-perm"
	raw := fmt.Sprintf("%s %s", src, dst)
	defer os.RemoveAll(src)
	defer os.RemoveAll(dst)

	ctx, err := CreateContext("apt-get", "")
	if err != nil {
		t.Fatalf("cannot create context")
	}

	for i := 0; i < 2; i++ {

		os.RemoveAll(src)
		if err := os.RemoveAll(dst); err != nil {
			t.Fatalf("[%d] cannot remove destination file\n", i)
		}

		// 1. create source
		fd, err := os.Create(src)
		if err != nil {
			t.Fatalf("[%d] cannot create test file | %s", i, err)
		}
		fd.Close()
		// 1. create destination
		fd, err = os.Create(src)
		if err != nil {
			t.Fatalf("[%d] cannot create test file | %s", i, err)
		}
		err = fd.Chmod(os.FileMode(0555))
		if err != nil {
			t.Fatalf("[%d] cannot set permissions | %s", i, err)
		}
		fd.Close()

		inst := new(copy)
		err = inst.Build(raw)
		if err != nil {
			t.Fatalf("[%d] unexpected error <%s>", i, err)
		}

		if i == 0 {
			_, err = inst.Exec(*ctx)
		} else {
			_, err = inst.DryRun(*ctx)
		}
		if err != nil {
			t.Fatalf("[%d] unexpected error <%s>", i, err)
		}

	}
}