main_test.go

package main

import (
	"flag"
	"fmt"
	"io/ioutil"
	"os"
	"os/exec"
	"regexp"
	"strings"
	"testing"

	"github.com/davecgh/go-spew/spew"
	log "github.com/sirupsen/logrus"
)

// ConfigState for spew
var sp spew.ConfigState = spew.ConfigState{
	Indent:                  " ",
	DisablePointerAddresses: true,
	DisableCapacities:       true,
	SortKeys:                true,
	SpewKeys:                true,
	DisableMethods:          true,
}

func show(args ...interface{}) {
	fmt.Println(args...)
}

func showf(format string, args ...interface{}) {
	fmt.Printf(format+"\\n", args...)
}

func dump(args ...interface{}) {
	sp.Dump(args...)
}

func fdump(filenames ...string) {
	for _, filename := range filenames {
		buf, err := ioutil.ReadFile(filename)
		if err == nil {
			show(string(buf))
		} else {
			show(err)
		}
	}
}

func sdump(args ...interface{}) string {
	return sp.Sdump(args...)
}

func grep(search string, data ...interface{}) {
	re := regexp.MustCompile(search)
	lines := strings.Split(sp.Sdump(data...), "\n")
	for _, line := range lines {
		//show(line)
		if re.Match([]byte(line)) {
			show(strings.TrimSpace(line))
		}
	}
}

func run(cli string) string {
	res, err := exec.Command("sh", "-c", cli).CombinedOutput()
	if err != nil {
		return err.Error()
	}
	return string(res)
}

type recovering func()

func capture(fn recovering) {
	defer func() {
		if r := recover(); r != nil {
			show("capture:", r)
		}
	}()
	fn()
}

func debug(arg ...interface{}) {
	log.Debug(arg...)
}

func trace(arg ...interface{}) {
	log.Trace(arg...)
}

func traceOn() {
	log.SetLevel(log.TraceLevel)
}

func traceOff() {
	log.SetLevel(log.DebugLevel)
}

func TestMain(m *testing.M) {
	flag.Parse()
	*DryRunFlag = true
	*useDefaultAPIKey = true
	log.SetOutput(ioutil.Discard)
	//decomment this if you to see error log in tests
	//log.SetOutput(os.Stderr)
	log.SetFormatter(&log.TextFormatter{
		DisableColors:    true,
		DisableTimestamp: true,
	})
	log.SetLevel(log.DebugLevel)
	Version = "test"
	os.Setenv("HOME", "/tmp/iosdk-test")
	run("rm -Rvf /tmp/iosdk-test ; mkdir /tmp/iosdk-test")
	os.Exit(m.Run())
}

func first(arg interface{}, err error) interface{} {
	return arg
}

func last(arg interface{}, err error) error {
	return err
}

func Example() {
	parse("init")
}