blob: 4740e6f7c64f2b69c624f0cabe956122bafbffbe [file] [log] [blame]
// Copyright 2020 The LUCI Authors.
//
// 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 sink
import (
"context"
"io/ioutil"
"net/http"
"os"
"time"
"google.golang.org/grpc"
"google.golang.org/grpc/metadata"
"google.golang.org/protobuf/types/known/durationpb"
"google.golang.org/protobuf/types/known/timestamppb"
"go.chromium.org/luci/common/clock/testclock"
"go.chromium.org/luci/grpc/prpc"
"go.chromium.org/luci/resultdb/pbutil"
pb "go.chromium.org/luci/resultdb/proto/v1"
sinkpb "go.chromium.org/luci/resultdb/sink/proto/v1"
. "github.com/smartystreets/goconvey/convey"
)
func reportTestResults(ctx context.Context, host, authToken string, in *sinkpb.ReportTestResultsRequest) (*sinkpb.ReportTestResultsResponse, error) {
sinkClient := sinkpb.NewSinkPRPCClient(&prpc.Client{
Host: host,
Options: &prpc.Options{Insecure: true},
})
// install the auth token into the context, if present
if authToken != "" {
ctx = metadata.AppendToOutgoingContext(ctx, AuthTokenKey, authTokenValue(authToken))
}
return sinkClient.ReportTestResults(ctx, in)
}
func testServerConfig(addr, tk string) ServerConfig {
return ServerConfig{
Address: addr,
AuthToken: tk,
ArtifactStreamClient: &http.Client{},
ArtifactStreamHost: "example.org",
Recorder: &mockRecorder{},
Invocation: "invocations/u-foo-1587421194_893166206",
invocationID: "u-foo-1587421194_893166206",
UpdateToken: "UpdateToken-ABC",
MaxBatchableArtifactSize: 2 * 1024 * 1024,
}
}
func testArtifactWithFile(writer func(f *os.File)) *sinkpb.Artifact {
f, err := ioutil.TempFile("", "test-artifact")
So(err, ShouldBeNil)
defer f.Close()
writer(f)
return &sinkpb.Artifact{
Body: &sinkpb.Artifact_FilePath{FilePath: f.Name()},
ContentType: "text/plain",
}
}
func testArtifactWithContents(contents []byte) *sinkpb.Artifact {
return &sinkpb.Artifact{
Body: &sinkpb.Artifact_Contents{Contents: contents},
ContentType: "text/plain",
}
}
// validTestResult returns a valid sinkpb.TestResult sample message.
func validTestResult() (*sinkpb.TestResult, func()) {
now := testclock.TestRecentTimeUTC
st := timestamppb.New(now.Add(-2 * time.Minute))
artf := testArtifactWithFile(func(f *os.File) {
_, err := f.WriteString("a sample artifact")
So(err, ShouldBeNil)
})
cleanup := func() { os.Remove(artf.GetFilePath()) }
return &sinkpb.TestResult{
TestId: "this is testID",
ResultId: "result_id1",
Expected: true,
Status: pb.TestStatus_PASS,
SummaryHtml: "HTML summary",
StartTime: st,
Duration: durationpb.New(time.Minute),
Tags: pbutil.StringPairs("k1", "v1"),
Artifacts: map[string]*sinkpb.Artifact{
"art1": artf,
},
TestMetadata: &pb.TestMetadata{
Name: "name",
Location: &pb.TestLocation{
Repo: "https://chromium.googlesource.com/chromium/src",
FileName: "//artifact_dir/a_test.cc",
Line: 54,
},
},
FailureReason: &pb.FailureReason{
PrimaryErrorMessage: "This is a failure message.",
},
}, cleanup
}
type mockRecorder struct {
pb.RecorderClient
batchCreateTestResults func(ctx context.Context, in *pb.BatchCreateTestResultsRequest) (*pb.BatchCreateTestResultsResponse, error)
batchCreateArtifacts func(ctx context.Context, in *pb.BatchCreateArtifactsRequest) (*pb.BatchCreateArtifactsResponse, error)
batchCreateTestExonerations func(ctx context.Context, in *pb.BatchCreateTestExonerationsRequest) (*pb.BatchCreateTestExonerationsResponse, error)
}
func (m *mockRecorder) BatchCreateTestResults(ctx context.Context, in *pb.BatchCreateTestResultsRequest, opts ...grpc.CallOption) (*pb.BatchCreateTestResultsResponse, error) {
if m.batchCreateTestResults != nil {
return m.batchCreateTestResults(ctx, in)
}
return nil, nil
}
func (m *mockRecorder) BatchCreateArtifacts(ctx context.Context, in *pb.BatchCreateArtifactsRequest, opts ...grpc.CallOption) (*pb.BatchCreateArtifactsResponse, error) {
if m.batchCreateArtifacts != nil {
return m.batchCreateArtifacts(ctx, in)
}
return nil, nil
}
func (m *mockRecorder) BatchCreateTestExonerations(ctx context.Context, in *pb.BatchCreateTestExonerationsRequest, opts ...grpc.CallOption) (*pb.BatchCreateTestExonerationsResponse, error) {
if m.batchCreateTestExonerations != nil {
return m.batchCreateTestExonerations(ctx, in)
}
return nil, nil
}