go.chromium.org/luci@v0.0.0-20240309015107-7cdc2e660f33/resultdb/internal/testresults/test_result.go

// 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 testresults

import (
	"context"
	"net/url"
	"strings"
	"time"

	"cloud.google.com/go/spanner"
	"google.golang.org/grpc/codes"
	"google.golang.org/protobuf/proto"
	"google.golang.org/protobuf/types/known/durationpb"
	"google.golang.org/protobuf/types/known/structpb"

	"go.chromium.org/luci/common/errors"
	"go.chromium.org/luci/grpc/appstatus"
	"go.chromium.org/luci/resultdb/internal/invocations"
	"go.chromium.org/luci/resultdb/internal/spanutil"
	pb "go.chromium.org/luci/resultdb/proto/v1"
)

// MustParseName retrieves the invocation ID, unescaped test id, and
// result ID.
//
// Panics if the name is invalid. Should be used only with trusted data.
//
// MustParseName is faster than pbutil.ParseTestResultName.
func MustParseName(name string) (invID invocations.ID, testID, resultID string) {
	parts := strings.Split(name, "/")
	if len(parts) != 6 || parts[0] != "invocations" || parts[2] != "tests" || parts[4] != "results" {
		panic(errors.Reason("malformed test result name: %q", name).Err())
	}

	invID = invocations.ID(parts[1])
	testID = parts[3]
	resultID = parts[5]

	unescaped, err := url.PathUnescape(testID)
	if err != nil {
		panic(errors.Annotate(err, "malformed test id %q", testID).Err())
	}
	testID = unescaped

	return
}

// Read reads specified TestResult within the transaction.
// If the TestResult does not exist, the returned error is annotated with
// NotFound GRPC code.
func Read(ctx context.Context, name string) (*pb.TestResult, error) {
	invID, testID, resultID := MustParseName(name)
	tr := &pb.TestResult{
		Name:     name,
		TestId:   testID,
		ResultId: resultID,
		Expected: true,
	}

	var maybeUnexpected spanner.NullBool
	var micros spanner.NullInt64
	var summaryHTML spanutil.Compressed
	var tmd spanutil.Compressed
	var fr spanutil.Compressed
	var properties spanutil.Compressed
	var skipReason spanner.NullInt64
	err := spanutil.ReadRow(ctx, "TestResults", invID.Key(testID, resultID), map[string]any{
		"Variant":         &tr.Variant,
		"VariantHash":     &tr.VariantHash,
		"IsUnexpected":    &maybeUnexpected,
		"Status":          &tr.Status,
		"SummaryHTML":     &summaryHTML,
		"StartTime":       &tr.StartTime,
		"RunDurationUsec": &micros,
		"Tags":            &tr.Tags,
		"TestMetadata":    &tmd,
		"FailureReason":   &fr,
		"Properties":      &properties,
		"SkipReason":      &skipReason,
	})
	switch {
	case spanner.ErrCode(err) == codes.NotFound:
		return nil, appstatus.Attachf(err, codes.NotFound, "%s not found", name)

	case err != nil:
		return nil, errors.Annotate(err, "failed to fetch %q", name).Err()
	}

	tr.SummaryHtml = string(summaryHTML)
	PopulateExpectedField(tr, maybeUnexpected)
	PopulateDurationField(tr, micros)
	PopulateSkipReasonField(tr, skipReason)
	if err := populateTestMetadata(tr, tmd); err != nil {
		return nil, errors.Annotate(err, "failed to unmarshal test metadata").Err()
	}
	if err := populateFailureReason(tr, fr); err != nil {
		return nil, errors.Annotate(err, "failed to unmarshal failure reason").Err()
	}
	if err := populateProperties(tr, properties); err != nil {
		return nil, errors.Annotate(err, "failed to unmarshal properties").Err()
	}
	return tr, nil
}

// PopulateDurationField populates tr.Duration from NullInt64.
func PopulateDurationField(tr *pb.TestResult, micros spanner.NullInt64) {
	tr.Duration = nil
	if micros.Valid {
		tr.Duration = durationpb.New(time.Duration(1000 * micros.Int64))
	}
}

// PopulateSkipReasonField populates tr.SkipReason from NullInt64.
func PopulateSkipReasonField(tr *pb.TestResult, skipReason spanner.NullInt64) {
	tr.SkipReason = pb.SkipReason_SKIP_REASON_UNSPECIFIED
	if skipReason.Valid {
		tr.SkipReason = pb.SkipReason(skipReason.Int64)
	}
}

// PopulateExpectedField populates tr.Expected from NullBool.
func PopulateExpectedField(tr *pb.TestResult, maybeUnexpected spanner.NullBool) {
	tr.Expected = !maybeUnexpected.Valid || !maybeUnexpected.Bool
}

func populateTestMetadata(tr *pb.TestResult, tmd spanutil.Compressed) error {
	if len(tmd) == 0 {
		return nil
	}

	tr.TestMetadata = &pb.TestMetadata{}
	return proto.Unmarshal(tmd, tr.TestMetadata)
}

func populateFailureReason(tr *pb.TestResult, fr spanutil.Compressed) error {
	if len(fr) == 0 {
		return nil
	}

	tr.FailureReason = &pb.FailureReason{}
	return proto.Unmarshal(fr, tr.FailureReason)
}

func populateProperties(tr *pb.TestResult, properties spanutil.Compressed) error {
	if len(properties) == 0 {
		return nil
	}

	tr.Properties = &structpb.Struct{}
	return proto.Unmarshal(properties, tr.Properties)
}