github.com/mikespook/gleam@v0.0.0-20170807230223-29a9392b522c/gleam_test.go (about)

     1  package gleam
     2  
     3  import (
     4  	"os"
     5  	"strings"
     6  	"testing"
     7  )
     8  
     9  var (
    10  	wd string
    11  )
    12  
    13  func init() {
    14  	var err error
    15  	if wd, err = os.Getwd(); err != nil {
    16  		panic(err)
    17  	}
    18  }
    19  
    20  func resetWD() {
    21  	if err := os.Chdir(wd); err != nil {
    22  		panic(err)
    23  	}
    24  }
    25  
    26  func TestGleam(t *testing.T) {
    27  	resetWD()
    28  
    29  	g := NewGleam("./scripts/")
    30  	if err := g.Init(); err != nil {
    31  		t.Fatal(err)
    32  	}
    33  
    34  	hostname, err := os.Hostname()
    35  	if err != nil {
    36  		t.Fatal(err)
    37  	}
    38  	if !strings.HasPrefix(g.config.ClientId, hostname) {
    39  		t.Fatalf("Config error: %+v", g.config)
    40  	}
    41  
    42  	if err := g.Final(); err != nil {
    43  		t.Fatal(err)
    44  	}
    45  }