diff --git a/pipeline/release/body_test.go b/pipeline/release/body_test.go index d5edf043e..5a28fa370 100644 --- a/pipeline/release/body_test.go +++ b/pipeline/release/body_test.go @@ -1,6 +1,7 @@ package release import ( + "flag" "io/ioutil" "os" "testing" @@ -11,6 +12,8 @@ import ( "github.com/stretchr/testify/assert" ) +var update = flag.Bool("update", false, "update .golden files") + func TestDescribeBody(t *testing.T) { var changelog = "\nfeature1: description\nfeature2: other description" var ctx = context.New(config.Project{}) @@ -28,10 +31,12 @@ func TestDescribeBody(t *testing.T) { out, err := describeBodyVersion(ctx, "go version go1.9 darwin/amd64") assert.NoError(t, err) - bts, err := ioutil.ReadFile("testdata/release1.txt") + var golden = "testdata/release1.golden" + if *update { + ioutil.WriteFile(golden, out.Bytes(), 0755) + } + bts, err := ioutil.ReadFile(golden) assert.NoError(t, err) - // ioutil.WriteFile("testdata/release1.txt", out.Bytes(), 0755) - assert.Equal(t, string(bts), out.String()) } @@ -43,9 +48,12 @@ func TestDescribeBodyNoDockerImagesNoBrews(t *testing.T) { out, err := describeBodyVersion(ctx, "go version go1.9 darwin/amd64") assert.NoError(t, err) - bts, err := ioutil.ReadFile("testdata/release2.txt") + var golden = "testdata/release2.golden" + if *update { + ioutil.WriteFile(golden, out.Bytes(), 0655) + } + bts, err := ioutil.ReadFile(golden) assert.NoError(t, err) - // ioutil.WriteFile("testdata/release2.txt", out.Bytes(), 0755) assert.Equal(t, string(bts), out.String()) } diff --git a/pipeline/release/testdata/release1.txt b/pipeline/release/testdata/release1.golden old mode 100644 new mode 100755 similarity index 100% rename from pipeline/release/testdata/release1.txt rename to pipeline/release/testdata/release1.golden diff --git a/pipeline/release/testdata/release2.txt b/pipeline/release/testdata/release2.golden old mode 100755 new mode 100644 similarity index 100% rename from pipeline/release/testdata/release2.txt rename to pipeline/release/testdata/release2.golden