| // Copyright 2021 The Go Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style |
| // license that can be found in the LICENSE file. |
| |
| package task |
| |
| import ( |
| "bytes" |
| "context" |
| "encoding/json" |
| "flag" |
| "fmt" |
| "image" |
| "image/png" |
| "io" |
| "net/http" |
| "net/http/httptest" |
| "os" |
| "path/filepath" |
| "reflect" |
| "testing" |
| |
| "github.com/google/go-cmp/cmp" |
| "golang.org/x/build/internal/workflow" |
| ) |
| |
| func TestTweetRelease(t *testing.T) { |
| tests := [...]struct { |
| name string |
| kind ReleaseKind |
| published []Published |
| security string |
| announcement string |
| randomSeed int64 |
| wantLog string |
| }{ |
| { |
| name: "minor", |
| kind: KindMinor, |
| published: []Published{ |
| {Version: "go1.17.1", Files: []WebsiteFile{{ |
| OS: "linux", Arch: "arm64", |
| Filename: "go1.17.1.linux-arm64.tar.gz", Size: 102606384, Kind: "archive"}}, |
| }, |
| {Version: "go1.16.8"}, |
| }, |
| security: "Includes security fixes for A and B.", |
| announcement: "https://groups.google.com/g/golang-announce/c/dx9d7IOseHw/m/KNH37k37AAAJ", |
| randomSeed: 234, |
| wantLog: `tweet text: |
| 🎊 Go 1.17.1 and 1.16.8 are released! |
| |
| 🔐 Security: Includes security fixes for A and B. |
| |
| 📢 Announcement: https://groups.google.com/g/golang-announce/c/dx9d7IOseHw/m/KNH37k37AAAJ |
| |
| ⬇️ Download: https://go.dev/dl/#go1.17.1 |
| |
| #golang |
| tweet image: |
| $ go install golang.org/dl/go1.17.1@latest |
| $ go1.17.1 download |
| Downloaded 0.0% ( 0 / 102606384 bytes) ... |
| Downloaded 50.0% ( 51303192 / 102606384 bytes) ... |
| Downloaded 100.0% (102606384 / 102606384 bytes) |
| Unpacking go1.17.1.linux-arm64.tar.gz ... |
| Success. You may now run 'go1.17.1' |
| $ go1.17.1 version |
| go version go1.17.1 linux/arm64` + "\n", |
| }, |
| { |
| name: "minor-solo", |
| kind: KindMinor, |
| published: []Published{{Version: "go1.11.1", Files: []WebsiteFile{{ |
| OS: "darwin", Arch: "amd64", |
| Filename: "go1.11.1.darwin-amd64.tar.gz", Size: 124181190, Kind: "archive"}}, |
| }}, |
| announcement: "https://groups.google.com/g/golang-announce/c/pFXKAfoVJqw", |
| randomSeed: 23, |
| wantLog: `tweet text: |
| 🎆 Go 1.11.1 is released! |
| |
| 📣 Announcement: https://groups.google.com/g/golang-announce/c/pFXKAfoVJqw |
| |
| 📦 Download: https://go.dev/dl/#go1.11.1 |
| |
| #golang |
| tweet image: |
| $ go install golang.org/dl/go1.11.1@latest |
| $ go1.11.1 download |
| Downloaded 0.0% ( 0 / 124181190 bytes) ... |
| Downloaded 50.0% ( 62090595 / 124181190 bytes) ... |
| Downloaded 100.0% (124181190 / 124181190 bytes) |
| Unpacking go1.11.1.darwin-amd64.tar.gz ... |
| Success. You may now run 'go1.11.1' |
| $ go1.11.1 version |
| go version go1.11.1 darwin/amd64` + "\n", |
| }, |
| { |
| name: "beta", |
| kind: KindBeta, |
| published: []Published{{Version: "go1.17beta1", Files: []WebsiteFile{{ |
| OS: "darwin", Arch: "amd64", |
| Filename: "go1.17beta1.darwin-amd64.tar.gz", Size: 135610703, Kind: "archive"}}, |
| }}, |
| announcement: "https://groups.google.com/g/golang-announce/c/i4EliPDV9Ok/m/MxA-nj53AAAJ", |
| randomSeed: 678, |
| wantLog: `tweet text: |
| ⚡️ Go 1.17 Beta 1 is released! |
| |
| ⚙️ Try it! File bugs! https://go.dev/issue/new |
| |
| 🗣 Announcement: https://groups.google.com/g/golang-announce/c/i4EliPDV9Ok/m/MxA-nj53AAAJ |
| |
| 📦 Download: https://go.dev/dl/#go1.17beta1 |
| |
| #golang |
| tweet image: |
| $ go install golang.org/dl/go1.17beta1@latest |
| $ go1.17beta1 download |
| Downloaded 0.0% ( 0 / 135610703 bytes) ... |
| Downloaded 50.0% ( 67805351 / 135610703 bytes) ... |
| Downloaded 100.0% (135610703 / 135610703 bytes) |
| Unpacking go1.17beta1.darwin-amd64.tar.gz ... |
| Success. You may now run 'go1.17beta1' |
| $ go1.17beta1 version |
| go version go1.17beta1 darwin/amd64` + "\n", |
| }, |
| { |
| name: "rc", |
| kind: KindRC, |
| published: []Published{{Version: "go1.17rc2", Files: []WebsiteFile{{ |
| OS: "windows", Arch: "arm64", |
| Filename: "go1.17rc2.windows-arm64.zip", Size: 116660997, Kind: "archive"}}, |
| }}, |
| announcement: "https://groups.google.com/g/golang-announce/c/yk30ovJGXWY/m/p9uUnKbbBQAJ", |
| randomSeed: 456, |
| wantLog: `tweet text: |
| 🎉 Go 1.17 Release Candidate 2 is released! |
| |
| 🏖 Run it in dev! Run it in prod! File bugs! https://go.dev/issue/new |
| |
| 🔈 Announcement: https://groups.google.com/g/golang-announce/c/yk30ovJGXWY/m/p9uUnKbbBQAJ |
| |
| 📦 Download: https://go.dev/dl/#go1.17rc2 |
| |
| #golang |
| tweet image: |
| $ go install golang.org/dl/go1.17rc2@latest |
| $ go1.17rc2 download |
| Downloaded 0.0% ( 0 / 116660997 bytes) ... |
| Downloaded 50.0% ( 58330498 / 116660997 bytes) ... |
| Downloaded 100.0% (116660997 / 116660997 bytes) |
| Unpacking go1.17rc2.windows-arm64.zip ... |
| Success. You may now run 'go1.17rc2' |
| $ go1.17rc2 version |
| go version go1.17rc2 windows/arm64` + "\n", |
| }, |
| { |
| name: "major", |
| kind: KindMajor, |
| published: []Published{{Version: "go1.21.0", Files: []WebsiteFile{{ |
| OS: "freebsd", Arch: "amd64", |
| Filename: "go1.21.0.freebsd-amd64.tar.gz", Size: 133579378, Kind: "archive"}}, |
| }}, |
| security: "Includes a super duper security fix (CVE-123).", |
| randomSeed: 123, |
| wantLog: `tweet text: |
| 🥳 Go 1.21.0 is released! |
| |
| 🔐 Security: Includes a super duper security fix (CVE-123). |
| |
| 📝 Release notes: https://go.dev/doc/go1.21 |
| |
| 📦 Download: https://go.dev/dl/#go1.21.0 |
| |
| #golang |
| tweet image: |
| $ go install golang.org/dl/go1.21.0@latest |
| $ go1.21.0 download |
| Downloaded 0.0% ( 0 / 133579378 bytes) ... |
| Downloaded 50.0% ( 66789689 / 133579378 bytes) ... |
| Downloaded 100.0% (133579378 / 133579378 bytes) |
| Unpacking go1.21.0.freebsd-amd64.tar.gz ... |
| Success. You may now run 'go1.21.0' |
| $ go1.21.0 version |
| go version go1.21.0 freebsd/amd64` + "\n", |
| }, |
| } |
| for _, tc := range tests { |
| t.Run(tc.name, func(t *testing.T) { |
| // Call the tweet task function in dry-run mode so it |
| // doesn't actually try to tweet, but capture its log. |
| var buf bytes.Buffer |
| ctx := &workflow.TaskContext{Context: context.Background(), Logger: fmtWriter{&buf}} |
| tweetURL, err := (TweetTasks{RandomSeed: tc.randomSeed}).TweetRelease(ctx, tc.kind, tc.published, tc.security, tc.announcement) |
| if err != nil { |
| t.Fatal("got a non-nil error:", err) |
| } |
| if got, want := tweetURL, "(dry-run)"; got != want { |
| t.Errorf("unexpected tweetURL: got = %q, want %q", got, want) |
| } |
| if diff := cmp.Diff(tc.wantLog, buf.String()); diff != "" { |
| t.Errorf("log mismatch (-want +got):\n%s", diff) |
| } |
| }) |
| } |
| } |
| |
| type fmtWriter struct{ w io.Writer } |
| |
| func (f fmtWriter) Printf(format string, v ...interface{}) { |
| fmt.Fprintf(f.w, format, v...) |
| } |
| |
| var updateFlag = flag.Bool("update", false, "Update golden files.") |
| |
| func TestDrawTerminal(t *testing.T) { |
| got, err := drawTerminal(`$ go install golang.org/dl/go1.18beta1@latest |
| $ go1.18beta1 download |
| Downloaded 0.0% ( 0 / 111109966 bytes) ... |
| Downloaded 50.0% ( 55554983 / 111109966 bytes) ... |
| Downloaded 100.0% (111109966 / 111109966 bytes) |
| Unpacking go1.18beta1.linux-s390x.tar.gz ... |
| Success. You may now run 'go1.18beta1' |
| $ go1.18beta1 version |
| go version go1.18beta1 linux/s390x`) |
| if err != nil { |
| t.Fatalf("drawTerminal: got error=%v, want nil", err) |
| } |
| if *updateFlag { |
| encodePNG(t, filepath.Join("testdata", "terminal.png"), got) |
| return |
| } |
| want := decodePNG(t, filepath.Join("testdata", "terminal.png")) |
| if !got.Bounds().Eq(want.Bounds()) { |
| t.Fatalf("drawTerminal: got image bounds=%v, want %v", got.Bounds(), want.Bounds()) |
| } |
| diff := func(a, b uint32) uint64 { |
| if a < b { |
| return uint64(b - a) |
| } |
| return uint64(a - b) |
| } |
| var total uint64 |
| for y := 0; y < want.Bounds().Dy(); y++ { |
| for x := 0; x < want.Bounds().Dx(); x++ { |
| r0, g0, b0, a0 := got.At(x, y).RGBA() |
| r1, g1, b1, a1 := want.At(x, y).RGBA() |
| const D = 0xffff * 20 / 100 // Diff threshold of 20% for RGB color components. |
| if diff(r0, r1) > D || diff(g0, g1) > D || diff(b0, b1) > D || a0 != a1 { |
| t.Errorf("at (%d, %d):\n got RGBA %v\nwant RGBA %v", x, y, got.At(x, y), want.At(x, y)) |
| } |
| total += diff(r0, r1) + diff(g0, g1) + diff(b0, b1) |
| } |
| } |
| if testing.Verbose() { |
| t.Logf("average pixel color diff: %v%%", 100*float64(total)/float64(0xffff*want.Bounds().Dx()*want.Bounds().Dy())) |
| } |
| } |
| |
| func encodePNG(t *testing.T, name string, m image.Image) { |
| t.Helper() |
| var buf bytes.Buffer |
| err := (&png.Encoder{CompressionLevel: png.BestCompression}).Encode(&buf, m) |
| if err != nil { |
| t.Fatal(err) |
| } |
| err = os.WriteFile(name, buf.Bytes(), 0644) |
| if err != nil { |
| t.Fatal(err) |
| } |
| } |
| |
| func decodePNG(t *testing.T, name string) image.Image { |
| t.Helper() |
| f, err := os.Open(name) |
| if err != nil { |
| t.Fatal(err) |
| } |
| defer f.Close() |
| m, err := png.Decode(f) |
| if err != nil { |
| t.Fatal(err) |
| } |
| return m |
| } |
| |
| func TestPostTweet(t *testing.T) { |
| mux := http.NewServeMux() |
| mux.HandleFunc("upload.twitter.com/1.1/media/upload.json", func(w http.ResponseWriter, req *http.Request) { |
| if got, want := req.Method, http.MethodPost; got != want { |
| t.Errorf("/1.1/media/upload.json: got method %s, want %s", got, want) |
| return |
| } |
| if got, want := req.FormValue("media_category"), "tweet_image"; got != want { |
| t.Errorf("/1.1/media/upload.json: got media_category=%q, want %q", got, want) |
| } |
| f, hdr, err := req.FormFile("media") |
| if err != nil { |
| t.Errorf("/1.1/media/upload.json: error getting image file: %v", err) |
| return |
| } |
| if got, want := hdr.Filename, "image.png"; got != want { |
| t.Errorf("/1.1/media/upload.json: got file name=%q, want %q", got, want) |
| } |
| if got, want := mustRead(f), "image-png-bytes"; got != want { |
| t.Errorf("/1.1/media/upload.json: got file content=%q, want %q", got, want) |
| return |
| } |
| mustWrite(w, `{"media_id_string": "media-123"}`) |
| }) |
| mux.HandleFunc("api.twitter.com/2/tweets", func(w http.ResponseWriter, req *http.Request) { |
| if got, want := req.Method, http.MethodPost; got != want { |
| t.Errorf("/2/tweets: got method %s, want %s", got, want) |
| return |
| } |
| if got, want := req.Header.Get("Content-Type"), "application/json"; got != want { |
| t.Errorf("/2/tweets: got Content-Type=%q, want %q", got, want) |
| return |
| } |
| var v struct { |
| Text string `json:"text"` |
| Media struct { |
| MediaIDs []string `json:"media_ids"` |
| } `json:"media"` |
| } |
| if err := json.NewDecoder(req.Body).Decode(&v); err != nil { |
| t.Errorf("/2/tweets: decode JSON error: %v", err) |
| return |
| } |
| if got, want := v.Text, "tweet-text"; got != want { |
| t.Errorf("/2/tweets: got status=%q, want %q", got, want) |
| } |
| if got, want := v.Media.MediaIDs, []string{"media-123"}; !reflect.DeepEqual(got, want) { |
| t.Errorf("/2/tweets: got media_ids=%q, want %q", got, want) |
| } |
| w.WriteHeader(http.StatusCreated) |
| mustWrite(w, `{"data": {"id": "tweet-123"}}`) |
| }) |
| cl := realTwitterClient{twitterAPI: &http.Client{Transport: localRoundTripper{mux}}} |
| |
| tweetURL, err := cl.PostTweet("tweet-text", []byte("image-png-bytes")) |
| if err != nil { |
| t.Fatal("PostTweet:", err) |
| } |
| if got, want := tweetURL, "https://twitter.com/username/status/tweet-123"; got != want { |
| t.Errorf("got tweetURL=%q, want %q", got, want) |
| } |
| } |
| |
| // localRoundTripper is an http.RoundTripper that executes HTTP transactions |
| // by using handler directly, instead of going over an HTTP connection. |
| type localRoundTripper struct { |
| handler http.Handler |
| } |
| |
| func (l localRoundTripper) RoundTrip(req *http.Request) (*http.Response, error) { |
| w := httptest.NewRecorder() |
| l.handler.ServeHTTP(w, req) |
| return w.Result(), nil |
| } |
| |
| func mustRead(r io.Reader) string { |
| b, err := io.ReadAll(r) |
| if err != nil { |
| panic(err) |
| } |
| return string(b) |
| } |
| |
| func mustWrite(w io.Writer, s string) { |
| _, err := io.WriteString(w, s) |
| if err != nil { |
| panic(err) |
| } |
| } |