// Copyright 2020 Google LLC
//
// Licensed under the Apache License, Ve
// 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"
 "reflect"
 "testing"

 "android.googlesource.com/platform/tools/treble.git/hacksaw/bind"
)

func TestBasicCompose(t *testing.T) {
 var err error
 var codebaseDir string
 if codebaseDir, err = ioutil.TempDir("", "codebase"); 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)
 }
 var wsTempDir string
 if wsTempDir, err = ioutil.TempDir("", "workspace"); 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)
 }
 fakePathBinder := bind.NewFakePathBinder()
 wsComposer := NewComposer(fakePathBinder)
 list, err := wsComposer.Compose(codebaseDir, wsTopDir)
 if err != nil {
  t.Error(err)
 }
 expectedList := []string{
  path.Join(wsTopDir, ".repo"),
  path.Join(wsTopDir, "project"),
 }
 if !reflect.DeepEqual(expectedList, list) {
  t.Errorf("Expected a list of binds %v and got %v", expectedList, list)
 }
 list, err = wsComposer.Dismantle(wsTopDir)
 if err != nil {
  t.Error(err)
 }
 expectedList = []string{
  path.Join(wsTopDir, "project"),
  path.Join(wsTopDir, ".repo"),
 }
 if !reflect.DeepEqual(expectedList, list) {
  t.Errorf("Expected a list of unbinds %v and got %v", expectedList, list)
 }
}

const projectListContents = `project2/subprojectb
project1
project2
project1/subprojecta`

func TestBindOrder(t *testing.T) {
 var err error
 var codebaseDir string
 if codebaseDir, err = ioutil.TempDir("", "codebase"); err != nil {
  t.Error(err)
 }
 defer os.RemoveAll(codebaseDir)
 project2Git := path.Join(codebaseDir, "project2", ".git")
 if err = os.MkdirAll(project2Git, os.ModePerm); err != nil {
  t.Error(err)
 }
 project1Git := path.Join(codebaseDir, "project1", ".git")
 if err = os.MkdirAll(project1Git, os.ModePerm); err != nil {
  t.Error(err)
 }
 subProjectAGit := path.Join(codebaseDir, "project1", "subprojecta", ".git")
 if err = os.MkdirAll(subProjectAGit, os.ModePerm); err != nil {
  t.Error(err)
 }
 subProjectBGit := path.Join(codebaseDir, "project2", "subprojectb", ".git")
 if err = os.MkdirAll(subProjectBGit, 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(projectListContents)
 listPath := path.Join(repoDir, "project.list")
 if err = ioutil.WriteFile(listPath, listContents, os.ModePerm); err != nil {
  t.Error(err)
 }
 var wsTempDir string
 if wsTempDir, err = ioutil.TempDir("", "workspace"); 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)
 }
 fakePathBinder := bind.NewFakePathBinder()
 wsComposer := NewComposer(fakePathBinder)
 bindList, err := wsComposer.Compose(codebaseDir, wsTopDir)
 if err != nil {
  t.Error(err)
 }
 expectedList := []string{
  path.Join(wsTopDir, ".repo"),
  path.Join(wsTopDir, "project1"),
  path.Join(wsTopDir, "project1", "subprojecta"),
  path.Join(wsTopDir, "project2"),
  path.Join(wsTopDir, "project2", "subprojectb"),
 }
 if !reflect.DeepEqual(expectedList, bindList) {
  t.Errorf("Expected a list of binds %v and got %v", expectedList, bindList)
 }
 unbindList, err := wsComposer.Dismantle(wsTopDir)
 if err != nil {
  t.Error(err)
 }
 expectedList = []string{
  path.Join(wsTopDir, "project2", "subprojectb"),
  path.Join(wsTopDir, "project2"),
  path.Join(wsTopDir, "project1", "subprojecta"),
  path.Join(wsTopDir, "project1"),
  path.Join(wsTopDir, ".repo"),
 }
 if !reflect.DeepEqual(expectedList, unbindList) {
  t.Errorf("Expected a list of unbinds %v and got %v", expectedList, unbindList)
 }
}