main_test.go

package main

import (
	"errors"
	"io/ioutil"
	"log"
	"path"
	"strings"
	"testing"

	"github.com/adamliesko/go-thanks/discover"
	"github.com/adamliesko/go-thanks/thank"
)

const goodAPIToken = "pass123"

func init() {
	log.SetOutput(ioutil.Discard)
}

func TestThankingTheGiants(t *testing.T) {
	*projectPath = path.Join("discover", "test", "project_dep")
	ft := &FakeGithubThanker{
		apiToken: goodAPIToken,
		thanked:  map[string]bool{},
	}
	err := thankGiants([]thank.Thanker{ft}, *projectPath)
	if err != nil {
		t.Fatal(err)
	}

	if len(ft.thanked) != 3 {
		t.Errorf("bad thanked count, got %d want %d", len(ft.thanked), 3)
	}
}

func TestThankingTheGiantsWithBadTokenFails(t *testing.T) {
	*projectPath = path.Join("discover", "test", "project_dep")
	ft := &FakeGithubThanker{
		apiToken: "bad-token",
		thanked:  map[string]bool{},
	}
	err := thankGiants([]thank.Thanker{ft}, *projectPath)
	if err == nil {
		t.Fatal(err)
	}

	if len(ft.thanked) != 0 {
		t.Errorf("bad thanked count, got %d want %d", len(ft.thanked), 0)
	}
}

func TestRunnerReturnsErrorIfNoThankerAuthenticated(t *testing.T) {
	*projectPath = path.Join("discover", "test", "project_dep")
	err := run()
	if err == nil {
		t.Fatal(err)
	}
}

type FakeGithubThanker struct {
	apiToken string
	thanked  map[string]bool
}

func (ft *FakeGithubThanker) CanThank(r discover.Repository) bool {
	return strings.HasPrefix(r.URL, "github.com")
}

func (ft *FakeGithubThanker) Thank(r discover.Repository) error {
	ft.thanked[r.URL] = true
	return nil
}

func (ft *FakeGithubThanker) Auth() error {
	if ft.apiToken == goodAPIToken {
		return nil
	}
	return errors.New("bad auth")
}