package main_test

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

    "golang.org/x/tools/go/loader"
)

const (
    bar = `package bar`
    foo = `package foo; import _ "test.com/bar"`
    foo_test = `package foo_test; import _ "test.com/bar"`
)

func TestRepro(t *testing.T) {
    wd, _ := os.Getwd()
    tmpDir, _ := ioutil.TempDir(wd, "")
    defer func() { os.RemoveAll(tmpDir) }()

    ioutil.WriteFile(path.Join(tmpDir, "foo.go"), []byte(foo), 0644)
    barDir := path.Join(tmpDir, "vendor", "test.com", "bar")
    os.MkdirAll(barDir, 0755)
    ioutil.WriteFile(path.Join(barDir, "bar.go"), []byte(bar), 0644)

    loadcfg := loader.Config{}
    loadcfg.ImportWithTests("./" + path.Base(tmpDir))
    // succeeds if non-test file imports vendored package
    if _, err := loadcfg.Load(); err != nil {
        t.Errorf("Error loading package without test")
    }

    // add test file that imports the vendored directory
    ioutil.WriteFile(path.Join(tmpDir, "foo_test.go"), []byte(foo_test), 0644)

    loadcfg = loader.Config{}
    loadcfg.ImportWithTests("./" + path.Base(tmpDir))
    // fails -- only change was adding a test file that imports the same vendored package
    if _, err := loadcfg.Load(); err != nil {
        t.Errorf("Error loading package that contains test that imports vendored library")
    }

    loadcfg = loader.Config{}
    loadcfg.Cwd = tmpDir
    loadcfg.ImportWithTests(".")
    // succeeds with test file if Cwd is set to the directory that is at the same level as the "vendor" directory
    if _, err := loadcfg.Load(); err != nil {
        t.Errorf("Error loading package after Cwd is set")
    }
}