mirror of
https://github.com/pomerium/pomerium.git
synced 2025-04-30 10:56:28 +02:00
de603f87de
* Add new configurable bootstrap writers (#2405) This PR adds the ability to configure different backends to use for storing modifications to the zero bootstrap config. The two currently implemented backends allow writing changes to a file or to a Kubernetes secret. Backend selection is determined by the scheme in a URI passed to the flag '--config-writeback-uri'. In a Kubernetes environment, where the bootstrap config is mounted into the pod from a secret, this option allows Pomerium to write changes back to the secret, as writes to the mounted secret file on disk are not persisted. * Use env vars for bootstrap config filepath/writeback uri * linter pass and code cleanup * Add new config writer options mechanism This moves the encryption cipher parameter out of the WriteConfig() method in the ConfigWriter interface and into a new ConfigWriterOptions struct. Options (e.g. cipher) can be applied to an existing ConfigWriter to allow customizing implementation-specific behavior. * Code cleanup/lint fixes * Move vendored k8s code into separate package, and add license header and package comment
67 lines
1.5 KiB
Go
67 lines
1.5 KiB
Go
package filesystem_test
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/pomerium/pomerium/internal/zero/bootstrap"
|
|
"github.com/pomerium/pomerium/internal/zero/bootstrap/writers"
|
|
"github.com/pomerium/pomerium/pkg/cryptutil"
|
|
cluster_api "github.com/pomerium/pomerium/pkg/zero/cluster"
|
|
)
|
|
|
|
func TestFileWriter(t *testing.T) {
|
|
cipher, err := cryptutil.NewAEADCipher(cryptutil.NewKey())
|
|
require.NoError(t, err)
|
|
|
|
txt := "test"
|
|
src := cluster_api.BootstrapConfig{
|
|
DatabrokerStorageConnection: &txt,
|
|
}
|
|
|
|
fd, err := os.CreateTemp(t.TempDir(), "test.data")
|
|
require.NoError(t, err)
|
|
require.NoError(t, fd.Close())
|
|
|
|
writer, err := writers.NewForURI(fmt.Sprintf("file://%s", fd.Name()))
|
|
require.NoError(t, err)
|
|
|
|
writer = writer.WithOptions(writers.ConfigWriterOptions{
|
|
Cipher: cipher,
|
|
})
|
|
require.NoError(t, bootstrap.SaveBootstrapConfig(context.Background(), writer, &src))
|
|
|
|
dst, err := bootstrap.LoadBootstrapConfigFromFile(fd.Name(), cipher)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, src, *dst)
|
|
}
|
|
|
|
func TestNewForURI(t *testing.T) {
|
|
for _, tc := range []struct {
|
|
uri string
|
|
err string
|
|
}{
|
|
{
|
|
uri: "file:///path/to/file",
|
|
},
|
|
{
|
|
uri: "file://path/to/file",
|
|
err: `invalid file uri "file://path/to/file" (did you mean "file:///path/to/file"?)`,
|
|
},
|
|
} {
|
|
w, err := writers.NewForURI(tc.uri)
|
|
if tc.err == "" {
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, w)
|
|
} else {
|
|
assert.EqualError(t, err, tc.err)
|
|
assert.Nil(t, w)
|
|
}
|
|
}
|
|
}
|