You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
458 lines
12 KiB
458 lines
12 KiB
// 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"
|
|
"os/exec"
|
|
"path"
|
|
"testing"
|
|
|
|
"android.googlesource.com/platform/tools/treble.git/hacksaw/bind"
|
|
"android.googlesource.com/platform/tools/treble.git/hacksaw/codebase"
|
|
"android.googlesource.com/platform/tools/treble.git/hacksaw/config"
|
|
)
|
|
|
|
func TestBasicCreate(t *testing.T) {
|
|
defer config.GetConfig().Reset()
|
|
codebaseDir, err := ioutil.TempDir("", "codebase")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer os.RemoveAll(codebaseDir)
|
|
gitDir := path.Join(codebaseDir, "project", ".git")
|
|
if err = os.MkdirAll(gitDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
repoDir := path.Join(codebaseDir, ".repo")
|
|
if err = os.Mkdir(repoDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
listContents := []byte("project")
|
|
listPath := path.Join(repoDir, "project.list")
|
|
if err = ioutil.WriteFile(listPath, listContents, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = codebase.Add("test-codebase", codebaseDir)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
// The top dir must be named "hacksaw"
|
|
// otherwise the mounters will reject any
|
|
// mount requests
|
|
wsTempDir, err := ioutil.TempDir("", "workspace")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer os.RemoveAll(wsTempDir)
|
|
wsTopDir := path.Join(wsTempDir, "hacksaw")
|
|
if err = os.Mkdir(wsTopDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
ws := New(bind.NewFakePathBinder(), wsTopDir)
|
|
if _, err = ws.Create("test-workspace", "test-codebase"); err != nil {
|
|
t.Error(err)
|
|
}
|
|
workspaceDir, err := ws.GetDir("test-workspace")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = os.Stat(workspaceDir)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestWorkspaceDuplicate(t *testing.T) {
|
|
defer config.GetConfig().Reset()
|
|
codebaseDir, err := ioutil.TempDir("", "codebase")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer os.RemoveAll(codebaseDir)
|
|
gitDir := path.Join(codebaseDir, "project", ".git")
|
|
if err = os.MkdirAll(gitDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
repoDir := path.Join(codebaseDir, ".repo")
|
|
if err = os.Mkdir(repoDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
listContents := []byte("project")
|
|
listPath := path.Join(repoDir, "project.list")
|
|
if err = ioutil.WriteFile(listPath, listContents, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = codebase.Add("test-codebase", codebaseDir)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
// The top dir must be named "hacksaw"
|
|
// otherwise the mounters will reject any
|
|
// mount requests
|
|
wsTempDir, err := ioutil.TempDir("", "workspace")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer os.RemoveAll(wsTempDir)
|
|
wsTopDir := path.Join(wsTempDir, "hacksaw")
|
|
if err = os.Mkdir(wsTopDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
ws := New(bind.NewFakePathBinder(), wsTopDir)
|
|
_, err = ws.Create("test-workspace", "test-codebase")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = ws.Create("test-workspace", "test-codebase")
|
|
if err == nil {
|
|
t.Error("Allowed workspace duplicate")
|
|
}
|
|
}
|
|
|
|
func TestCreateWorkspaceFromBadCodebase(t *testing.T) {
|
|
defer config.GetConfig().Reset()
|
|
codebaseDir, err := ioutil.TempDir("", "test")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer os.RemoveAll(codebaseDir)
|
|
_, err = codebase.Add("test-codebase", codebaseDir)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
wsTempDir, err := ioutil.TempDir("", "workspace")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer os.RemoveAll(wsTempDir)
|
|
wsTopDir := path.Join(wsTempDir, "hacksaw")
|
|
if err = os.Mkdir(wsTopDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
ws := New(bind.NewFakePathBinder(), wsTopDir)
|
|
if _, err = ws.Create("test-workspace", "does-not-exist"); err == nil {
|
|
t.Error("Allowed bad codebase")
|
|
}
|
|
}
|
|
|
|
func TestList(t *testing.T) {
|
|
defer config.GetConfig().Reset()
|
|
codebaseDir, err := ioutil.TempDir("", "test")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer os.RemoveAll(codebaseDir)
|
|
gitDir := path.Join(codebaseDir, "project", ".git")
|
|
if err = os.MkdirAll(gitDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
repoDir := path.Join(codebaseDir, ".repo")
|
|
if err = os.Mkdir(repoDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
listContents := []byte("project")
|
|
listPath := path.Join(repoDir, "project.list")
|
|
if err = ioutil.WriteFile(listPath, listContents, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = codebase.Add("test-codebase", codebaseDir)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
wsTempDir, err := ioutil.TempDir("", "workspace")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer os.RemoveAll(wsTempDir)
|
|
wsTopDir := path.Join(wsTempDir, "hacksaw")
|
|
if err = os.Mkdir(wsTopDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
ws := New(bind.NewFakePathBinder(), wsTopDir)
|
|
if _, err = ws.Create("test-workspace", "test-codebase"); err != nil {
|
|
t.Error(err)
|
|
}
|
|
list := ws.List()
|
|
cb, ok := list["test-workspace"]
|
|
if !ok || cb != "test-codebase" {
|
|
t.Error("Added workspace not listed")
|
|
}
|
|
}
|
|
|
|
func TestRemove(t *testing.T) {
|
|
defer config.GetConfig().Reset()
|
|
codebaseDir, err := ioutil.TempDir("", "test")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer os.RemoveAll(codebaseDir)
|
|
gitDir := path.Join(codebaseDir, "project", ".git")
|
|
if err = os.MkdirAll(gitDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
repoDir := path.Join(codebaseDir, ".repo")
|
|
if err = os.Mkdir(repoDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
listContents := []byte("project")
|
|
listPath := path.Join(repoDir, "project.list")
|
|
if err = ioutil.WriteFile(listPath, listContents, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = codebase.Add("test-codebase", codebaseDir)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
wsTempDir, err := ioutil.TempDir("", "workspace")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer os.RemoveAll(wsTempDir)
|
|
wsTopDir := path.Join(wsTempDir, "hacksaw")
|
|
if err = os.Mkdir(wsTopDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
ws := New(bind.NewFakePathBinder(), wsTopDir)
|
|
if _, err = ws.Create("test-workspace", "test-codebase"); err != nil {
|
|
t.Error(err)
|
|
}
|
|
workspaceDir, err := ws.GetDir("test-workspace")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = os.Stat(workspaceDir)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
cfg, err := ws.Remove("test-workspace")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, ok := cfg.Workspaces["test-codebase"]
|
|
if ok {
|
|
t.Error("Removed workspace test-codebase is still in the configuration")
|
|
}
|
|
_, err = os.Stat(workspaceDir)
|
|
if err == nil {
|
|
t.Error("Workspace test-workspace was removed but its directory remains")
|
|
} else if os.IsNotExist(err) {
|
|
// This is the expected error
|
|
} else {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
func TestEdit(t *testing.T) {
|
|
defer config.GetConfig().Reset()
|
|
codebaseDir, err := ioutil.TempDir("", "codebase")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer os.RemoveAll(codebaseDir)
|
|
projectDir := path.Join(codebaseDir, "project")
|
|
if err = os.MkdirAll(projectDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
cmd := exec.Command("git", "-C", projectDir, "init")
|
|
output, err := cmd.CombinedOutput()
|
|
if err != nil {
|
|
t.Errorf("Command\n%s\nfailed with the following:\n%s\n%s",
|
|
cmd.String(), err.Error(), output)
|
|
}
|
|
cmd = exec.Command("git", "-C", projectDir, "commit", `--message="Initial commit"`, "--allow-empty")
|
|
output, err = cmd.CombinedOutput()
|
|
if err != nil {
|
|
t.Errorf("Command\n%s\nfailed with the following:\n%s\n%s",
|
|
cmd.String(), err.Error(), output)
|
|
}
|
|
repoDir := path.Join(codebaseDir, ".repo")
|
|
if err = os.Mkdir(repoDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
listContents := []byte("project")
|
|
listPath := path.Join(repoDir, "project.list")
|
|
if err = ioutil.WriteFile(listPath, listContents, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = codebase.Add("test-codebase", codebaseDir)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
wsTempDir, err := ioutil.TempDir("", "workspace")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer os.RemoveAll(wsTempDir)
|
|
wsTopDir := path.Join(wsTempDir, "hacksaw")
|
|
if err = os.Mkdir(wsTopDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
ws := New(bind.NewFakePathBinder(), wsTopDir)
|
|
if _, err = ws.Create("test-workspace", "test-codebase"); err != nil {
|
|
t.Error(err)
|
|
}
|
|
workspaceDir, err := ws.GetDir("test-workspace")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = os.Stat(workspaceDir)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
editPath := path.Join(workspaceDir, "project")
|
|
branchName, wsProjectDir, err := ws.Edit(editPath)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
if branchName == "" {
|
|
t.Error("Editing returned an empty branch")
|
|
}
|
|
if wsProjectDir == "" {
|
|
t.Error("Editing returned an empty project path")
|
|
}
|
|
cmd = exec.Command("git", "-C", wsProjectDir, "show", branchName)
|
|
output, err = cmd.CombinedOutput()
|
|
if err != nil {
|
|
t.Errorf("Command\n%s\nfailed with the following:\n%s\n%s",
|
|
cmd.String(), err.Error(), output)
|
|
}
|
|
//Recreate workspace and try editing again
|
|
_, err = ws.Remove("test-workspace")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = ws.Create("test-workspace", "test-codebase")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, _, err = ws.Edit(editPath)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
}
|
|
|
|
const projectList = `read-only-project
|
|
editable-project`
|
|
|
|
func TestRecreate(t *testing.T) {
|
|
defer config.GetConfig().Reset()
|
|
codebaseDir, err := ioutil.TempDir("", "codebase")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer os.RemoveAll(codebaseDir)
|
|
roProjectDir := path.Join(codebaseDir, "read-only-project")
|
|
if err = os.MkdirAll(roProjectDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
cmd := exec.Command("git", "-C", roProjectDir, "init")
|
|
output, err := cmd.CombinedOutput()
|
|
if err != nil {
|
|
t.Errorf("Command\n%s\nfailed with the following:\n%s\n%s",
|
|
cmd.String(), err.Error(), output)
|
|
}
|
|
cmd = exec.Command("git", "-C", roProjectDir, "commit", `--message="Initial commit"`, "--allow-empty")
|
|
output, err = cmd.CombinedOutput()
|
|
if err != nil {
|
|
t.Errorf("Command\n%s\nfailed with the following:\n%s\n%s",
|
|
cmd.String(), err.Error(), output)
|
|
}
|
|
linkPath := path.Join(codebaseDir, "symlink")
|
|
if err = os.Symlink(roProjectDir, linkPath); err != nil {
|
|
t.Error(err)
|
|
}
|
|
rwProjectDir := path.Join(codebaseDir, "editable-project")
|
|
if err = os.MkdirAll(rwProjectDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
cmd = exec.Command("git", "-C", rwProjectDir, "init")
|
|
output, err = cmd.CombinedOutput()
|
|
if err != nil {
|
|
t.Errorf("Command\n%s\nfailed with the following:\n%s\n%s",
|
|
cmd.String(), err.Error(), output)
|
|
}
|
|
cmd = exec.Command("git", "-C", rwProjectDir, "commit", `--message="Initial commit"`, "--allow-empty")
|
|
output, err = cmd.CombinedOutput()
|
|
if err != nil {
|
|
t.Errorf("Command\n%s\nfailed with the following:\n%s\n%s",
|
|
cmd.String(), err.Error(), output)
|
|
}
|
|
repoDir := path.Join(codebaseDir, ".repo")
|
|
if err = os.Mkdir(repoDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
listContents := []byte(projectList)
|
|
listPath := path.Join(repoDir, "project.list")
|
|
if err = ioutil.WriteFile(listPath, listContents, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = codebase.Add("test-codebase", codebaseDir)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
wsTempDir, err := ioutil.TempDir("", "workspace")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
defer os.RemoveAll(wsTempDir)
|
|
wsTopDir := path.Join(wsTempDir, "hacksaw")
|
|
if err = os.Mkdir(wsTopDir, os.ModePerm); err != nil {
|
|
t.Error(err)
|
|
}
|
|
pathBinder := bind.NewFakePathBinder()
|
|
ws := New(pathBinder, wsTopDir)
|
|
if _, err = ws.Create("test-workspace", "test-codebase"); err != nil {
|
|
t.Error(err)
|
|
}
|
|
workspaceDir, err := ws.GetDir("test-workspace")
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
editPath := path.Join(workspaceDir, "editable-project")
|
|
_, _, err = ws.Edit(editPath)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
emptyFilePath := path.Join(editPath, "empty-edit")
|
|
emptyFile, err := os.Create(emptyFilePath)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
emptyFile.Close()
|
|
if _, err = ws.Recreate("test-workspace"); err != nil {
|
|
t.Error(err)
|
|
}
|
|
_, err = os.Stat(emptyFilePath)
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
wsRoProjectDir := path.Join(workspaceDir, "read-only-project")
|
|
isRoPathBound := false
|
|
pathList, err := pathBinder.List()
|
|
if err != nil {
|
|
t.Error(err)
|
|
}
|
|
for _, path := range pathList {
|
|
if path == wsRoProjectDir {
|
|
isRoPathBound = true
|
|
}
|
|
}
|
|
if !isRoPathBound {
|
|
t.Error("Read only project was not mounted to the workspace")
|
|
}
|
|
}
|