sqlboiler/boilingcore/imports_test.go

package boilingcore

import (
	"reflect"
	"sort"
	"testing"

	"github.com/pkg/errors"
	"github.com/vattle/sqlboiler/bdb"
)

func TestImportsSort(t *testing.T) {
	t.Parallel()

	a1 := importList{
		`"fmt"`,
		`"errors"`,
	}
	a2 := importList{
		`_ "github.com/lib/pq"`,
		`_ "github.com/gorilla/n"`,
		`"github.com/gorilla/mux"`,
		`"github.com/gorilla/websocket"`,
	}

	a1Expected := importList{`"errors"`, `"fmt"`}
	a2Expected := importList{
		`"github.com/gorilla/mux"`,
		`_ "github.com/gorilla/n"`,
		`"github.com/gorilla/websocket"`,
		`_ "github.com/lib/pq"`,
	}

	sort.Sort(a1)
	if !reflect.DeepEqual(a1, a1Expected) {
		t.Errorf("Expected a1 to match a1Expected, got: %v", a1)
	}

	for i, v := range a1 {
		if v != a1Expected[i] {
			t.Errorf("Expected a1[%d] to match a1Expected[%d]:\n%s\n%s\n", i, i, v, a1Expected[i])
		}
	}

	sort.Sort(a2)
	if !reflect.DeepEqual(a2, a2Expected) {
		t.Errorf("Expected a2 to match a2expected, got: %v", a2)
	}

	for i, v := range a2 {
		if v != a2Expected[i] {
			t.Errorf("Expected a2[%d] to match a2Expected[%d]:\n%s\n%s\n", i, i, v, a1Expected[i])
		}
	}
}

func TestImportsAddAndRemove(t *testing.T) {
	t.Parallel()

	var imp imports
	imp.Add("value", false)
	if len(imp.standard) != 1 {
		t.Errorf("expected len 1, got %d", len(imp.standard))
	}
	if imp.standard[0] != "value" {
		t.Errorf("expected %q to be added", "value")
	}
	imp.Add("value2", true)
	if len(imp.thirdParty) != 1 {
		t.Errorf("expected len 1, got %d", len(imp.thirdParty))
	}
	if imp.thirdParty[0] != "value2" {
		t.Errorf("expected %q to be added", "value2")
	}

	imp.Remove("value")
	if len(imp.standard) != 0 {
		t.Errorf("expected len 0, got %d", len(imp.standard))
	}
	imp.Remove("value")
	if len(imp.standard) != 0 {
		t.Errorf("expected len 0, got %d", len(imp.standard))
	}
	imp.Remove("value2")
	if len(imp.thirdParty) != 0 {
		t.Errorf("expected len 0, got %d", len(imp.thirdParty))
	}

	// Test deleting last element in len 2 slice
	imp.Add("value3", false)
	imp.Add("value4", false)
	if len(imp.standard) != 2 {
		t.Errorf("expected len 2, got %d", len(imp.standard))
	}
	imp.Remove("value4")
	if len(imp.standard) != 1 {
		t.Errorf("expected len 1, got %d", len(imp.standard))
	}
	if imp.standard[0] != "value3" {
		t.Errorf("expected %q, got %q", "value3", imp.standard[0])
	}
	// Test deleting first element in len 2 slice
	imp.Add("value4", false)
	imp.Remove("value3")
	if len(imp.standard) != 1 {
		t.Errorf("expected len 1, got %d", len(imp.standard))
	}
	if imp.standard[0] != "value4" {
		t.Errorf("expected %q, got %q", "value4", imp.standard[0])
	}
	imp.Remove("value2")
	if len(imp.thirdParty) != 0 {
		t.Errorf("expected len 0, got %d", len(imp.thirdParty))
	}

	// Test deleting last element in len 2 slice
	imp.Add("value5", true)
	imp.Add("value6", true)
	if len(imp.thirdParty) != 2 {
		t.Errorf("expected len 2, got %d", len(imp.thirdParty))
	}
	imp.Remove("value6")
	if len(imp.thirdParty) != 1 {
		t.Errorf("expected len 1, got %d", len(imp.thirdParty))
	}
	if imp.thirdParty[0] != "value5" {
		t.Errorf("expected %q, got %q", "value5", imp.thirdParty[0])
	}
	// Test deleting first element in len 2 slice
	imp.Add("value6", true)
	imp.Remove("value5")
	if len(imp.thirdParty) != 1 {
		t.Errorf("expected len 1, got %d", len(imp.thirdParty))
	}
	if imp.thirdParty[0] != "value6" {
		t.Errorf("expected %q, got %q", "value6", imp.thirdParty[0])
	}
}

func TestMapImportsAddAndRemove(t *testing.T) {
	t.Parallel()

	imp := mapImports{}
	imp.Add("cat", "value", false)
	if len(imp["cat"].standard) != 1 {
		t.Errorf("expected len 1, got %d", len(imp["cat"].standard))
	}
	if imp["cat"].standard[0] != "value" {
		t.Errorf("expected %q to be added", "value")
	}
	imp.Add("cat", "value2", true)
	if len(imp["cat"].thirdParty) != 1 {
		t.Errorf("expected len 1, got %d", len(imp["cat"].thirdParty))
	}
	if imp["cat"].thirdParty[0] != "value2" {
		t.Errorf("expected %q to be added", "value2")
	}

	imp.Remove("cat", "value")
	if len(imp["cat"].standard) != 0 {
		t.Errorf("expected len 0, got %d", len(imp["cat"].standard))
	}
	imp.Remove("cat", "value")
	if len(imp["cat"].standard) != 0 {
		t.Errorf("expected len 0, got %d", len(imp["cat"].standard))
	}
	imp.Remove("cat", "value2")
	if len(imp["cat"].thirdParty) != 0 {
		t.Errorf("expected len 0, got %d", len(imp["cat"].thirdParty))
	}
	// If there are no elements left in key, test key is deleted
	_, ok := imp["cat"]
	if ok {
		t.Errorf("expected cat key to be deleted when list empty")
	}

	// Test deleting last element in len 2 slice
	imp.Add("cat", "value3", false)
	imp.Add("cat", "value4", false)
	if len(imp["cat"].standard) != 2 {
		t.Errorf("expected len 2, got %d", len(imp["cat"].standard))
	}
	imp.Remove("cat", "value4")
	if len(imp["cat"].standard) != 1 {
		t.Errorf("expected len 1, got %d", len(imp["cat"].standard))
	}
	if imp["cat"].standard[0] != "value3" {
		t.Errorf("expected %q, got %q", "value3", imp["cat"].standard[0])
	}
	// Test deleting first element in len 2 slice
	imp.Add("cat", "value4", false)
	imp.Remove("cat", "value3")
	if len(imp["cat"].standard) != 1 {
		t.Errorf("expected len 1, got %d", len(imp["cat"].standard))
	}
	if imp["cat"].standard[0] != "value4" {
		t.Errorf("expected %q, got %q", "value4", imp["cat"].standard[0])
	}
	imp.Remove("cat", "value2")
	if len(imp["cat"].thirdParty) != 0 {
		t.Errorf("expected len 0, got %d", len(imp["cat"].thirdParty))
	}

	// Test deleting last element in len 2 slice
	imp.Add("dog", "value5", true)
	imp.Add("dog", "value6", true)
	if len(imp["dog"].thirdParty) != 2 {
		t.Errorf("expected len 2, got %d", len(imp["dog"].thirdParty))
	}
	imp.Remove("dog", "value6")
	if len(imp["dog"].thirdParty) != 1 {
		t.Errorf("expected len 1, got %d", len(imp["dog"].thirdParty))
	}
	if imp["dog"].thirdParty[0] != "value5" {
		t.Errorf("expected %q, got %q", "value5", imp["dog"].thirdParty[0])
	}
	// Test deleting first element in len 2 slice
	imp.Add("dog", "value6", true)
	imp.Remove("dog", "value5")
	if len(imp["dog"].thirdParty) != 1 {
		t.Errorf("expected len 1, got %d", len(imp["dog"].thirdParty))
	}
	if imp["dog"].thirdParty[0] != "value6" {
		t.Errorf("expected %q, got %q", "value6", imp["dog"].thirdParty[0])
	}
}

func TestCombineTypeImports(t *testing.T) {
	t.Parallel()

	imports1 := imports{
		standard: importList{
			`"errors"`,
			`"fmt"`,
		},
		thirdParty: importList{
			`"github.com/vattle/sqlboiler/boil"`,
		},
	}

	importsExpected := imports{
		standard: importList{
			`"errors"`,
			`"fmt"`,
			`"time"`,
		},
		thirdParty: importList{
			`"github.com/vattle/sqlboiler/boil"`,
			`"gopkg.in/nullbio/null.v6"`,
		},
	}

	cols := []bdb.Column{
		{
			Type: "null.Time",
		},
		{
			Type: "null.Time",
		},
		{
			Type: "time.Time",
		},
		{
			Type: "null.Float",
		},
	}

	imps := newImporter()

	res1 := combineTypeImports(imports1, imps.basedOnType, cols)

	if !reflect.DeepEqual(res1, importsExpected) {
		t.Errorf("Expected res1 to match importsExpected, got:\n\n%#v\n", res1)
	}

	imports2 := imports{
		standard: importList{
			`"errors"`,
			`"fmt"`,
			`"time"`,
		},
		thirdParty: importList{
			`"github.com/vattle/sqlboiler/boil"`,
			`"gopkg.in/nullbio/null.v6"`,
		},
	}

	res2 := combineTypeImports(imports2, imps.basedOnType, cols)

	if !reflect.DeepEqual(res2, importsExpected) {
		t.Errorf("Expected res2 to match importsExpected, got:\n\n%#v\n", res1)
	}
}

func TestCombineImports(t *testing.T) {
	t.Parallel()

	a := imports{
		standard:   importList{"fmt"},
		thirdParty: importList{"github.com/vattle/sqlboiler", "gopkg.in/nullbio/null.v6"},
	}
	b := imports{
		standard:   importList{"os"},
		thirdParty: importList{"github.com/vattle/sqlboiler"},
	}

	c := combineImports(a, b)

	if c.standard[0] != "fmt" && c.standard[1] != "os" {
		t.Errorf("Wanted: fmt, os got: %#v", c.standard)
	}
	if c.thirdParty[0] != "github.com/vattle/sqlboiler" && c.thirdParty[1] != "gopkg.in/nullbio/null.v6" {
		t.Errorf("Wanted: github.com/vattle/sqlboiler, gopkg.in/nullbio/null.v6 got: %#v", c.thirdParty)
	}
}

func TestRemoveDuplicates(t *testing.T) {
	t.Parallel()

	hasDups := func(possible []string) error {
		for i := 0; i < len(possible)-1; i++ {
			for j := i + 1; j < len(possible); j++ {
				if possible[i] == possible[j] {
					return errors.Errorf("found duplicate: %s [%d] [%d]", possible[i], i, j)
				}
			}
		}

		return nil
	}

	if len(removeDuplicates([]string{})) != 0 {
		t.Error("It should have returned an empty slice")
	}

	oneItem := []string{"patrick"}
	slice := removeDuplicates(oneItem)
	if ln := len(slice); ln != 1 {
		t.Error("Length was wrong:", ln)
	} else if oneItem[0] != slice[0] {
		t.Errorf("Slices differ: %#v %#v", oneItem, slice)
	}

	slice = removeDuplicates([]string{"hello", "patrick", "hello"})
	if ln := len(slice); ln != 2 {
		t.Error("Length was wrong:", ln)
	}
	if err := hasDups(slice); err != nil {
		t.Error(err)
	}

	slice = removeDuplicates([]string{"five", "patrick", "hello", "hello", "patrick", "hello", "hello"})
	if ln := len(slice); ln != 3 {
		t.Error("Length was wrong:", ln)
	}
	if err := hasDups(slice); err != nil {
		t.Error(err)
	}
}

func TestCombineStringSlices(t *testing.T) {
	t.Parallel()

	var a, b []string
	slice := combineStringSlices(a, b)
	if ln := len(slice); ln != 0 {
		t.Error("Len was wrong:", ln)
	}

	a = []string{"1", "2"}
	slice = combineStringSlices(a, b)
	if ln := len(slice); ln != 2 {
		t.Error("Len was wrong:", ln)
	} else if slice[0] != a[0] || slice[1] != a[1] {
		t.Errorf("Slice mismatch: %#v %#v", a, slice)
	}

	b = a
	a = nil
	slice = combineStringSlices(a, b)
	if ln := len(slice); ln != 2 {
		t.Error("Len was wrong:", ln)
	} else if slice[0] != b[0] || slice[1] != b[1] {
		t.Errorf("Slice mismatch: %#v %#v", b, slice)
	}

	a = b
	b = []string{"3", "4"}
	slice = combineStringSlices(a, b)
	if ln := len(slice); ln != 4 {
		t.Error("Len was wrong:", ln)
	} else if slice[0] != a[0] || slice[1] != a[1] || slice[2] != b[0] || slice[3] != b[1] {
		t.Errorf("Slice mismatch: %#v + %#v != #%v", a, b, slice)
	}
}