// Copyright 2023 The Gitea Authors. All rights reserved.// SPDX-License-Identifier: MIT package unit import (	"testing" 	"forgejo.org/modules/setting" 	"github.com/stretchr/testify/assert"	"github.com/stretchr/testify/require") func TestLoadUnitConfig(t *testing.T) {	t.Run("regular", func(t *testing.T) {		defer func(disabledRepoUnits, defaultRepoUnits, defaultForkRepoUnits []Type) {			DisabledRepoUnitsSet(disabledRepoUnits)			DefaultRepoUnits = defaultRepoUnits			DefaultForkRepoUnits = defaultForkRepoUnits		}(DisabledRepoUnitsGet(), DefaultRepoUnits, DefaultForkRepoUnits)		defer func(disabledRepoUnits, defaultRepoUnits, defaultForkRepoUnits []string) {			setting.Repository.DisabledRepoUnits = disabledRepoUnits			setting.Repository.DefaultRepoUnits = defaultRepoUnits			setting.Repository.DefaultForkRepoUnits = defaultForkRepoUnits		}(setting.Repository.DisabledRepoUnits, setting.Repository.DefaultRepoUnits, setting.Repository.DefaultForkRepoUnits) 		setting.Repository.DisabledRepoUnits = []string{"repo.issues"}		setting.Repository.DefaultRepoUnits = []string{"repo.code", "repo.releases", "repo.issues", "repo.pulls"}		setting.Repository.DefaultForkRepoUnits = []string{"repo.releases"}		require.NoError(t, LoadUnitConfig())		assert.Equal(t, []Type{TypeIssues}, DisabledRepoUnitsGet())		assert.Equal(t, []Type{TypeCode, TypeReleases, TypePullRequests}, DefaultRepoUnits)		assert.Equal(t, []Type{TypeReleases}, DefaultForkRepoUnits)	})	t.Run("invalid", func(t *testing.T) {		defer func(disabledRepoUnits, defaultRepoUnits, defaultForkRepoUnits []Type) {			DisabledRepoUnitsSet(disabledRepoUnits)			DefaultRepoUnits = defaultRepoUnits			DefaultForkRepoUnits = defaultForkRepoUnits		}(DisabledRepoUnitsGet(), DefaultRepoUnits, DefaultForkRepoUnits)		defer func(disabledRepoUnits, defaultRepoUnits, defaultForkRepoUnits []string) {			setting.Repository.DisabledRepoUnits = disabledRepoUnits			setting.Repository.DefaultRepoUnits = defaultRepoUnits			setting.Repository.DefaultForkRepoUnits = defaultForkRepoUnits		}(setting.Repository.DisabledRepoUnits, setting.Repository.DefaultRepoUnits, setting.Repository.DefaultForkRepoUnits) 		setting.Repository.DisabledRepoUnits = []string{"repo.issues", "invalid.1"}		setting.Repository.DefaultRepoUnits = []string{"repo.code", "invalid.2", "repo.releases", "repo.issues", "repo.pulls"}		setting.Repository.DefaultForkRepoUnits = []string{"invalid.3", "repo.releases"}		require.NoError(t, LoadUnitConfig())		assert.Equal(t, []Type{TypeIssues}, DisabledRepoUnitsGet())		assert.Equal(t, []Type{TypeCode, TypeReleases, TypePullRequests}, DefaultRepoUnits)		assert.Equal(t, []Type{TypeReleases}, DefaultForkRepoUnits)	})	t.Run("duplicate", func(t *testing.T) {		defer func(disabledRepoUnits, defaultRepoUnits, defaultForkRepoUnits []Type) {			DisabledRepoUnitsSet(disabledRepoUnits)			DefaultRepoUnits = defaultRepoUnits			DefaultForkRepoUnits = defaultForkRepoUnits		}(DisabledRepoUnitsGet(), DefaultRepoUnits, DefaultForkRepoUnits)		defer func(disabledRepoUnits, defaultRepoUnits, defaultForkRepoUnits []string) {			setting.Repository.DisabledRepoUnits = disabledRepoUnits			setting.Repository.DefaultRepoUnits = defaultRepoUnits			setting.Repository.DefaultForkRepoUnits = defaultForkRepoUnits		}(setting.Repository.DisabledRepoUnits, setting.Repository.DefaultRepoUnits, setting.Repository.DefaultForkRepoUnits) 		setting.Repository.DisabledRepoUnits = []string{"repo.issues", "repo.issues"}		setting.Repository.DefaultRepoUnits = []string{"repo.code", "repo.releases", "repo.issues", "repo.pulls", "repo.code"}		setting.Repository.DefaultForkRepoUnits = []string{"repo.releases", "repo.releases"}		require.NoError(t, LoadUnitConfig())		assert.Equal(t, []Type{TypeIssues}, DisabledRepoUnitsGet())		assert.Equal(t, []Type{TypeCode, TypeReleases, TypePullRequests}, DefaultRepoUnits)		assert.Equal(t, []Type{TypeReleases}, DefaultForkRepoUnits)	})	t.Run("empty_default", func(t *testing.T) {		defer func(disabledRepoUnits, defaultRepoUnits, defaultForkRepoUnits []Type) {			DisabledRepoUnitsSet(disabledRepoUnits)			DefaultRepoUnits = defaultRepoUnits			DefaultForkRepoUnits = defaultForkRepoUnits		}(DisabledRepoUnitsGet(), DefaultRepoUnits, DefaultForkRepoUnits)		defer func(disabledRepoUnits, defaultRepoUnits, defaultForkRepoUnits []string) {			setting.Repository.DisabledRepoUnits = disabledRepoUnits			setting.Repository.DefaultRepoUnits = defaultRepoUnits			setting.Repository.DefaultForkRepoUnits = defaultForkRepoUnits		}(setting.Repository.DisabledRepoUnits, setting.Repository.DefaultRepoUnits, setting.Repository.DefaultForkRepoUnits) 		setting.Repository.DisabledRepoUnits = []string{"repo.issues", "repo.issues"}		setting.Repository.DefaultRepoUnits = []string{}		setting.Repository.DefaultForkRepoUnits = []string{"repo.releases", "repo.releases"}		require.NoError(t, LoadUnitConfig())		assert.Equal(t, []Type{TypeIssues}, DisabledRepoUnitsGet())		assert.ElementsMatch(t, []Type{TypeCode, TypePullRequests, TypeReleases, TypeWiki, TypePackages, TypeProjects, TypeActions}, DefaultRepoUnits)		assert.Equal(t, []Type{TypeReleases}, DefaultForkRepoUnits)	})}