Fix review suggestions

Signed-off-by: Vadim Markovtsev <vadim@sourced.tech>
This commit is contained in:
Vadim Markovtsev 2017-10-26 15:37:18 +02:00
parent 250519bb51
commit c97a180da5
3 changed files with 25 additions and 13 deletions

View File

@ -23,6 +23,7 @@ var (
overcomeLanguages []string overcomeLanguages []string
samples []*sample samples []*sample
samplesDir string samplesDir string
cloned bool
) )
func TestMain(m *testing.M) { func TestMain(m *testing.M) {
@ -35,7 +36,9 @@ func TestMain(m *testing.M) {
if err := cloneLinguist(linguistURL); err != nil { if err := cloneLinguist(linguistURL); err != nil {
log.Fatal(err) log.Fatal(err)
} }
if cloned {
defer os.RemoveAll(filepath.Dir(samplesDir)) defer os.RemoveAll(filepath.Dir(samplesDir))
}
var err error var err error
samples, err = getSamples(samplesDir) samples, err = getSamples(samplesDir)
@ -47,17 +50,24 @@ func TestMain(m *testing.M) {
} }
func cloneLinguist(linguistURL string) error { func cloneLinguist(linguistURL string) error {
repoLinguist, err := ioutil.TempDir("", "linguist-") repoLinguist := os.Getenv(linguistClonedEnvVar)
cloned = repoLinguist == ""
if cloned {
var err error
repoLinguist, err = ioutil.TempDir("", "linguist-")
if err != nil { if err != nil {
return err return err
} }
}
samplesDir = filepath.Join(repoLinguist, "samples") samplesDir = filepath.Join(repoLinguist, "samples")
if cloned {
cmd := exec.Command("git", "clone", linguistURL, repoLinguist) cmd := exec.Command("git", "clone", linguistURL, repoLinguist)
if err := cmd.Run(); err != nil { if err := cmd.Run(); err != nil {
return err return err
} }
}
cwd, err := os.Getwd() cwd, err := os.Getwd()
if err != nil { if err != nil {
@ -68,7 +78,7 @@ func cloneLinguist(linguistURL string) error {
return err return err
} }
cmd = exec.Command("git", "checkout", data.LinguistCommit) cmd := exec.Command("git", "checkout", data.LinguistCommit)
if err := cmd.Run(); err != nil { if err := cmd.Run(); err != nil {
return err return err
} }

View File

@ -15,6 +15,7 @@ import (
) )
const linguistURL = "https://github.com/github/linguist.git" const linguistURL = "https://github.com/github/linguist.git"
const linguistClonedEnvVar = "ENRY_TEST_REPO"
type EnryTestSuite struct { type EnryTestSuite struct {
suite.Suite suite.Suite
@ -29,8 +30,8 @@ func TestEnryTestSuite(t *testing.T) {
func (s *EnryTestSuite) SetupSuite() { func (s *EnryTestSuite) SetupSuite() {
var err error var err error
s.repoLinguist = os.Getenv("ENRY_TEST_REPO") s.repoLinguist = os.Getenv(linguistClonedEnvVar)
s.cloned = len(s.repoLinguist) == 0 s.cloned = s.repoLinguist == ""
if s.cloned { if s.cloned {
s.repoLinguist, err = ioutil.TempDir("", "linguist-") s.repoLinguist, err = ioutil.TempDir("", "linguist-")
assert.NoError(s.T(), err) assert.NoError(s.T(), err)

View File

@ -14,6 +14,7 @@ import (
const ( const (
linguistURL = "https://github.com/github/linguist.git" linguistURL = "https://github.com/github/linguist.git"
linguistClonedEnvVar = "ENRY_TEST_REPO"
commit = "d5c8db3fb91963c4b2762ca2ea2ff7cfac109f68" commit = "d5c8db3fb91963c4b2762ca2ea2ff7cfac109f68"
samplesDir = "samples" samplesDir = "samples"
languagesFile = "lib/linguist/languages.yml" languagesFile = "lib/linguist/languages.yml"
@ -89,8 +90,8 @@ func TestGeneratorTestSuite(t *testing.T) {
func (s *GeneratorTestSuite) SetupSuite() { func (s *GeneratorTestSuite) SetupSuite() {
var err error var err error
s.tmpLinguist = os.Getenv("ENRY_TEST_REPO") s.tmpLinguist = os.Getenv(linguistClonedEnvVar)
s.cloned = len(s.tmpLinguist) == 0 s.cloned = s.tmpLinguist == ""
if s.cloned { if s.cloned {
s.tmpLinguist, err = ioutil.TempDir("", "linguist-") s.tmpLinguist, err = ioutil.TempDir("", "linguist-")
assert.NoError(s.T(), err) assert.NoError(s.T(), err)