argo-cd/util/github_app/repos_test.go

package github_app

import (
	"context"
	"testing"

	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/mock"
	"github.com/stretchr/testify/require"

	"github.com/argoproj/argo-cd/v3/applicationset/services/github_app_auth"
	"github.com/argoproj/argo-cd/v3/pkg/apis/application/v1alpha1"
	"github.com/argoproj/argo-cd/v3/util/db/mocks"
)

type ArgocdRepositoryMock struct {
	mock *mock.Mock
}

func (a ArgocdRepositoryMock) GetRepoCredsBySecretName(ctx context.Context, secretName string) (*v1alpha1.RepoCreds, error) {
	args := a.mock.Called(ctx, secretName)

	return args.Get(0).(*v1alpha1.RepoCreds), args.Error(1)
}

func Test_repoAsCredentials_GetAuth(t *testing.T) {
	tests := []struct {
		name    string
		repo    v1alpha1.RepoCreds
		want    *github_app_auth.Authentication
		wantErr bool
	}{
		{name: "missing", wantErr: true},
		{name: "found", repo: v1alpha1.RepoCreds{
			GithubAppId:             123,
			GithubAppInstallationId: 456,
			GithubAppPrivateKey:     "private key",
		}, want: &github_app_auth.Authentication{
			Id:                123,
			InstallationId:    456,
			EnterpriseBaseURL: "",
			PrivateKey:        "private key",
		}, wantErr: false},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			m := &mocks.RepoCredsDB{}
			m.EXPECT().GetRepoCredsBySecretName(mock.Anything, mock.Anything).Return(&tt.repo, nil)
			creds := NewAuthCredentials(m)

			auth, err := creds.GetAuthSecret(t.Context(), "https://github.com/foo")
			if tt.wantErr {
				assert.Error(t, err)
				return
			}
			require.NoError(t, err)
			assert.Equal(t, tt.want, auth)
		})
	}
}