go/src/infra/appengine/cr-rev/backend/import_test.go

// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

package main

import (
	"testing"

	"github.com/golang/mock/gomock"

	gitilesProto "go.chromium.org/luci/common/proto/gitiles"
	"go.chromium.org/luci/common/testing/ftt"
	"go.chromium.org/luci/common/testing/truth/assert"
	"go.chromium.org/luci/common/testing/truth/should"
	"go.chromium.org/luci/server/auth/authtest"

	"go.chromium.org/infra/appengine/cr-rev/backend/gitiles"
	"go.chromium.org/infra/appengine/cr-rev/backend/repoimport"
	"go.chromium.org/infra/appengine/cr-rev/common"
	"go.chromium.org/infra/appengine/cr-rev/config"
)

func TestInitialImport(t *testing.T) {
	ctx := authtest.MockAuthConfig(t.Context())
	mockCtrl := gomock.NewController(t)
	defer mockCtrl.Finish()

	ftt.Run("invalid host", t, func(t *ftt.Test) {
		cfg := &config.Config{
			Hosts: []*config.Host{
				{
					Name: "invalid/name",
				},
			},
		}
		assert.ErrIsLike(t, setupImport(ctx, cfg), " hostname invalid/name.googlesource.com is not a valid Gitiles host")
	})
}

func TestInitialHostImport(t *testing.T) {
	mockCtrl := gomock.NewController(t)
	defer mockCtrl.Finish()
	controller := repoimport.NewMockController(mockCtrl)

	ftt.Run("Skip repos", t, func(t *ftt.Test) {
		// Setup gitiles
		fakeGitilesClient := &gitilesProto.Fake{}
		fakeGitilesClient.SetRepository("foo", nil, nil)
		fakeGitilesClient.SetRepository("bar", nil, nil)
		ctx := gitiles.SetClient(t.Context(), fakeGitilesClient)

		// Setup config
		barRepoConfig := &config.Repository{
			Name: "bar",
			Indexing: &config.Repository_DoNotIndex{
				DoNotIndex: true,
			},
		}
		host := &config.Host{
			Name: "host",
			Repos: []*config.Repository{
				barRepoConfig,
			},
		}

		// setup mock
		controller.EXPECT().Index(common.GitRepository{
			Host: "host",
			Name: "foo",
		}).Times(1)

		assert.NoErr(t, initialHostImport(ctx, controller, host))
		// We expect only one Gitiles calls (to list projects):
		assert.Loosely(t, fakeGitilesClient.GetCallLogs(), should.HaveLength(1))
	})
}