// Copyright 2015 Google Inc. All rights reserved.
//
// 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
//
//     http://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 blueprint

import (
 "reflect"
 "testing"
)

var (
 testModuleA = &moduleInfo{variant: variant{name: "testModuleA"}}
 testModuleB = &moduleInfo{variant: variant{name: "testModuleB"}}
 testModuleC = &moduleInfo{variant: variant{name: "testModuleC"}}
 testModuleD = &moduleInfo{variant: variant{name: "testModuleD"}}
 testModuleE = &moduleInfo{variant: variant{name: "testModuleE"}}
 testModuleF = &moduleInfo{variant: variant{name: "testModuleF"}}
)

var spliceModulesTestCases = []struct {
 in         moduleList
 at         int
 with       moduleList
 out        moduleList
 outAt      int
 reallocate bool
}{
 {
  // Insert at the beginning
  in:         moduleList{testModuleA, testModuleB, testModuleC},
  at:         0,
  with:       moduleList{testModuleD, testModuleE},
  out:        moduleList{testModuleD, testModuleE, testModuleB, testModuleC},
  outAt:      1,
  reallocate: true,
 },
 {
  // Insert in the middle
  in:         moduleList{testModuleA, testModuleB, testModuleC},
  at:         1,
  with:       moduleList{testModuleD, testModuleE},
  out:        moduleList{testModuleA, testModuleD, testModuleE, testModuleC},
  outAt:      2,
  reallocate: true,
 },
 {
  // Insert at the end
  in:         moduleList{testModuleA, testModuleB, testModuleC},
  at:         2,
  with:       moduleList{testModuleD, testModuleE},
  out:        moduleList{testModuleA, testModuleB, testModuleD, testModuleE},
  outAt:      3,
  reallocate: true,
 },
 {
  // Insert over a single element
  in:         moduleList{testModuleA},
  at:         0,
  with:       moduleList{testModuleD, testModuleE},
  out:        moduleList{testModuleD, testModuleE},
  outAt:      1,
  reallocate: true,
 },
 {
  // Insert at the beginning without reallocating
  in:         moduleList{testModuleA, testModuleB, testModuleC, nil}[0:3],
  at:         0,
  with:       moduleList{testModuleD, testModuleE},
  out:        moduleList{testModuleD, testModuleE, testModuleB, testModuleC},
  outAt:      1,
  reallocate: false,
 },
 {
  // Insert in the middle without reallocating
  in:         moduleList{testModuleA, testModuleB, testModuleC, nil}[0:3],
  at:         1,
  with:       moduleList{testModuleD, testModuleE},
  out:        moduleList{testModuleA, testModuleD, testModuleE, testModuleC},
  outAt:      2,
  reallocate: false,
 },
 {
  // Insert at the end without reallocating
  in:         moduleList{testModuleA, testModuleB, testModuleC, nil}[0:3],
  at:         2,
  with:       moduleList{testModuleD, testModuleE},
  out:        moduleList{testModuleA, testModuleB, testModuleD, testModuleE},
  outAt:      3,
  reallocate: false,
 },
 {
  // Insert over a single element without reallocating
  in:         moduleList{testModuleA, nil}[0:1],
  at:         0,
  with:       moduleList{testModuleD, testModuleE},
  out:        moduleList{testModuleD, testModuleE},
  outAt:      1,
  reallocate: false,
 },
}

func TestSpliceModules(t *testing.T) {
 for _, testCase := range spliceModulesTestCases {
  in := make(moduleList, len(testCase.in), cap(testCase.in))
  copy(in, testCase.in)
  origIn := in
  got, gotAt := spliceModules(in, testCase.at, testCase.with)
  if !reflect.DeepEqual(got, testCase.out) {
   t.Errorf("test case: %v, %v -> %v", testCase.in, testCase.at, testCase.with)
   t.Errorf("incorrect output:")
   t.Errorf("  expected: %v", testCase.out)
   t.Errorf("       got: %v", got)
  }
  if gotAt != testCase.outAt {
   t.Errorf("test case: %v, %v -> %v", testCase.in, testCase.at, testCase.with)
   t.Errorf("incorrect index:")
   t.Errorf("  expected: %d", testCase.outAt)
   t.Errorf("       got: %d", gotAt)
  }
  if sameArray(origIn, got) != !testCase.reallocate {
   t.Errorf("test case: %v, %v -> %v", testCase.in, testCase.at, testCase.with)
   not := ""
   if !testCase.reallocate {
    not = " not"
   }
   t.Errorf("  expected to%s reallocate", not)
  }
 }
}

func sameArray(a, b moduleList) bool {
 return &a[0:cap(a)][cap(a)-1] == &b[0:cap(b)][cap(b)-1]
}