v811_spc009/tools/treble/hacksaw/workspace/copier_test.go

// Copyright 2020 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     https://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package workspace

import (
	"io/ioutil"
	"os"
	"path"
	"testing"
)

func TestCopy(t *testing.T) {
	// Setup codebase
	codebaseDir, err := ioutil.TempDir("", "codebase")
	if err != nil {
		t.Error(err)
	}
	defer os.RemoveAll(codebaseDir)
	project1GitDir := path.Join(codebaseDir, "project1", ".git")
	if err = os.MkdirAll(project1GitDir, os.ModePerm); err != nil {
		t.Error(err)
	}
	emptyBytes := []byte{}
	project1File := path.Join(codebaseDir, "project1", "projectfile")
	if err = ioutil.WriteFile(project1File, emptyBytes, os.ModePerm); err != nil {
		t.Error(err)
	}
	symlink := path.Join(codebaseDir, "symlink")
	if err = os.Symlink(path.Join(codebaseDir, "project1"), symlink); err != nil {
		t.Error(err)
	}
	project2GitDir := path.Join(codebaseDir, "dirwithprojects", "project2", ".git")
	if err = os.MkdirAll(project2GitDir, os.ModePerm); err != nil {
		t.Error(err)
	}
	dirWithoutProjects := path.Join(codebaseDir, "dirwithoutprojects")
	if err = os.Mkdir(dirWithoutProjects, os.ModePerm); err != nil {
		t.Error(err)
	}
	projectSiblingFile := path.Join(codebaseDir, "dirwithprojects", "projectsiblingfile")
	if err = ioutil.WriteFile(projectSiblingFile, emptyBytes, os.ModePerm); err != nil {
		t.Error(err)
	}
	noProjectFile1 := path.Join(dirWithoutProjects, "noprojectfile1")
	if err = ioutil.WriteFile(noProjectFile1, emptyBytes, os.ModePerm); err != nil {
		t.Error(err)
	}
	noProjectFile2 := path.Join(dirWithoutProjects, "noprojectfile2")
	if err = ioutil.WriteFile(noProjectFile2, emptyBytes, os.ModePerm); err != nil {
		t.Error(err)
	}
	topFile := path.Join(codebaseDir, "topfile")
	if err = ioutil.WriteFile(topFile, emptyBytes, os.ModePerm); err != nil {
		t.Error(err)
	}
	gitProjects := []string{
		"project1",
		"dirwithprojects/project2",
	}

	// Set up workspace
	workspaceDir, err := ioutil.TempDir("", "workspace")
	if err != nil {
		t.Error(err)
	}
	//This dir may already exist if the projects have been mounted
	wsDirWithProjects := path.Join(workspaceDir, "dirwithprojects")
	if err = os.Mkdir(wsDirWithProjects, os.ModePerm); err != nil {
		t.Error(err)
	}

	copier := NewFileCopier()
	if err = copier.Copy(codebaseDir, gitProjects, workspaceDir); err != nil {
		t.Error(err)
	}

	wsTopFile := path.Join(workspaceDir, "topfile")
	_, err = os.Stat(wsTopFile)
	if err != nil {
		t.Error(err)
	}
	wsNoProjectFile1 := path.Join(workspaceDir, "dirwithoutprojects", "noprojectfile1")
	_, err = os.Stat(wsNoProjectFile1)
	if err != nil {
		t.Error(err)
	}
	wsNoProjectFile2 := path.Join(workspaceDir, "dirwithoutprojects", "noprojectfile2")
	_, err = os.Stat(wsNoProjectFile2)
	if err != nil {
		t.Error(err)
	}
	wsProjectSiblingFile := path.Join(workspaceDir, "dirwithprojects", "projectsiblingfile")
	_, err = os.Stat(wsProjectSiblingFile)
	if err != nil {
		t.Error(err)
	}
	wsSymlink := path.Join(workspaceDir, "symlink")
	_, err = os.Stat(wsSymlink)
	if err != nil {
		t.Error(err)
	}
	//TODO: check why this is failing
	//	if linkInfo.Mode() & os.ModeSymlink != os.ModeSymlink {
	//		t.Error("Symlink not copied as symlink")
	//	}
	wsProject2Dir := path.Join(workspaceDir, "dirwithprojects", "project2")
	_, err = os.Stat(wsProject2Dir)
	if err == nil {
		t.Error("Project2 mistakenly copied")
	} else if os.IsNotExist(err) {
		//This error is expected
	} else {
		t.Error(err)
	}
	wsProject1Dir := path.Join(workspaceDir, "project1")
	_, err = os.Stat(wsProject1Dir)
	if err == nil {
		t.Error("Project1 mistakenly copied")
	} else if os.IsNotExist(err) {
		//This error is expected
	} else {
		t.Error(err)
	}
}