testing.go

package radiko

import (
	"io/ioutil"
	"net/http"
	"os"
	"path/filepath"
	"testing"
)

const areaIDTokyo = "JP13"

var (
	outsideJP bool

	testdataDir string
)

func init() {
	// Skip tests if outside Japan.
	OUTSIDEJP := os.Getenv("GO_RADIKO_OUTSIDE_JP")
	if len(OUTSIDEJP) > 0 {
		outsideJP = true
	}

	GOPATH := os.Getenv("GOPATH")
	testdataDir = filepath.Join(GOPATH, "src", "github.com/chikulla/go-radiko", "testdata")
}

// For skipping tests.
// radiko.jp restricts use from outside Japan.
func isOutsideJP() bool {
	return outsideJP
}

// Should restore defaultHTTPClient if SetHTTPClient is called.
func teardownHTTPClient() {
	SetHTTPClient(&http.Client{Timeout: defaultHTTPTimeout})
}

func createTestTempDir(t *testing.T) (string, func()) {
	dir, err := ioutil.TempDir("", "test-go-radiko")
	if err != nil {
		t.Fatalf("Failed to create temp dir: %s", err)
	}

	return dir, func() { os.RemoveAll(dir) }
}