gotosocial/testrig/config.go

/*
   GoToSocial
   Copyright (C) 2021 GoToSocial Authors admin@gotosocial.org

   This program is free software: you can redistribute it and/or modify
   it under the terms of the GNU Affero General Public License as published by
   the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU Affero General Public License for more details.

   You should have received a copy of the GNU Affero General Public License
   along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

package testrig

import (
	"reflect"

	"github.com/coreos/go-oidc/v3/oidc"
	"github.com/spf13/viper"
	"github.com/superseriousbusiness/gotosocial/internal/config"
)

// InitTestConfig resets + initializes the viper configuration with test defaults.
func InitTestConfig() {
	// reset viper to an empty state
	viper.Reset()

	// get the field names of config.Keys
	keyFields := reflect.VisibleFields(reflect.TypeOf(config.Keys))

	// get the field values of config.Keys
	keyValues := reflect.ValueOf(config.Keys)

	// get the field values of TestDefaults
	testDefaults := reflect.ValueOf(TestDefaults)

	// for each config field...
	for _, field := range keyFields {
		// the viper config key should be the value of the key
		key, ok := keyValues.FieldByName(field.Name).Interface().(string)
		if !ok {
			panic("could not convert config.Keys value to string")
		}

		// the value should be the test default corresponding to the given fieldName
		value := testDefaults.FieldByName(field.Name).Interface()

		// actually set the value in viper -- this will override everything
		viper.Set(key, value)
	}
}

// TestDefaults returns a Values struct with values set that are suitable for local testing.
var TestDefaults = config.Values{
	LogLevel:        "trace",
	ApplicationName: "gotosocial",
	ConfigPath:      "",
	Host:            "localhost:8080",
	AccountDomain:   "localhost:8080",
	Protocol:        "http",
	BindAddress:     "127.0.0.1",
	Port:            8080,
	TrustedProxies:  []string{"127.0.0.1/32"},

	DbType:     "sqlite",
	DbAddress:  ":memory:",
	DbPort:     5432,
	DbUser:     "postgres",
	DbPassword: "postgres",
	DbDatabase: "postgres",

	WebTemplateBaseDir: "./web/template/",
	WebAssetBaseDir:    "./web/assets/",

	AccountsRegistrationOpen: true,
	AccountsApprovalRequired: true,
	AccountsReasonRequired:   true,

	MediaImageMaxSize:        1048576, // 1mb
	MediaVideoMaxSize:        5242880, // 5mb
	MediaDescriptionMinChars: 0,
	MediaDescriptionMaxChars: 500,

	StorageBackend:       "local",
	StorageBasePath:      "/gotosocial/storage",
	StorageServeProtocol: "http",
	StorageServeHost:     "localhost:8080",
	StorageServeBasePath: "/fileserver",

	StatusesMaxChars:           5000,
	StatusesCWMaxChars:         100,
	StatusesPollMaxOptions:     6,
	StatusesPollOptionMaxChars: 50,
	StatusesMediaMaxFiles:      6,

	LetsEncryptEnabled:      false,
	LetsEncryptPort:         0,
	LetsEncryptCertDir:      "",
	LetsEncryptEmailAddress: "",

	OIDCEnabled:          false,
	OIDCIdpName:          "",
	OIDCSkipVerification: false,
	OIDCIssuer:           "",
	OIDCClientID:         "",
	OIDCClientSecret:     "",
	OIDCScopes:           []string{oidc.ScopeOpenID, "profile", "email", "groups"},

	SMTPHost:     "",
	SMTPPort:     0,
	SMTPUsername: "",
	SMTPPassword: "",
	SMTPFrom:     "GoToSocial",

	SyslogEnabled:  false,
	SyslogProtocol: "udp",
	SyslogAddress:  "localhost:514",
}