| // Copyright 2020 Google Inc. All Rights Reserved. |
| // |
| // Licensed under the Apache License, Version 2.0 (the "License"); |
| // you may not use this file except in compliance with the License. |
| // You may obtain a copy of the License at |
| // |
| // http://www.apache.org/licenses/LICENSE-2.0 |
| // |
| // Unless required by applicable law or agreed to in writing, software |
| // distributed under the License is distributed on an "AS IS" BASIS, |
| // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| // See the License for the specific language governing permissions and |
| // limitations under the License. |
| |
| package utils |
| |
| import ( |
| "errors" |
| "fmt" |
| "strings" |
| "testing" |
| |
| "google.golang.org/grpc/codes" |
| "google.golang.org/grpc/status" |
| ) |
| |
| const ( |
| testInstanceURL = "cos-review.googlesource.com" |
| ) |
| |
| func testCLLink(clID, instanceURL string) string { |
| return fmt.Sprintf("<a href=\"%s/c/%s\" target=\"_blank\">CL %s</a>", instanceURL, clID, clID) |
| } |
| |
| func TestGerritErrCode(t *testing.T) { |
| tests := map[string]struct { |
| inputErr error |
| expectedCode string |
| }{ |
| "Empty Error": { |
| inputErr: errors.New(""), |
| expectedCode: "500", |
| }, |
| "Mapped Error Code": { |
| inputErr: errors.New("failed to fetch \"https://cos-internal-review.googlesource.com/a/changes/?n=1&o=CURRENT_REVISION&q=1\", status code 403"), |
| expectedCode: "403", |
| }, |
| "Irregular Code": { |
| inputErr: errors.New("failed to fetch \"https://cos-internal-review.googlesource.com/a/changes/?n=1&o=CURRENT_REVISION&q=1\", status code 689"), |
| expectedCode: "689", |
| }, |
| } |
| for name, test := range tests { |
| t.Run(name, func(t *testing.T) { |
| code := GerritErrCode(test.inputErr) |
| if code != test.expectedCode { |
| t.Errorf("expected HTTP code %s, got %s", test.expectedCode, code) |
| } |
| }) |
| } |
| } |
| |
| func TestGitilesErrCode(t *testing.T) { |
| tests := map[string]struct { |
| inputErr error |
| expectedCode string |
| }{ |
| "Default Error Code": { |
| inputErr: errors.New("code = e desc = not a desc"), |
| expectedCode: "500", |
| }, |
| "Mapped Error Code": { |
| inputErr: status.New(codes.NotFound, "not found").Err(), |
| expectedCode: "404", |
| }, |
| "403 Code Edge Case": { |
| inputErr: status.New(codes.Internal, "unexpected HTTP 403 from Gitiles").Err(), |
| expectedCode: "403", |
| }, |
| } |
| for name, test := range tests { |
| t.Run(name, func(t *testing.T) { |
| code := GitilesErrCode(test.inputErr) |
| if code != test.expectedCode { |
| t.Errorf("expected HTTP code %s, got %s", test.expectedCode, code) |
| } |
| }) |
| } |
| } |
| |
| func TestBothBuildsNotFound(t *testing.T) { |
| source := "cos-stable-81-12871-103-0" |
| target := "cos-stable-81-12871-117-0" |
| croslandURL := "https://www.google.com" |
| expectedCode := "404" |
| expectedErrHeader := "Build Not Found" |
| expectedErrStr := strings.Join([]string{ |
| "The builds associated with input", |
| source, |
| "and", |
| target, |
| "cannot be found. It may be possible that the inputs are either invalid or both belong", |
| "to pre-Cusky builds. If both of the inputs belong to pre-Cusky builds, note that this tool only supports changelogs", |
| "between Cusky builds. Otherwise, please input valid build numbers (example: 13310.1035.0) or valid image names", |
| "(example: cos-rc-85-13310-1034-0).", |
| }, " ") |
| expectedHTMLErrStr := fmt.Sprintf("%s %s and %s %s<br><br>%s %s <a href=%s target=\"_blank\">%s</a>. %s %s", |
| "The builds associated with input", |
| source, |
| target, |
| "could not be found.", |
| "It may be possible that the inputs are either invalid or both belong to pre-Cusky builds.", |
| "If both of the inputs belong to pre-Cusky builds, please check", |
| croslandLink(croslandURL, source, target), |
| croslandLink(croslandURL, source, target), |
| "Otherwise, please input valid build numbers", |
| "(example: 13310.1035.0) or valid image names (example: cos-rc-85-13310-1034-0).", |
| ) |
| err := BothBuildsNotFound(croslandURL, source, target, source, target) |
| if err.HTTPCode() != expectedCode { |
| t.Errorf("expected HTTP code %s, got %s", expectedCode, err.HTTPCode()) |
| } else if err.Header() != expectedErrHeader { |
| t.Errorf("expected error header \"%s\", got %s", expectedErrHeader, err.Header()) |
| } else if err.Error() != expectedErrStr { |
| t.Errorf("expected error string %s, got %s", expectedErrStr, err.Error()) |
| } else if err.HTMLError() != expectedHTMLErrStr { |
| t.Errorf("expected html error string %s, got %s", expectedErrStr, err.HTMLError()) |
| } else if err.Retryable() { |
| t.Errorf("expected retryable = false, got true") |
| } |
| } |
| |
| func TestBuildNotFound(t *testing.T) { |
| buildNumber := "cos-stable-81-12871-117-0" |
| expectedCode := "404" |
| expectedErrHeader := "Build Not Found" |
| expectedErrStr := strings.Join([]string{ |
| "The build associated with input", |
| buildNumber, |
| "cannot be found. It may be possible that the input is either invalid or belongs to a", |
| "pre-Cusky build. If you entered a pre-Cusky build number or image name, note that changelog between", |
| "pre-Cusky and Cusky builds are not supported. Otherwise, please input a valid build number", |
| "(example: 13310.1035.0) or a valid image name (example: cos-rc-85-13310-1034-0).", |
| }, " ") |
| expectedHTMLErrStr := fmt.Sprintf("%s %s %s<br><br>%s %s %s %s", |
| "The build associated with input", |
| buildNumber, |
| "cannot be found.", |
| "It may be possible that either the input is either invalid or belongs to a", |
| "pre-Cusky build. If you entered a pre-Cusky build number or image name, note that changelog between", |
| "pre-Cusky and Cusky builds are not supported. Otherwise, please input a valid build number", |
| "(example: 13310.1035.0) or a valid image name (example: cos-rc-85-13310-1034-0).", |
| ) |
| err := BuildNotFound(buildNumber) |
| if err.HTTPCode() != expectedCode { |
| t.Errorf("expected HTTP code %s, got %s", expectedCode, err.HTTPCode()) |
| } else if err.Header() != expectedErrHeader { |
| t.Errorf("expected error header \"%s\", got %s", expectedErrHeader, err.Header()) |
| } else if err.Error() != expectedErrStr { |
| t.Errorf("expected error string %s, got %s", expectedErrStr, err.Error()) |
| } else if err.HTMLError() != expectedHTMLErrStr { |
| t.Errorf("expected html error string %s, got %s", expectedErrStr, err.HTMLError()) |
| } else if err.Retryable() { |
| t.Errorf("expected retryable = false, got true") |
| } |
| } |
| |
| func TestCLNotFound(t *testing.T) { |
| clID := "1540" |
| expectedCode := "404" |
| expectedErrHeader := "CL Not Found" |
| expectedErrStr := fmt.Sprintf("No CL was found matching the identifier: %s. Please enter either the CL-number (example: 3206) or a Commit-SHA (example: I7e549d7753cc7acec2b44bb5a305347a97719ab9) of a submitted CL.", clID) |
| err := CLNotFound(clID) |
| if err.HTTPCode() != expectedCode { |
| t.Errorf("expected HTTP code %s, got %s", expectedCode, err.HTTPCode()) |
| } else if err.Header() != expectedErrHeader { |
| t.Errorf("expected error header \"%s\", got %s", expectedErrHeader, err.Header()) |
| } else if err.Error() != expectedErrStr { |
| t.Errorf("expected error string %s, got %s", expectedErrStr, err.Error()) |
| } else if err.HTMLError() != expectedErrStr { |
| t.Errorf("expected html error string %s, got %s", expectedErrStr, err.HTMLError()) |
| } else if err.Retryable() { |
| t.Errorf("expected retryable = false, got true") |
| } |
| } |
| |
| func TestCLLandingNotFound(t *testing.T) { |
| clID := "1540" |
| expectedCode := "406" |
| expectedErrHeader := "No Build Found" |
| expectedErrStr := fmt.Sprintf("No build was found containing CL %s.", clID) |
| link := testCLLink(clID, testInstanceURL) |
| expectedHTMLErrStr := fmt.Sprintf("No build was found containing %s.", link) |
| err := CLLandingNotFound(clID, testInstanceURL) |
| if err.HTTPCode() != expectedCode { |
| t.Errorf("expected HTTP code %s, got %s", expectedCode, err.HTTPCode()) |
| } else if err.Header() != expectedErrHeader { |
| t.Errorf("expected error header \"%s\", got %s", expectedErrHeader, err.Header()) |
| } else if err.Error() != expectedErrStr { |
| t.Errorf("expected error string %s, got %s", expectedErrStr, err.Error()) |
| } else if err.HTMLError() != expectedHTMLErrStr { |
| t.Errorf("expected html error string %s, got %s", expectedHTMLErrStr, err.HTMLError()) |
| } else if !err.Retryable() { |
| t.Errorf("expected retryable = true, got false") |
| } |
| } |
| |
| func TestCLNotUsed(t *testing.T) { |
| clID := "1540" |
| repo := "cos/tools" |
| branch := "master" |
| expectedCode := "406" |
| expectedErrHeader := "CL Not Used" |
| expectedErrStr := fmt.Sprintf("CL %s modifies the %s repository on the %s branch, which has not been used in COS builds since the CL's submission.", clID, repo, branch) |
| link := testCLLink(clID, testInstanceURL) |
| expectedHTMLErrStr := fmt.Sprintf("%s modifies the %s repository on the %s branch, which has not been used in COS builds since the CL's submission.", link, repo, branch) |
| err := CLNotUsed(clID, repo, branch, testInstanceURL) |
| if err.HTTPCode() != expectedCode { |
| t.Errorf("expected HTTP code %s, got %s", expectedCode, err.HTTPCode()) |
| } else if err.Header() != expectedErrHeader { |
| t.Errorf("expected error header \"%s\", got %s", expectedErrHeader, err.Header()) |
| } else if err.Error() != expectedErrStr { |
| t.Errorf("expected error string %s, got %s", expectedErrStr, err.Error()) |
| } else if err.HTMLError() != expectedHTMLErrStr { |
| t.Errorf("expected html error string %s, got %s", expectedHTMLErrStr, err.HTMLError()) |
| } else if err.Retryable() { |
| t.Errorf("expected retryable = false, got true") |
| } |
| } |
| |
| func TestCLTooRecent(t *testing.T) { |
| clID := "1540" |
| expectedCode := "406" |
| expectedErrHeader := "CL Too Recent" |
| expectedErrStr := fmt.Sprintf("CL %s was submitted too recently to be included in any builds. Please wait a couple hours and try again.", clID) |
| link := testCLLink(clID, testInstanceURL) |
| expectedHTMLErrStr := fmt.Sprintf("%s was submitted too recently to be included in any builds. Please wait a couple hours and try again.", link) |
| err := CLTooRecent(clID, testInstanceURL) |
| if err.HTTPCode() != expectedCode { |
| t.Errorf("expected HTTP code %s, got %s", expectedCode, err.HTTPCode()) |
| } else if err.Header() != expectedErrHeader { |
| t.Errorf("expected error header \"%s\", got %s", expectedErrHeader, err.Header()) |
| } else if err.Error() != expectedErrStr { |
| t.Errorf("expected error string %s, got %s", expectedErrStr, err.Error()) |
| } else if err.HTMLError() != expectedHTMLErrStr { |
| t.Errorf("expected html error string %s, got %s", expectedHTMLErrStr, err.HTMLError()) |
| } else if err.Retryable() { |
| t.Errorf("expected retryable = false, got true") |
| } |
| } |
| |
| func TestCLNotSubmitted(t *testing.T) { |
| clID := "1540" |
| expectedCode := "406" |
| expectedErrHeader := "CL Not Submitted" |
| expectedErrStr := fmt.Sprintf("CL %s has not been submitted yet. A CL will not enter any build until it is successfully submitted.", clID) |
| link := testCLLink(clID, testInstanceURL) |
| expectedHTMLErrStr := fmt.Sprintf("%s has not been submitted yet. A CL will not enter any build until it is successfully submitted.", link) |
| err := CLNotSubmitted(clID, testInstanceURL) |
| if err.HTTPCode() != expectedCode { |
| t.Errorf("expected HTTP code %s, got %s", expectedCode, err.HTTPCode()) |
| } else if err.Header() != expectedErrHeader { |
| t.Errorf("expected error header \"%s\", got %s", expectedErrHeader, err.Header()) |
| } else if err.Error() != expectedErrStr { |
| t.Errorf("expected error string %s, got %s", expectedErrStr, err.Error()) |
| } else if err.HTMLError() != expectedHTMLErrStr { |
| t.Errorf("expected html error string %s, got %s", expectedHTMLErrStr, err.HTMLError()) |
| } |
| } |
| |
| func TestCLInvalidRelease(t *testing.T) { |
| clID := "1540" |
| release := "master" |
| expectedCode := "406" |
| expectedErrHeader := "Invalid Release Branch" |
| expectedErrStr := fmt.Sprintf("CL %s maps to release %s, which is not a valid release", clID, release) |
| link := testCLLink(clID, testInstanceURL) |
| expectedHTMLErrStr := fmt.Sprintf("%s maps to release %s, which is not a valid release", link, release) |
| err := CLInvalidRelease(clID, release, testInstanceURL) |
| if err.HTTPCode() != expectedCode { |
| t.Errorf("expected HTTP code %s, got %s", expectedCode, err.HTTPCode()) |
| } else if err.Header() != expectedErrHeader { |
| t.Errorf("expected error header \"%s\", got %s", expectedErrHeader, err.Header()) |
| } else if err.Error() != expectedErrStr { |
| t.Errorf("expected error string %s, got %s", expectedErrStr, err.Error()) |
| } else if err.HTMLError() != expectedHTMLErrStr { |
| t.Errorf("expected html error string %s, got %s", expectedHTMLErrStr, err.HTMLError()) |
| } else if err.Retryable() { |
| t.Errorf("expected retryable = false, got true") |
| } |
| } |
