go.fuchsia.dev/infra@v0.0.0-20240507153436-9b593402251b/cmd/recipe_wrapper/git/git_test.go

// Copyright 2024 The Fuchsia 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 git

import (
	"testing"

	buildbucketpb "go.chromium.org/luci/buildbucket/proto"
	"google.golang.org/protobuf/types/known/structpb"
)

func TestChooseRef(t *testing.T) {
	tests := []struct {
		name  string
		ref   string
		props map[string]any
		want  string
	}{
		{
			name: "no ref set",
			ref:  "",
			want: FallbackRef,
		},
		{
			name: "ref from commit",
			ref:  "refs/heads/foo",
			want: "refs/heads/foo",
		},
		{
			name: "ref overridden",
			ref:  "refs/heads/overridden",
			props: map[string]any{
				"recipes_integration_ref_override": "refs/heads/foo",
			},
			want: "refs/heads/foo",
		},
		{
			name: "mapped ref",
			ref:  "refs/heads/original",
			props: map[string]any{
				"recipes_integration_ref_mapping": map[string]any{
					"refs/heads/original": "refs/heads/mapped",
				},
			},
			want: "refs/heads/mapped",
		},
	}
	for _, tc := range tests {
		t.Run(tc.name, func(t *testing.T) {
			props, err := structpb.NewStruct(tc.props)
			if err != nil {
				t.Fatal(err)
			}
			build := &buildbucketpb.Build{
				Input: &buildbucketpb.Build_Input{
					Properties:    props,
					GitilesCommit: &buildbucketpb.GitilesCommit{Ref: tc.ref},
				},
			}
			got, err := chooseRef(build)
			if err != nil {
				t.Fatal(err)
			}
			if tc.want != got {
				t.Errorf("Expected ref %q, got %q", tc.want, got)
			}
		})
	}
}