| // Copyright 2019 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 middleware |
| |
| import ( |
| "context" |
| "fmt" |
| "io/ioutil" |
| "net/http" |
| "net/http/httptest" |
| "testing" |
| |
| "golang.org/x/pkgsite/internal" |
| "golang.org/x/pkgsite/internal/experiment" |
| ) |
| |
| func TestLatestMinorVersion(t *testing.T) { |
| for _, test := range []struct { |
| name string |
| latest internal.LatestInfo |
| in string |
| want string |
| }{ |
| { |
| name: "package version is not latest", |
| latest: internal.LatestInfo{MinorVersion: "v1.2.3"}, |
| in: ` |
| <div class="DetailsHeader-badge $$GODISCOVERY_LATESTMINORCLASS$$" |
| data-version="v1.0.0" data-mpath="p1/p2" data-ppath="p1/p2/p3" data-pagetype="pkg"> |
| <span>Latest</span> |
| <a href="p1/p2@$$GODISCOVERY_LATESTMINORVERSION$$/p3">Go to latest</a> |
| </div>`, |
| want: ` |
| <div class="DetailsHeader-badge DetailsHeader-badge--goToLatest" |
| data-version="v1.0.0" data-mpath="p1/p2" data-ppath="p1/p2/p3" data-pagetype="pkg"> |
| <span>Latest</span> |
| <a href="p1/p2@v1.2.3/p3">Go to latest</a> |
| </div>`, |
| }, |
| { |
| name: "package version is latest", |
| latest: internal.LatestInfo{MinorVersion: "v1.2.3", UnitExistsAtMinor: true}, |
| in: ` |
| <div class="DetailsHeader-badge $$GODISCOVERY_LATESTMINORCLASS$$" |
| data-version="v1.2.3" data-mpath="p1/p2" data-ppath="p1/p2/p3" data-pagetype="pkg"> |
| <span>Latest</span> |
| <a href="p1/p2@$$GODISCOVERY_LATESTMINORVERSION$$/p3">Go to latest</a> |
| </div>`, |
| want: ` |
| <div class="DetailsHeader-badge DetailsHeader-badge--latest" |
| data-version="v1.2.3" data-mpath="p1/p2" data-ppath="p1/p2/p3" data-pagetype="pkg"> |
| <span>Latest</span> |
| <a href="p1/p2@v1.2.3/p3">Go to latest</a> |
| </div>`, |
| }, |
| { |
| name: "package version with build is latest", |
| latest: internal.LatestInfo{MinorVersion: "v1.2.3+build", UnitExistsAtMinor: true}, |
| in: ` |
| <div class="DetailsHeader-badge $$GODISCOVERY_LATESTMINORCLASS$$" |
| data-version="v1.2.3+build" data-mpath="p1/p2" data-ppath="p1/p2/p3" data-pagetype="pkg"> |
| <span>Latest</span> |
| <a href="p1/p2@$$GODISCOVERY_LATESTMINORVERSION$$/p3">Go to latest</a> |
| </div>`, |
| want: ` |
| <div class="DetailsHeader-badge DetailsHeader-badge--latest" |
| data-version="v1.2.3+build" data-mpath="p1/p2" data-ppath="p1/p2/p3" data-pagetype="pkg"> |
| <span>Latest</span> |
| <a href="p1/p2@v1.2.3+build/p3">Go to latest</a> |
| </div>`, |
| }, |
| { |
| name: "module version is not latest", |
| latest: internal.LatestInfo{MinorVersion: "v1.2.3"}, |
| in: ` |
| <div class="DetailsHeader-badge $$GODISCOVERY_LATESTMINORCLASS$$" |
| data-version="v1.0.0" data-mpath="p1/p2" data-ppath="" data-pagetype="pkg"> |
| <span>Latest</span> |
| <a href="mod/p1/p2@$$GODISCOVERY_LATESTMINORVERSION$$">Go to latest</a> |
| </div>`, |
| want: ` |
| <div class="DetailsHeader-badge DetailsHeader-badge--goToLatest" |
| data-version="v1.0.0" data-mpath="p1/p2" data-ppath="" data-pagetype="pkg"> |
| <span>Latest</span> |
| <a href="mod/p1/p2@v1.2.3">Go to latest</a> |
| </div>`, |
| }, |
| { |
| name: "module version is latest", |
| latest: internal.LatestInfo{MinorVersion: "v1.2.3", UnitExistsAtMinor: true}, |
| in: ` |
| <div class="DetailsHeader-badge $$GODISCOVERY_LATESTMINORCLASS$$" |
| data-version="v1.2.3" data-mpath="p1/p2" data-ppath="" data-pagetype="pkg"> |
| <span>Latest</span> |
| <a href="mod/p1/p2@$$GODISCOVERY_LATESTMINORVERSION$$">Go to latest</a> |
| </div>`, |
| want: ` |
| <div class="DetailsHeader-badge DetailsHeader-badge--latest" |
| data-version="v1.2.3" data-mpath="p1/p2" data-ppath="" data-pagetype="pkg"> |
| <span>Latest</span> |
| <a href="mod/p1/p2@v1.2.3">Go to latest</a> |
| </div>`, |
| }, |
| { |
| name: "package not in module latest", |
| latest: internal.LatestInfo{MinorVersion: "v1.2.3", UnitExistsAtMinor: false}, |
| in: ` |
| <div class="DetailsHeader-badge $$GODISCOVERY_LATESTMINORCLASS$$" |
| data-version="v1.2.3" data-mpath="p1/p2" data-ppath="" data-pagetype="pkg"> |
| <span>Latest</span> |
| <a href="mod/p1/p2@$$GODISCOVERY_LATESTMINORVERSION$$">Go to latest</a> |
| </div>`, |
| want: ` |
| <div class="DetailsHeader-badge DetailsHeader-badge--notAtLatest" |
| data-version="v1.2.3" data-mpath="p1/p2" data-ppath="" data-pagetype="pkg"> |
| <span>Latest</span> |
| <a href="mod/p1/p2@v1.2.3">Go to latest</a> |
| </div>`, |
| }, |
| { |
| name: "latest func returns empty string", |
| latest: internal.LatestInfo{}, |
| in: ` |
| <div class="DetailsHeader-badge $$GODISCOVERY_LATESTMINORCLASS$$" |
| data-version="v1.2.3" data-mpath="p1/p2" data-ppath="" data-pagetype="pkg"> |
| <span>Latest</span> |
| <a href="mod/p1/p2@$$GODISCOVERY_LATESTMINORVERSION$$">Go to latest</a> |
| </div>`, |
| want: ` |
| <div class="DetailsHeader-badge DetailsHeader-badge--unknown" |
| data-version="v1.2.3" data-mpath="p1/p2" data-ppath="" data-pagetype="pkg"> |
| <span>Latest</span> |
| <a href="mod/p1/p2@">Go to latest</a> |
| </div>`, |
| }, |
| { |
| name: "no regexp match", |
| latest: internal.LatestInfo{MinorVersion: "v1.2.3"}, |
| in: ` |
| <div class="DetailsHeader-badge $$GODISCOVERY_LATESTMINORCLASS$$"> |
| <span>Latest</span> |
| <a href="mod/p1/p2@$$GODISCOVERY_LATESTMINORVERSION$$">Go to latest</a> |
| </div>`, |
| want: ` |
| <div class="DetailsHeader-badge $$GODISCOVERY_LATESTMINORCLASS$$"> |
| <span>Latest</span> |
| <a href="mod/p1/p2@$$GODISCOVERY_LATESTMINORVERSION$$">Go to latest</a> |
| </div>`, |
| }, |
| } { |
| t.Run(test.name, func(t *testing.T) { |
| handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { |
| fmt.Fprint(w, test.in) |
| }) |
| lfunc := func(context.Context, string, string) internal.LatestInfo { return test.latest } |
| lv := LatestVersions(lfunc)(handler) |
| addExp := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { |
| lv.ServeHTTP(w, r.WithContext(experiment.NewContext(r.Context(), internal.ExperimentNotAtLatest))) |
| }) |
| ts := httptest.NewServer(addExp) |
| defer ts.Close() |
| resp, err := ts.Client().Get(ts.URL) |
| if err != nil { |
| t.Fatal(err) |
| } |
| got, err := ioutil.ReadAll(resp.Body) |
| if err != nil { |
| t.Fatal(err) |
| } |
| _ = resp.Body.Close() |
| if string(got) != test.want { |
| t.Errorf("\ngot %s\nwant %s", got, test.want) |
| } |
| }) |
| } |
| } |
| |
| func TestLatestMajorVersion(t *testing.T) { |
| for _, test := range []struct { |
| name string |
| latest internal.LatestInfo |
| modulePaths []string |
| in string |
| want string |
| }{ |
| { |
| name: "module path is not at latest", |
| latest: internal.LatestInfo{MajorModulePath: "foo.com/bar/v3", MajorUnitPath: "foo.com/bar/v3"}, |
| modulePaths: []string{ |
| "foo.com/bar", |
| "foo.com/bar/v2", |
| "foo.com/bar/v3", |
| }, |
| in: ` |
| <div class="DetailsHeader-banner$$GODISCOVERY_LATESTMAJORCLASS$$"> |
| data-version="v1.0.0" data-mpath="foo.com/bar" data-ppath="foo.com/bar/far" data-pagetype="pkg"> |
| <p> |
| The highest tagged major version is <a href="/$$GODISCOVERY_LATESTMAJORVERSIONURL$$">$$GODISCOVERY_LATESTMAJORVERSION$$</a>. |
| </p> |
| </div>`, |
| want: ` |
| <div class="DetailsHeader-banner"> |
| data-version="v1.0.0" data-mpath="foo.com/bar" data-ppath="foo.com/bar/far" data-pagetype="pkg"> |
| <p> |
| The highest tagged major version is <a href="/foo.com/bar/v3">v3</a>. |
| </p> |
| </div>`, |
| }, |
| { |
| name: "module path is at latest", |
| latest: internal.LatestInfo{MajorModulePath: "foo.com/bar/v3", MajorUnitPath: "foo.com/bar/v3"}, |
| modulePaths: []string{ |
| "foo.com/bar", |
| "foo.com/bar/v2", |
| "foo.com/bar/v3", |
| }, |
| in: ` |
| <div class="DetailsHeader-banner$$GODISCOVERY_LATESTMAJORCLASS$$"> |
| data-version="v3.0.0" data-mpath="foo.com/bar/v3" data-ppath="foo.com/bar/far" data-pagetype="pkg"> |
| <p> |
| The highest tagged major version is <a href="/$$GODISCOVERY_LATESTMAJORVERSIONURL$$">$$GODISCOVERY_LATESTMAJORVERSION$$</a>. |
| </p> |
| </div>`, |
| want: ` |
| <div class="DetailsHeader-banner DetailsHeader-banner--latest"> |
| data-version="v3.0.0" data-mpath="foo.com/bar/v3" data-ppath="foo.com/bar/far" data-pagetype="pkg"> |
| <p> |
| The highest tagged major version is <a href="/foo.com/bar/v3">v3</a>. |
| </p> |
| </div>`, |
| }, |
| { |
| name: "full path is not at the latest", |
| latest: internal.LatestInfo{MajorModulePath: "foo.com/bar/v3", MajorUnitPath: "foo.com/bar/v3/far"}, |
| modulePaths: []string{ |
| "foo.com/bar", |
| "foo.com/bar/v2", |
| "foo.com/bar/v3", |
| }, |
| in: ` |
| <div class="DetailsHeader-banner$$GODISCOVERY_LATESTMAJORCLASS$$"> |
| data-version="v1.0.0" data-mpath="foo.com/bar" data-ppath="foo.com/bar/far" data-pagetype="pkg"> |
| <p> |
| The highest tagged major version is <a href="/$$GODISCOVERY_LATESTMAJORVERSIONURL$$">$$GODISCOVERY_LATESTMAJORVERSION$$</a>. |
| </p> |
| </div>`, |
| want: ` |
| <div class="DetailsHeader-banner"> |
| data-version="v1.0.0" data-mpath="foo.com/bar" data-ppath="foo.com/bar/far" data-pagetype="pkg"> |
| <p> |
| The highest tagged major version is <a href="/foo.com/bar/v3/far">v3</a>. |
| </p> |
| </div>`, |
| }, |
| } { |
| t.Run(test.name, func(t *testing.T) { |
| handler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { |
| fmt.Fprint(w, test.in) |
| }) |
| lfunc := func(context.Context, string, string) internal.LatestInfo { return test.latest } |
| ts := httptest.NewServer(LatestVersions(lfunc)(handler)) |
| defer ts.Close() |
| resp, err := ts.Client().Get(ts.URL) |
| if err != nil { |
| t.Fatal(err) |
| } |
| got, err := ioutil.ReadAll(resp.Body) |
| if err != nil { |
| t.Fatal(err) |
| } |
| _ = resp.Body.Close() |
| if string(got) != test.want { |
| t.Errorf("\ngot %s\nwant %s", got, test.want) |
| } |
| }) |
| } |
| } |
