mirror of
				https://github.com/siderolabs/talos.git
				synced 2025-10-31 00:11:36 +01:00 
			
		
		
		
	`config.Container` implements a multi-doc container which implements both `Container` interface (encoding, validation, etc.), and `Conifg` interface (accessing parts of the config). Refactor `generate` and `bundle` packages to support multi-doc, and provide backwards compatibility. Implement a first (mostly example) machine config document for SideroLink API URL. Many places don't properly support multi-doc yet (e.g. config patches). Signed-off-by: Andrey Smirnov <andrey.smirnov@talos-systems.com>
		
			
				
	
	
		
			88 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			88 lines
		
	
	
		
			1.9 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| // This Source Code Form is subject to the terms of the Mozilla Public
 | |
| // License, v. 2.0. If a copy of the MPL was not distributed with this
 | |
| // file, You can obtain one at http://mozilla.org/MPL/2.0/.
 | |
| 
 | |
| package environment_test
 | |
| 
 | |
| import (
 | |
| 	"testing"
 | |
| 
 | |
| 	"github.com/siderolabs/go-procfs/procfs"
 | |
| 	"github.com/stretchr/testify/assert"
 | |
| 	"github.com/stretchr/testify/require"
 | |
| 
 | |
| 	"github.com/siderolabs/talos/internal/pkg/environment"
 | |
| 	"github.com/siderolabs/talos/pkg/machinery/config"
 | |
| 	"github.com/siderolabs/talos/pkg/machinery/config/container"
 | |
| 	"github.com/siderolabs/talos/pkg/machinery/config/types/v1alpha1"
 | |
| )
 | |
| 
 | |
| func TestGet(t *testing.T) {
 | |
| 	t.Parallel()
 | |
| 
 | |
| 	for _, test := range []struct {
 | |
| 		name    string
 | |
| 		cmdline string
 | |
| 		cfg     map[string]string
 | |
| 
 | |
| 		expected []string
 | |
| 	}{
 | |
| 		{
 | |
| 			name: "empty",
 | |
| 		},
 | |
| 		{
 | |
| 			name: "machine config only",
 | |
| 			cfg: map[string]string{
 | |
| 				"http_proxy": "http://proxy.example.com:8080",
 | |
| 			},
 | |
| 			expected: []string{
 | |
| 				"http_proxy=http://proxy.example.com:8080",
 | |
| 			},
 | |
| 		},
 | |
| 		{
 | |
| 			name:    "cmdline only",
 | |
| 			cmdline: "talos.environment=foo=bar talos.environment=bar=baz",
 | |
| 			expected: []string{
 | |
| 				"foo=bar",
 | |
| 				"bar=baz",
 | |
| 			},
 | |
| 		},
 | |
| 		{
 | |
| 			name:    "cmdline and machine config",
 | |
| 			cmdline: "talos.environment=foo=bar",
 | |
| 			cfg: map[string]string{
 | |
| 				"http_proxy": "http://proxy.example.com:8080",
 | |
| 			},
 | |
| 			expected: []string{
 | |
| 				"foo=bar",
 | |
| 				"http_proxy=http://proxy.example.com:8080",
 | |
| 			},
 | |
| 		},
 | |
| 	} {
 | |
| 		test := test
 | |
| 
 | |
| 		t.Run(test.name, func(t *testing.T) {
 | |
| 			t.Parallel()
 | |
| 
 | |
| 			cmdline := procfs.NewCmdline(test.cmdline)
 | |
| 
 | |
| 			var cfg config.Config
 | |
| 
 | |
| 			if test.cfg != nil {
 | |
| 				var err error
 | |
| 
 | |
| 				cfg, err = container.New(&v1alpha1.Config{
 | |
| 					MachineConfig: &v1alpha1.MachineConfig{
 | |
| 						MachineEnv: test.cfg,
 | |
| 					},
 | |
| 				})
 | |
| 				require.NoError(t, err)
 | |
| 			}
 | |
| 
 | |
| 			result := environment.GetCmdline(cmdline, cfg)
 | |
| 
 | |
| 			assert.Equal(t, test.expected, result)
 | |
| 		})
 | |
| 	}
 | |
| }
 |