hercules/cmd/hercules/root_test.go

package main

import (
	"io/ioutil"
	"log"
	"os"
	"path/filepath"
	"runtime"
	"testing"

	"github.com/stretchr/testify/assert"
	"gopkg.in/src-d/go-billy.v4/osfs"
	"gopkg.in/src-d/go-git.v4"
	"gopkg.in/src-d/go-git.v4/plumbing/cache"
	"gopkg.in/src-d/go-git.v4/storage/filesystem"
)

func TestLoadRepository(t *testing.T) {
	repo := loadRepository("https://github.com/src-d/hercules", "", true, "")
	assert.NotNil(t, repo)
	log.Println("TestLoadRepository: 1/3")

	tempdir, err := ioutil.TempDir("", "hercules-")
	assert.Nil(t, err)
	if err != nil {
		assert.FailNow(t, "ioutil.TempDir")
	}
	defer os.RemoveAll(tempdir)
	backend := filesystem.NewStorage(osfs.New(tempdir), cache.NewObjectLRUDefault())
	cloneOptions := &git.CloneOptions{URL: "https://github.com/src-d/hercules"}
	_, err = git.Clone(backend, nil, cloneOptions)
	assert.Nil(t, err)
	if err != nil {
		assert.FailNow(t, "filesystem.NewStorage")
	}

	repo = loadRepository(tempdir, "", true, "")
	assert.NotNil(t, repo)
	log.Println("TestLoadRepository: 2/3")

	_, filename, _, _ := runtime.Caller(0)
	sivafile := filepath.Join(filepath.Dir(filename), "test_data", "hercules.siva")
	repo = loadRepository(sivafile, "", true, "")
	assert.NotNil(t, repo)
	log.Println("TestLoadRepository: 3/3")

	assert.Panics(t, func() { loadRepository("https://github.com/src-d/porn", "", true, "") })
	assert.Panics(t, func() { loadRepository(filepath.Dir(filename), "", true, "") })
	assert.Panics(t, func() { loadRepository("/xxx", "", true, "") })
}