| 
									
										
										
										
											2015-02-12 18:55:55 +08:00
										 |  |  | package setting | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | import ( | 
					
						
							| 
									
										
										
										
											2019-06-20 22:15:21 +08:00
										 |  |  | 	"bufio" | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 	"math/rand" | 
					
						
							| 
									
										
										
										
											2021-02-01 17:13:09 +08:00
										 |  |  | 	"net/url" | 
					
						
							| 
									
										
										
										
											2015-02-12 20:31:41 +08:00
										 |  |  | 	"os" | 
					
						
							| 
									
										
										
										
											2019-04-25 14:29:07 +08:00
										 |  |  | 	"path" | 
					
						
							| 
									
										
										
										
											2015-02-12 18:55:55 +08:00
										 |  |  | 	"path/filepath" | 
					
						
							| 
									
										
										
										
											2017-06-19 21:36:08 +08:00
										 |  |  | 	"runtime" | 
					
						
							| 
									
										
										
										
											2019-06-20 22:15:21 +08:00
										 |  |  | 	"strings" | 
					
						
							| 
									
										
										
										
											2015-02-12 18:55:55 +08:00
										 |  |  | 	"testing" | 
					
						
							| 
									
										
										
										
											2020-09-14 21:57:38 +08:00
										 |  |  | 	"time" | 
					
						
							| 
									
										
										
										
											2015-02-12 18:55:55 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 	"github.com/stretchr/testify/assert" | 
					
						
							| 
									
										
										
										
											2020-03-11 17:04:48 +08:00
										 |  |  | 	"github.com/stretchr/testify/require" | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-12 13:27:47 +08:00
										 |  |  | 	"gopkg.in/ini.v1" | 
					
						
							| 
									
										
										
										
											2015-02-12 18:55:55 +08:00
										 |  |  | ) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2019-06-12 13:27:47 +08:00
										 |  |  | const ( | 
					
						
							|  |  |  | 	windows = "windows" | 
					
						
							|  |  |  | ) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2015-02-12 18:55:55 +08:00
										 |  |  | func TestLoadingSettings(t *testing.T) { | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 	skipStaticRootValidation = true | 
					
						
							| 
									
										
										
										
											2015-02-12 18:55:55 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 	t.Run("Given the default ini files", func(t *testing.T) { | 
					
						
							|  |  |  | 		cfg := NewCfg() | 
					
						
							|  |  |  | 		err := cfg.Load(CommandLineArgs{HomePath: "../../", Config: "../../conf/defaults.ini"}) | 
					
						
							|  |  |  | 		require.Nil(t, err) | 
					
						
							| 
									
										
										
										
											2015-02-12 18:55:55 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 		require.Equal(t, "admin", cfg.AdminUser) | 
					
						
							|  |  |  | 		require.Equal(t, "http://localhost:3000/", cfg.RendererCallbackUrl) | 
					
						
							|  |  |  | 	}) | 
					
						
							| 
									
										
										
										
											2015-02-12 20:31:41 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 	t.Run("default.ini should have no semi-colon commented entries", func(t *testing.T) { | 
					
						
							|  |  |  | 		file, err := os.Open("../../conf/defaults.ini") | 
					
						
							|  |  |  | 		if err != nil { | 
					
						
							|  |  |  | 			t.Errorf("failed to load defaults.ini file: %v", err) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 		defer func() { | 
					
						
							|  |  |  | 			err := file.Close() | 
					
						
							|  |  |  | 			require.Nil(t, err) | 
					
						
							|  |  |  | 		}() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		scanner := bufio.NewScanner(file) | 
					
						
							|  |  |  | 		for scanner.Scan() { | 
					
						
							|  |  |  | 			// This only catches values commented out with ";" and will not catch those that are commented out with "#".
 | 
					
						
							|  |  |  | 			if strings.HasPrefix(scanner.Text(), ";") { | 
					
						
							|  |  |  | 				t.Errorf("entries in defaults.ini must not be commented or environment variables will not work: %v", scanner.Text()) | 
					
						
							| 
									
										
										
										
											2019-06-20 22:15:21 +08:00
										 |  |  | 			} | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 		} | 
					
						
							|  |  |  | 	}) | 
					
						
							| 
									
										
										
										
											2021-07-22 22:50:27 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 	t.Run("sample.ini should load successfully", func(t *testing.T) { | 
					
						
							|  |  |  | 		customInitPath := CustomInitPath | 
					
						
							|  |  |  | 		CustomInitPath = "conf/sample.ini" | 
					
						
							|  |  |  | 		cfg := NewCfg() | 
					
						
							|  |  |  | 		err := cfg.Load(CommandLineArgs{HomePath: "../../"}) | 
					
						
							|  |  |  | 		require.Nil(t, err) | 
					
						
							|  |  |  | 		// Restore CustomInitPath to avoid side effects.
 | 
					
						
							|  |  |  | 		CustomInitPath = customInitPath | 
					
						
							|  |  |  | 	}) | 
					
						
							| 
									
										
										
										
											2018-04-30 22:21:04 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 	t.Run("Should be able to override via environment variables", func(t *testing.T) { | 
					
						
							|  |  |  | 		err := os.Setenv("GF_SECURITY_ADMIN_USER", "superduper") | 
					
						
							|  |  |  | 		require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2015-02-12 20:31:41 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 		cfg := NewCfg() | 
					
						
							|  |  |  | 		err = cfg.Load(CommandLineArgs{HomePath: "../../"}) | 
					
						
							|  |  |  | 		require.Nil(t, err) | 
					
						
							| 
									
										
										
										
											2015-04-09 18:16:59 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 		require.Equal(t, "superduper", cfg.AdminUser) | 
					
						
							|  |  |  | 		require.Equal(t, filepath.Join(HomePath, "data"), cfg.DataPath) | 
					
						
							|  |  |  | 		require.Equal(t, filepath.Join(cfg.DataPath, "log"), cfg.LogsPath) | 
					
						
							|  |  |  | 	}) | 
					
						
							| 
									
										
										
										
											2018-04-30 22:21:04 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 	t.Run("Should replace password when defined in environment", func(t *testing.T) { | 
					
						
							|  |  |  | 		err := os.Setenv("GF_SECURITY_ADMIN_PASSWORD", "supersecret") | 
					
						
							|  |  |  | 		require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2016-06-29 00:37:59 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 		cfg := NewCfg() | 
					
						
							|  |  |  | 		err = cfg.Load(CommandLineArgs{HomePath: "../../"}) | 
					
						
							|  |  |  | 		require.Nil(t, err) | 
					
						
							| 
									
										
										
										
											2016-06-29 00:37:59 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 		require.Contains(t, appliedEnvOverrides, "GF_SECURITY_ADMIN_PASSWORD=*********") | 
					
						
							|  |  |  | 	}) | 
					
						
							| 
									
										
										
										
											2018-04-30 22:21:04 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 	t.Run("Should replace password in URL when url environment is defined", func(t *testing.T) { | 
					
						
							|  |  |  | 		err := os.Setenv("GF_DATABASE_URL", "mysql://user:secret@localhost:3306/database") | 
					
						
							|  |  |  | 		require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2016-06-29 00:37:59 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 		cfg := NewCfg() | 
					
						
							|  |  |  | 		err = cfg.Load(CommandLineArgs{HomePath: "../../"}) | 
					
						
							|  |  |  | 		require.Nil(t, err) | 
					
						
							| 
									
										
										
										
											2016-06-29 00:37:59 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 		require.Contains(t, appliedEnvOverrides, "GF_DATABASE_URL=mysql://user:xxxxx@localhost:3306/database") | 
					
						
							|  |  |  | 	}) | 
					
						
							| 
									
										
										
										
											2015-04-09 18:16:59 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 	t.Run("Should get property map from command line args array", func(t *testing.T) { | 
					
						
							| 
									
										
										
										
											2021-11-09 00:56:56 +08:00
										 |  |  | 		cfg := NewCfg() | 
					
						
							|  |  |  | 		props := cfg.getCommandLineProperties([]string{"cfg:test=value", "cfg:map.test=1"}) | 
					
						
							| 
									
										
										
										
											2015-04-09 18:16:59 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 		require.Equal(t, 2, len(props)) | 
					
						
							|  |  |  | 		require.Equal(t, "value", props["test"]) | 
					
						
							|  |  |  | 		require.Equal(t, "1", props["map.test"]) | 
					
						
							|  |  |  | 	}) | 
					
						
							| 
									
										
										
										
											2015-04-09 18:16:59 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 	t.Run("Should be able to override via command line", func(t *testing.T) { | 
					
						
							|  |  |  | 		if runtime.GOOS == windows { | 
					
						
							| 
									
										
										
										
											2018-04-30 22:21:04 +08:00
										 |  |  | 			cfg := NewCfg() | 
					
						
							| 
									
										
										
										
											2021-08-25 21:11:22 +08:00
										 |  |  | 			err := cfg.Load(CommandLineArgs{ | 
					
						
							| 
									
										
										
										
											2015-04-12 15:15:49 +08:00
										 |  |  | 				HomePath: "../../", | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 				Args:     []string{`cfg:paths.data=c:\tmp\data`, `cfg:paths.logs=c:\tmp\logs`}, | 
					
						
							| 
									
										
										
										
											2015-04-09 18:16:59 +08:00
										 |  |  | 			}) | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			require.Nil(t, err) | 
					
						
							|  |  |  | 			require.Equal(t, `c:\tmp\data`, cfg.DataPath) | 
					
						
							|  |  |  | 			require.Equal(t, `c:\tmp\logs`, cfg.LogsPath) | 
					
						
							|  |  |  | 		} else { | 
					
						
							|  |  |  | 			cfg := NewCfg() | 
					
						
							|  |  |  | 			err := cfg.Load(CommandLineArgs{ | 
					
						
							|  |  |  | 				HomePath: "../../", | 
					
						
							|  |  |  | 				Args:     []string{"cfg:paths.data=/tmp/data", "cfg:paths.logs=/tmp/logs"}, | 
					
						
							|  |  |  | 			}) | 
					
						
							|  |  |  | 			require.Nil(t, err) | 
					
						
							| 
									
										
										
										
											2015-04-09 18:16:59 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			require.Equal(t, "/tmp/data", cfg.DataPath) | 
					
						
							|  |  |  | 			require.Equal(t, "/tmp/logs", cfg.LogsPath) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	}) | 
					
						
							| 
									
										
										
										
											2015-04-09 18:16:59 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 	t.Run("Should be able to override defaults via command line", func(t *testing.T) { | 
					
						
							|  |  |  | 		cfg := NewCfg() | 
					
						
							|  |  |  | 		err := cfg.Load(CommandLineArgs{ | 
					
						
							|  |  |  | 			HomePath: "../../", | 
					
						
							|  |  |  | 			Args: []string{ | 
					
						
							|  |  |  | 				"cfg:default.server.domain=test2", | 
					
						
							|  |  |  | 			}, | 
					
						
							|  |  |  | 			Config: filepath.Join(HomePath, "pkg/setting/testdata/override.ini"), | 
					
						
							| 
									
										
										
										
											2015-04-09 18:16:59 +08:00
										 |  |  | 		}) | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 		require.Nil(t, err) | 
					
						
							| 
									
										
										
										
											2015-04-09 18:16:59 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 		require.Equal(t, "test2", cfg.Domain) | 
					
						
							|  |  |  | 	}) | 
					
						
							| 
									
										
										
										
											2015-02-12 20:31:41 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 	t.Run("Defaults can be overridden in specified config file", func(t *testing.T) { | 
					
						
							|  |  |  | 		if runtime.GOOS == windows { | 
					
						
							| 
									
										
										
										
											2018-04-30 22:21:04 +08:00
										 |  |  | 			cfg := NewCfg() | 
					
						
							| 
									
										
										
										
											2021-08-25 21:11:22 +08:00
										 |  |  | 			err := cfg.Load(CommandLineArgs{ | 
					
						
							| 
									
										
										
										
											2016-06-02 20:32:17 +08:00
										 |  |  | 				HomePath: "../../", | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 				Config:   filepath.Join(HomePath, "pkg/setting/testdata/override_windows.ini"), | 
					
						
							|  |  |  | 				Args:     []string{`cfg:default.paths.data=c:\tmp\data`}, | 
					
						
							| 
									
										
										
										
											2016-06-02 20:32:17 +08:00
										 |  |  | 			}) | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			require.Nil(t, err) | 
					
						
							| 
									
										
										
										
											2016-06-02 20:32:17 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			require.Equal(t, `c:\tmp\override`, cfg.DataPath) | 
					
						
							|  |  |  | 		} else { | 
					
						
							| 
									
										
										
										
											2018-09-04 19:42:55 +08:00
										 |  |  | 			cfg := NewCfg() | 
					
						
							| 
									
										
										
										
											2021-08-25 21:11:22 +08:00
										 |  |  | 			err := cfg.Load(CommandLineArgs{ | 
					
						
							| 
									
										
										
										
											2018-09-04 19:42:55 +08:00
										 |  |  | 				HomePath: "../../", | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 				Config:   filepath.Join(HomePath, "pkg/setting/testdata/override.ini"), | 
					
						
							|  |  |  | 				Args:     []string{"cfg:default.paths.data=/tmp/data"}, | 
					
						
							| 
									
										
										
										
											2018-09-04 19:42:55 +08:00
										 |  |  | 			}) | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			require.Nil(t, err) | 
					
						
							| 
									
										
										
										
											2018-09-04 19:42:55 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			require.Equal(t, "/tmp/override", cfg.DataPath) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	}) | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 	t.Run("Command line overrides specified config file", func(t *testing.T) { | 
					
						
							|  |  |  | 		if runtime.GOOS == windows { | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 			cfg := NewCfg() | 
					
						
							| 
									
										
										
										
											2021-08-25 21:11:22 +08:00
										 |  |  | 			err := cfg.Load(CommandLineArgs{ | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 				HomePath: "../../", | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 				Config:   filepath.Join(HomePath, "pkg/setting/testdata/override_windows.ini"), | 
					
						
							|  |  |  | 				Args:     []string{`cfg:paths.data=c:\tmp\data`}, | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 			}) | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			require.Nil(t, err) | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			require.Equal(t, `c:\tmp\data`, cfg.DataPath) | 
					
						
							|  |  |  | 		} else { | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 			cfg := NewCfg() | 
					
						
							| 
									
										
										
										
											2021-08-25 21:11:22 +08:00
										 |  |  | 			err := cfg.Load(CommandLineArgs{ | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 				HomePath: "../../", | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 				Config:   filepath.Join(HomePath, "pkg/setting/testdata/override.ini"), | 
					
						
							|  |  |  | 				Args:     []string{"cfg:paths.data=/tmp/data"}, | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 			}) | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			require.Nil(t, err) | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			require.Equal(t, "/tmp/data", cfg.DataPath) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	}) | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 	t.Run("Can use environment variables in config values", func(t *testing.T) { | 
					
						
							|  |  |  | 		if runtime.GOOS == windows { | 
					
						
							|  |  |  | 			err := os.Setenv("GF_DATA_PATH", `c:\tmp\env_override`) | 
					
						
							|  |  |  | 			require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 			cfg := NewCfg() | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			err = cfg.Load(CommandLineArgs{ | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 				HomePath: "../../", | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 				Args:     []string{"cfg:paths.data=${GF_DATA_PATH}"}, | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 			}) | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			require.Nil(t, err) | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			require.Equal(t, `c:\tmp\env_override`, cfg.DataPath) | 
					
						
							|  |  |  | 		} else { | 
					
						
							|  |  |  | 			err := os.Setenv("GF_DATA_PATH", "/tmp/env_override") | 
					
						
							|  |  |  | 			require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 			cfg := NewCfg() | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			err = cfg.Load(CommandLineArgs{ | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 				HomePath: "../../", | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 				Args:     []string{"cfg:paths.data=${GF_DATA_PATH}"}, | 
					
						
							| 
									
										
										
										
											2019-11-07 18:24:54 +08:00
										 |  |  | 			}) | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			require.Nil(t, err) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 			require.Equal(t, "/tmp/env_override", cfg.DataPath) | 
					
						
							|  |  |  | 		} | 
					
						
							|  |  |  | 	}) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	t.Run("instance_name default to hostname even if hostname env is empty", func(t *testing.T) { | 
					
						
							|  |  |  | 		cfg := NewCfg() | 
					
						
							|  |  |  | 		err := cfg.Load(CommandLineArgs{ | 
					
						
							|  |  |  | 			HomePath: "../../", | 
					
						
							|  |  |  | 		}) | 
					
						
							|  |  |  | 		require.Nil(t, err) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		hostname, err := os.Hostname() | 
					
						
							|  |  |  | 		require.Nil(t, err) | 
					
						
							|  |  |  | 		require.Equal(t, hostname, InstanceName) | 
					
						
							|  |  |  | 	}) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	t.Run("Reading callback_url should add trailing slash", func(t *testing.T) { | 
					
						
							|  |  |  | 		cfg := NewCfg() | 
					
						
							|  |  |  | 		err := cfg.Load(CommandLineArgs{ | 
					
						
							|  |  |  | 			HomePath: "../../", | 
					
						
							|  |  |  | 			Args:     []string{"cfg:rendering.callback_url=http://myserver/renderer"}, | 
					
						
							|  |  |  | 		}) | 
					
						
							|  |  |  | 		require.Nil(t, err) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		require.Equal(t, "http://myserver/renderer/", cfg.RendererCallbackUrl) | 
					
						
							|  |  |  | 	}) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	t.Run("Only sync_ttl should return the value sync_ttl", func(t *testing.T) { | 
					
						
							|  |  |  | 		cfg := NewCfg() | 
					
						
							|  |  |  | 		err := cfg.Load(CommandLineArgs{ | 
					
						
							|  |  |  | 			HomePath: "../../", | 
					
						
							|  |  |  | 			Args:     []string{"cfg:auth.proxy.sync_ttl=2"}, | 
					
						
							|  |  |  | 		}) | 
					
						
							|  |  |  | 		require.Nil(t, err) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		require.Equal(t, 2, cfg.AuthProxySyncTTL) | 
					
						
							|  |  |  | 	}) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	t.Run("Test reading string values from .ini file", func(t *testing.T) { | 
					
						
							| 
									
										
										
										
											2019-04-25 14:29:07 +08:00
										 |  |  | 		iniFile, err := ini.Load(path.Join(HomePath, "pkg/setting/testdata/invalid.ini")) | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 		require.Nil(t, err) | 
					
						
							| 
									
										
										
										
											2019-04-25 14:29:07 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 		t.Run("If key is found - should return value from ini file", func(t *testing.T) { | 
					
						
							| 
									
										
										
										
											2020-09-08 17:33:04 +08:00
										 |  |  | 			value := valueAsString(iniFile.Section("server"), "alt_url", "") | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			require.Equal(t, "https://grafana.com/", value) | 
					
						
							| 
									
										
										
										
											2019-04-25 14:29:07 +08:00
										 |  |  | 		}) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 		t.Run("If key is not found - should return default value", func(t *testing.T) { | 
					
						
							| 
									
										
										
										
											2020-09-08 17:33:04 +08:00
										 |  |  | 			value := valueAsString(iniFile.Section("server"), "extra_url", "default_url_val") | 
					
						
							| 
									
										
										
										
											2021-10-26 19:19:07 +08:00
										 |  |  | 			require.Equal(t, "default_url_val", value) | 
					
						
							| 
									
										
										
										
											2019-04-25 14:29:07 +08:00
										 |  |  | 		}) | 
					
						
							|  |  |  | 	}) | 
					
						
							| 
									
										
										
										
											2015-02-12 18:55:55 +08:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2020-03-11 17:04:48 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-11-03 02:26:19 +08:00
										 |  |  | func TestParseAppURLAndSubURL(t *testing.T) { | 
					
						
							| 
									
										
										
										
											2020-03-11 17:04:48 +08:00
										 |  |  | 	testCases := []struct { | 
					
						
							|  |  |  | 		rootURL           string | 
					
						
							|  |  |  | 		expectedAppURL    string | 
					
						
							|  |  |  | 		expectedAppSubURL string | 
					
						
							|  |  |  | 	}{ | 
					
						
							|  |  |  | 		{rootURL: "http://localhost:3000/", expectedAppURL: "http://localhost:3000/"}, | 
					
						
							|  |  |  | 		{rootURL: "http://localhost:3000", expectedAppURL: "http://localhost:3000/"}, | 
					
						
							|  |  |  | 		{rootURL: "http://localhost:3000/grafana", expectedAppURL: "http://localhost:3000/grafana/", expectedAppSubURL: "/grafana"}, | 
					
						
							|  |  |  | 		{rootURL: "http://localhost:3000/grafana/", expectedAppURL: "http://localhost:3000/grafana/", expectedAppSubURL: "/grafana"}, | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	for _, tc := range testCases { | 
					
						
							|  |  |  | 		f := ini.Empty() | 
					
						
							| 
									
										
										
										
											2021-11-09 00:56:56 +08:00
										 |  |  | 		cfg := NewCfg() | 
					
						
							| 
									
										
										
										
											2020-03-11 17:04:48 +08:00
										 |  |  | 		s, err := f.NewSection("server") | 
					
						
							|  |  |  | 		require.NoError(t, err) | 
					
						
							|  |  |  | 		_, err = s.NewKey("root_url", tc.rootURL) | 
					
						
							|  |  |  | 		require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2021-11-09 00:56:56 +08:00
										 |  |  | 		appURL, appSubURL, err := cfg.parseAppUrlAndSubUrl(s) | 
					
						
							| 
									
										
										
										
											2020-03-11 17:04:48 +08:00
										 |  |  | 		require.NoError(t, err) | 
					
						
							|  |  |  | 		require.Equal(t, tc.expectedAppURL, appURL) | 
					
						
							|  |  |  | 		require.Equal(t, tc.expectedAppSubURL, appSubURL) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | } | 
					
						
							| 
									
										
										
										
											2020-11-03 02:26:19 +08:00
										 |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-09-14 21:57:38 +08:00
										 |  |  | func TestAuthDurationSettings(t *testing.T) { | 
					
						
							| 
									
										
										
										
											2020-11-03 02:26:19 +08:00
										 |  |  | 	const maxInactiveDaysTest = 240 * time.Hour | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2020-09-14 21:57:38 +08:00
										 |  |  | 	f := ini.Empty() | 
					
						
							|  |  |  | 	cfg := NewCfg() | 
					
						
							|  |  |  | 	sec, err := f.NewSection("auth") | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2022-06-01 18:29:15 +08:00
										 |  |  | 	_, err = sec.NewKey("login_maximum_inactive_lifetime_duration", "10d") | 
					
						
							| 
									
										
										
										
											2020-09-14 21:57:38 +08:00
										 |  |  | 	require.NoError(t, err) | 
					
						
							|  |  |  | 	err = readAuthSettings(f, cfg) | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							|  |  |  | 	require.Equal(t, maxInactiveDaysTest, cfg.LoginMaxInactiveLifetime) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	f = ini.Empty() | 
					
						
							|  |  |  | 	sec, err = f.NewSection("auth") | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							|  |  |  | 	_, err = sec.NewKey("login_maximum_inactive_lifetime_duration", "824h") | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2020-11-03 02:26:19 +08:00
										 |  |  | 	maxInactiveDurationTest, err := time.ParseDuration("824h") | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2020-09-14 21:57:38 +08:00
										 |  |  | 	err = readAuthSettings(f, cfg) | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							|  |  |  | 	require.Equal(t, maxInactiveDurationTest, cfg.LoginMaxInactiveLifetime) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	f = ini.Empty() | 
					
						
							|  |  |  | 	sec, err = f.NewSection("auth") | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2022-06-01 18:29:15 +08:00
										 |  |  | 	_, err = sec.NewKey("login_maximum_lifetime_duration", "24d") | 
					
						
							| 
									
										
										
										
											2020-09-14 21:57:38 +08:00
										 |  |  | 	require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2020-11-03 02:26:19 +08:00
										 |  |  | 	maxLifetimeDaysTest, err := time.ParseDuration("576h") | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2020-09-14 21:57:38 +08:00
										 |  |  | 	err = readAuthSettings(f, cfg) | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							|  |  |  | 	require.Equal(t, maxLifetimeDaysTest, cfg.LoginMaxLifetime) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 	f = ini.Empty() | 
					
						
							|  |  |  | 	sec, err = f.NewSection("auth") | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							|  |  |  | 	_, err = sec.NewKey("login_maximum_lifetime_duration", "824h") | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2020-11-03 02:26:19 +08:00
										 |  |  | 	maxLifetimeDurationTest, err := time.ParseDuration("824h") | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2020-09-14 21:57:38 +08:00
										 |  |  | 	err = readAuthSettings(f, cfg) | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							|  |  |  | 	require.Equal(t, maxLifetimeDurationTest, cfg.LoginMaxLifetime) | 
					
						
							| 
									
										
										
										
											2021-01-07 15:58:46 +08:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	f = ini.Empty() | 
					
						
							|  |  |  | 	sec, err = f.NewSection("auth") | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							|  |  |  | 	_, err = sec.NewKey("login_maximum_lifetime_duration", "") | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							|  |  |  | 	maxLifetimeDurationTest, err = time.ParseDuration("720h") | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							|  |  |  | 	err = readAuthSettings(f, cfg) | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							|  |  |  | 	require.Equal(t, maxLifetimeDurationTest, cfg.LoginMaxLifetime) | 
					
						
							| 
									
										
										
										
											2020-09-14 21:57:38 +08:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2021-02-01 17:13:09 +08:00
										 |  |  | 
 | 
					
						
							|  |  |  | func TestGetCDNPath(t *testing.T) { | 
					
						
							|  |  |  | 	var err error | 
					
						
							|  |  |  | 	cfg := NewCfg() | 
					
						
							|  |  |  | 	cfg.BuildVersion = "v7.5.0-11124" | 
					
						
							|  |  |  | 	cfg.CDNRootURL, err = url.Parse("http://cdn.grafana.com") | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-02-02 19:34:59 +08:00
										 |  |  | 	require.Equal(t, "http://cdn.grafana.com/grafana-oss/v7.5.0-11124/", cfg.GetContentDeliveryURL("grafana-oss")) | 
					
						
							|  |  |  | 	require.Equal(t, "http://cdn.grafana.com/grafana/v7.5.0-11124/", cfg.GetContentDeliveryURL("grafana")) | 
					
						
							|  |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func TestGetContentDeliveryURLWhenNoCDNRootURLIsSet(t *testing.T) { | 
					
						
							|  |  |  | 	cfg := NewCfg() | 
					
						
							|  |  |  | 	require.Equal(t, "", cfg.GetContentDeliveryURL("grafana-oss")) | 
					
						
							| 
									
										
										
										
											2021-02-01 17:13:09 +08:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func TestGetCDNPathWithPreReleaseVersionAndSubPath(t *testing.T) { | 
					
						
							|  |  |  | 	var err error | 
					
						
							|  |  |  | 	cfg := NewCfg() | 
					
						
							|  |  |  | 	cfg.BuildVersion = "v7.5.0-11124pre" | 
					
						
							|  |  |  | 	cfg.CDNRootURL, err = url.Parse("http://cdn.grafana.com/sub") | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2021-09-10 22:48:13 +08:00
										 |  |  | 	require.Equal(t, "http://cdn.grafana.com/sub/grafana-oss/v7.5.0-11124pre/", cfg.GetContentDeliveryURL("grafana-oss")) | 
					
						
							|  |  |  | 	require.Equal(t, "http://cdn.grafana.com/sub/grafana/v7.5.0-11124pre/", cfg.GetContentDeliveryURL("grafana")) | 
					
						
							| 
									
										
										
										
											2021-02-01 17:13:09 +08:00
										 |  |  | } | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | // Adding a case for this in case we switch to proper semver version strings
 | 
					
						
							|  |  |  | func TestGetCDNPathWithAlphaVersion(t *testing.T) { | 
					
						
							|  |  |  | 	var err error | 
					
						
							|  |  |  | 	cfg := NewCfg() | 
					
						
							|  |  |  | 	cfg.BuildVersion = "v7.5.0-alpha.11124" | 
					
						
							|  |  |  | 	cfg.CDNRootURL, err = url.Parse("http://cdn.grafana.com") | 
					
						
							|  |  |  | 	require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2021-09-10 22:48:13 +08:00
										 |  |  | 	require.Equal(t, "http://cdn.grafana.com/grafana-oss/v7.5.0-alpha.11124/", cfg.GetContentDeliveryURL("grafana-oss")) | 
					
						
							|  |  |  | 	require.Equal(t, "http://cdn.grafana.com/grafana/v7.5.0-alpha.11124/", cfg.GetContentDeliveryURL("grafana")) | 
					
						
							| 
									
										
										
										
											2021-02-01 17:13:09 +08:00
										 |  |  | } | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 
 | 
					
						
							|  |  |  | func TestAlertingEnabled(t *testing.T) { | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 	anyBoolean := func() bool { | 
					
						
							|  |  |  | 		return rand.Int63()%2 == 0 | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 	testCases := []struct { | 
					
						
							|  |  |  | 		desc                   string | 
					
						
							|  |  |  | 		unifiedAlertingEnabled string | 
					
						
							|  |  |  | 		legacyAlertingEnabled  string | 
					
						
							|  |  |  | 		featureToggleSet       bool | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 		isEnterprise           bool | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 		verifyCfg              func(*testing.T, Cfg, *ini.File) | 
					
						
							|  |  |  | 	}{ | 
					
						
							|  |  |  | 		{ | 
					
						
							|  |  |  | 			desc:                   "when legacy alerting is enabled and unified is disabled", | 
					
						
							|  |  |  | 			legacyAlertingEnabled:  "true", | 
					
						
							|  |  |  | 			unifiedAlertingEnabled: "false", | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 			isEnterprise:           anyBoolean(), | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			verifyCfg: func(t *testing.T, cfg Cfg, f *ini.File) { | 
					
						
							|  |  |  | 				err := readAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.readFeatureToggles(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.ReadUnifiedAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2021-11-25 03:56:07 +08:00
										 |  |  | 				assert.NotNil(t, cfg.UnifiedAlerting.Enabled) | 
					
						
							|  |  |  | 				assert.Equal(t, *cfg.UnifiedAlerting.Enabled, false) | 
					
						
							|  |  |  | 				assert.NotNil(t, AlertingEnabled) | 
					
						
							|  |  |  | 				assert.Equal(t, *AlertingEnabled, true) | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			}, | 
					
						
							|  |  |  | 		}, | 
					
						
							|  |  |  | 		{ | 
					
						
							|  |  |  | 			desc:                   "when legacy alerting is disabled and unified is enabled", | 
					
						
							|  |  |  | 			legacyAlertingEnabled:  "false", | 
					
						
							|  |  |  | 			unifiedAlertingEnabled: "true", | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 			isEnterprise:           anyBoolean(), | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			verifyCfg: func(t *testing.T, cfg Cfg, f *ini.File) { | 
					
						
							|  |  |  | 				err := readAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.readFeatureToggles(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.ReadUnifiedAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2021-11-25 03:56:07 +08:00
										 |  |  | 				assert.NotNil(t, cfg.UnifiedAlerting.Enabled) | 
					
						
							|  |  |  | 				assert.Equal(t, *cfg.UnifiedAlerting.Enabled, true) | 
					
						
							|  |  |  | 				assert.NotNil(t, AlertingEnabled) | 
					
						
							|  |  |  | 				assert.Equal(t, *AlertingEnabled, false) | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			}, | 
					
						
							|  |  |  | 		}, | 
					
						
							|  |  |  | 		{ | 
					
						
							| 
									
										
										
										
											2021-11-25 03:56:07 +08:00
										 |  |  | 			desc:                   "when both alerting are enabled", | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			legacyAlertingEnabled:  "true", | 
					
						
							|  |  |  | 			unifiedAlertingEnabled: "true", | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 			isEnterprise:           anyBoolean(), | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			verifyCfg: func(t *testing.T, cfg Cfg, f *ini.File) { | 
					
						
							|  |  |  | 				err := readAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.readFeatureToggles(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.ReadUnifiedAlertingSettings(f) | 
					
						
							|  |  |  | 				require.Error(t, err) | 
					
						
							|  |  |  | 			}, | 
					
						
							|  |  |  | 		}, | 
					
						
							|  |  |  | 		{ | 
					
						
							| 
									
										
										
										
											2021-11-25 03:56:07 +08:00
										 |  |  | 			desc:                   "when legacy alerting is invalid (or not defined) and unified is disabled", | 
					
						
							|  |  |  | 			legacyAlertingEnabled:  "", | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			unifiedAlertingEnabled: "false", | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 			isEnterprise:           anyBoolean(), | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			verifyCfg: func(t *testing.T, cfg Cfg, f *ini.File) { | 
					
						
							|  |  |  | 				err := readAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.readFeatureToggles(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.ReadUnifiedAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2021-11-25 03:56:07 +08:00
										 |  |  | 				assert.NotNil(t, cfg.UnifiedAlerting.Enabled) | 
					
						
							|  |  |  | 				assert.Equal(t, *cfg.UnifiedAlerting.Enabled, false) | 
					
						
							|  |  |  | 				assert.NotNil(t, AlertingEnabled) | 
					
						
							|  |  |  | 				assert.Equal(t, *AlertingEnabled, true) | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			}, | 
					
						
							|  |  |  | 		}, | 
					
						
							|  |  |  | 		{ | 
					
						
							| 
									
										
										
										
											2021-11-25 03:56:07 +08:00
										 |  |  | 			desc:                   "when legacy alerting is invalid (or not defined) and unified is enabled", | 
					
						
							|  |  |  | 			legacyAlertingEnabled:  "", | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			unifiedAlertingEnabled: "true", | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 			isEnterprise:           anyBoolean(), | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			verifyCfg: func(t *testing.T, cfg Cfg, f *ini.File) { | 
					
						
							|  |  |  | 				err := readAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.readFeatureToggles(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.ReadUnifiedAlertingSettings(f) | 
					
						
							| 
									
										
										
										
											2021-11-25 03:56:07 +08:00
										 |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				assert.NotNil(t, cfg.UnifiedAlerting.Enabled) | 
					
						
							|  |  |  | 				assert.Equal(t, *cfg.UnifiedAlerting.Enabled, true) | 
					
						
							|  |  |  | 				assert.NotNil(t, AlertingEnabled) | 
					
						
							|  |  |  | 				assert.Equal(t, *AlertingEnabled, false) | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			}, | 
					
						
							|  |  |  | 		}, | 
					
						
							|  |  |  | 		{ | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 			desc:                   "when legacy alerting is enabled and unified is invalid (or not defined) [OSS]", | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			legacyAlertingEnabled:  "true", | 
					
						
							| 
									
										
										
										
											2021-11-25 03:56:07 +08:00
										 |  |  | 			unifiedAlertingEnabled: "", | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 			isEnterprise:           false, | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			verifyCfg: func(t *testing.T, cfg Cfg, f *ini.File) { | 
					
						
							|  |  |  | 				err := readAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.readFeatureToggles(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.ReadUnifiedAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2022-05-30 23:47:15 +08:00
										 |  |  | 				assert.NotNil(t, cfg.UnifiedAlerting.Enabled) | 
					
						
							|  |  |  | 				assert.Equal(t, true, *cfg.UnifiedAlerting.Enabled) | 
					
						
							| 
									
										
										
										
											2021-11-25 03:56:07 +08:00
										 |  |  | 				assert.NotNil(t, AlertingEnabled) | 
					
						
							| 
									
										
										
										
											2022-05-30 23:47:15 +08:00
										 |  |  | 				assert.Equal(t, false, *AlertingEnabled) | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			}, | 
					
						
							|  |  |  | 		}, | 
					
						
							|  |  |  | 		{ | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 			desc:                   "when legacy alerting is enabled and unified is invalid (or not defined) [Enterprise]", | 
					
						
							|  |  |  | 			legacyAlertingEnabled:  "true", | 
					
						
							|  |  |  | 			unifiedAlertingEnabled: "", | 
					
						
							|  |  |  | 			isEnterprise:           true, | 
					
						
							|  |  |  | 			verifyCfg: func(t *testing.T, cfg Cfg, f *ini.File) { | 
					
						
							|  |  |  | 				err := readAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.readFeatureToggles(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.ReadUnifiedAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				assert.NotNil(t, cfg.UnifiedAlerting.Enabled) | 
					
						
							| 
									
										
										
										
											2022-05-30 23:47:15 +08:00
										 |  |  | 				assert.Equal(t, true, *cfg.UnifiedAlerting.Enabled) | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 				assert.NotNil(t, AlertingEnabled) | 
					
						
							| 
									
										
										
										
											2022-05-30 23:47:15 +08:00
										 |  |  | 				assert.Equal(t, false, *AlertingEnabled) | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 			}, | 
					
						
							|  |  |  | 		}, | 
					
						
							|  |  |  | 		{ | 
					
						
							|  |  |  | 			desc:                   "when legacy alerting is disabled and unified is invalid (or not defined) [OSS]", | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			legacyAlertingEnabled:  "false", | 
					
						
							|  |  |  | 			unifiedAlertingEnabled: "invalid", | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 			isEnterprise:           false, | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			verifyCfg: func(t *testing.T, cfg Cfg, f *ini.File) { | 
					
						
							|  |  |  | 				err := readAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.readFeatureToggles(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.ReadUnifiedAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2021-11-25 03:56:07 +08:00
										 |  |  | 				assert.NotNil(t, cfg.UnifiedAlerting.Enabled) | 
					
						
							|  |  |  | 				assert.Equal(t, *cfg.UnifiedAlerting.Enabled, true) | 
					
						
							|  |  |  | 				assert.NotNil(t, AlertingEnabled) | 
					
						
							|  |  |  | 				assert.Equal(t, *AlertingEnabled, false) | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			}, | 
					
						
							|  |  |  | 		}, | 
					
						
							|  |  |  | 		{ | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 			desc:                   "when legacy alerting is disabled and unified is invalid (or not defined) [Enterprise]", | 
					
						
							|  |  |  | 			legacyAlertingEnabled:  "false", | 
					
						
							|  |  |  | 			unifiedAlertingEnabled: "invalid", | 
					
						
							|  |  |  | 			isEnterprise:           true, | 
					
						
							|  |  |  | 			verifyCfg: func(t *testing.T, cfg Cfg, f *ini.File) { | 
					
						
							|  |  |  | 				err := readAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.readFeatureToggles(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.ReadUnifiedAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				assert.NotNil(t, cfg.UnifiedAlerting.Enabled) | 
					
						
							| 
									
										
										
										
											2022-05-30 23:47:15 +08:00
										 |  |  | 				assert.Equal(t, *cfg.UnifiedAlerting.Enabled, true) | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 				assert.NotNil(t, AlertingEnabled) | 
					
						
							|  |  |  | 				assert.Equal(t, *AlertingEnabled, false) | 
					
						
							|  |  |  | 			}, | 
					
						
							|  |  |  | 		}, | 
					
						
							|  |  |  | 		{ | 
					
						
							|  |  |  | 			desc:                   "when both are invalid (or not defined) [OSS]", | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			legacyAlertingEnabled:  "invalid", | 
					
						
							|  |  |  | 			unifiedAlertingEnabled: "invalid", | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 			isEnterprise:           false, | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			verifyCfg: func(t *testing.T, cfg Cfg, f *ini.File) { | 
					
						
							|  |  |  | 				err := readAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.readFeatureToggles(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.ReadUnifiedAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2022-05-30 23:47:15 +08:00
										 |  |  | 				assert.NotNil(t, cfg.UnifiedAlerting.Enabled) | 
					
						
							|  |  |  | 				assert.True(t, *cfg.UnifiedAlerting.Enabled) | 
					
						
							| 
									
										
										
										
											2021-11-25 03:56:07 +08:00
										 |  |  | 				assert.Nil(t, AlertingEnabled) | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			}, | 
					
						
							|  |  |  | 		}, | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 		{ | 
					
						
							|  |  |  | 			desc:                   "when both are invalid (or not defined) [Enterprise]", | 
					
						
							|  |  |  | 			legacyAlertingEnabled:  "invalid", | 
					
						
							|  |  |  | 			unifiedAlertingEnabled: "invalid", | 
					
						
							|  |  |  | 			isEnterprise:           true, | 
					
						
							|  |  |  | 			verifyCfg: func(t *testing.T, cfg Cfg, f *ini.File) { | 
					
						
							|  |  |  | 				err := readAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.readFeatureToggles(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.ReadUnifiedAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				assert.NotNil(t, cfg.UnifiedAlerting.Enabled) | 
					
						
							| 
									
										
										
										
											2022-05-30 23:47:15 +08:00
										 |  |  | 				assert.True(t, *cfg.UnifiedAlerting.Enabled) | 
					
						
							|  |  |  | 				assert.Nil(t, AlertingEnabled) | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 			}, | 
					
						
							|  |  |  | 		}, | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 		{ | 
					
						
							| 
									
										
										
										
											2021-11-25 03:56:07 +08:00
										 |  |  | 			desc:                   "when both are false", | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			legacyAlertingEnabled:  "false", | 
					
						
							|  |  |  | 			unifiedAlertingEnabled: "false", | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 			isEnterprise:           anyBoolean(), | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			verifyCfg: func(t *testing.T, cfg Cfg, f *ini.File) { | 
					
						
							|  |  |  | 				err := readAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.readFeatureToggles(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							|  |  |  | 				err = cfg.ReadUnifiedAlertingSettings(f) | 
					
						
							|  |  |  | 				require.NoError(t, err) | 
					
						
							| 
									
										
										
										
											2021-11-25 03:56:07 +08:00
										 |  |  | 				assert.NotNil(t, cfg.UnifiedAlerting.Enabled) | 
					
						
							|  |  |  | 				assert.Equal(t, *cfg.UnifiedAlerting.Enabled, false) | 
					
						
							|  |  |  | 				assert.NotNil(t, AlertingEnabled) | 
					
						
							|  |  |  | 				assert.Equal(t, *AlertingEnabled, false) | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			}, | 
					
						
							|  |  |  | 		}, | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 	var isEnterpriseOld = IsEnterprise | 
					
						
							|  |  |  | 	t.Cleanup(func() { | 
					
						
							|  |  |  | 		IsEnterprise = isEnterpriseOld | 
					
						
							|  |  |  | 	}) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 	for _, tc := range testCases { | 
					
						
							|  |  |  | 		t.Run(tc.desc, func(t *testing.T) { | 
					
						
							| 
									
										
										
										
											2021-11-30 03:51:15 +08:00
										 |  |  | 			IsEnterprise = tc.isEnterprise | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			t.Cleanup(func() { | 
					
						
							| 
									
										
										
										
											2021-11-25 03:56:07 +08:00
										 |  |  | 				AlertingEnabled = nil | 
					
						
							| 
									
										
										
										
											2021-09-29 22:16:40 +08:00
										 |  |  | 			}) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 			f := ini.Empty() | 
					
						
							|  |  |  | 			cfg := NewCfg() | 
					
						
							|  |  |  | 			unifiedAlertingSec, err := f.NewSection("unified_alerting") | 
					
						
							|  |  |  | 			require.NoError(t, err) | 
					
						
							|  |  |  | 			_, err = unifiedAlertingSec.NewKey("enabled", tc.unifiedAlertingEnabled) | 
					
						
							|  |  |  | 			require.NoError(t, err) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 			alertingSec, err := f.NewSection("alerting") | 
					
						
							|  |  |  | 			require.NoError(t, err) | 
					
						
							|  |  |  | 			_, err = alertingSec.NewKey("enabled", tc.legacyAlertingEnabled) | 
					
						
							|  |  |  | 			require.NoError(t, err) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 			tc.verifyCfg(t, *cfg, f) | 
					
						
							|  |  |  | 		}) | 
					
						
							|  |  |  | 	} | 
					
						
							|  |  |  | } |