| 
									
										
										
										
											2024-08-30 17:59:42 +08:00
										 |  |  | package setting | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | import ( | 
					
						
							|  |  |  | 	"testing" | 
					
						
							| 
									
										
										
										
											2024-12-13 04:41:01 +08:00
										 |  |  | 	"time" | 
					
						
							| 
									
										
										
										
											2024-08-30 17:59:42 +08:00
										 |  |  | 
 | 
					
						
							|  |  |  | 	"github.com/stretchr/testify/assert" | 
					
						
							|  |  |  | ) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | func TestCfg_setUnifiedStorageConfig(t *testing.T) { | 
					
						
							|  |  |  | 	t.Run("read unified_storage configs", func(t *testing.T) { | 
					
						
							|  |  |  | 		cfg := NewCfg() | 
					
						
							|  |  |  | 		err := cfg.Load(CommandLineArgs{HomePath: "../../", Config: "../../conf/defaults.ini"}) | 
					
						
							|  |  |  | 		assert.NoError(t, err) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		s, err := cfg.Raw.NewSection("unified_storage.playlists.playlist.grafana.app") | 
					
						
							|  |  |  | 		assert.NoError(t, err) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		_, err = s.NewKey("dualWriterMode", "2") | 
					
						
							|  |  |  | 		assert.NoError(t, err) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2024-09-25 03:03:15 +08:00
										 |  |  | 		_, err = s.NewKey("dualWriterPeriodicDataSyncJobEnabled", "true") | 
					
						
							|  |  |  | 		assert.NoError(t, err) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2024-12-13 04:41:01 +08:00
										 |  |  | 		_, err = s.NewKey("dataSyncerRecordsLimit", "1001") | 
					
						
							|  |  |  | 		assert.NoError(t, err) | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		_, err = s.NewKey("dataSyncerInterval", "10m") | 
					
						
							|  |  |  | 		assert.NoError(t, err) | 
					
						
							|  |  |  | 
 | 
					
						
							| 
									
										
										
										
											2024-08-30 17:59:42 +08:00
										 |  |  | 		cfg.setUnifiedStorageConfig() | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		value, exists := cfg.UnifiedStorage["playlists.playlist.grafana.app"] | 
					
						
							|  |  |  | 
 | 
					
						
							|  |  |  | 		assert.Equal(t, exists, true) | 
					
						
							| 
									
										
										
										
											2024-09-25 03:03:15 +08:00
										 |  |  | 		assert.Equal(t, value, UnifiedStorageConfig{ | 
					
						
							|  |  |  | 			DualWriterMode:                       2, | 
					
						
							|  |  |  | 			DualWriterPeriodicDataSyncJobEnabled: true, | 
					
						
							| 
									
										
										
										
											2024-12-13 04:41:01 +08:00
										 |  |  | 			DataSyncerRecordsLimit:               1001, | 
					
						
							|  |  |  | 			DataSyncerInterval:                   time.Minute * 10, | 
					
						
							| 
									
										
										
										
											2024-09-25 03:03:15 +08:00
										 |  |  | 		}) | 
					
						
							| 
									
										
										
										
											2024-08-30 17:59:42 +08:00
										 |  |  | 	}) | 
					
						
							|  |  |  | } |