mirror of https://github.com/hashicorp/consul
76 lines
1.9 KiB
Go
76 lines
1.9 KiB
Go
package consul
|
|
|
|
import (
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/consul/agent/structs"
|
|
"github.com/hashicorp/consul/testrpc"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestLeader_SystemMetadata_CRUD(t *testing.T) {
|
|
if testing.Short() {
|
|
t.Skip("too slow for testing.Short")
|
|
}
|
|
|
|
// This test is a little strange because it is testing behavior that
|
|
// doesn't have an exposed RPC. We're just testing the full round trip of
|
|
// raft+fsm For now,
|
|
|
|
dir1, srv := testServerWithConfig(t, func(c *Config) {
|
|
// We disable connect here so we skip inserting intention-migration
|
|
// related system metadata in the background.
|
|
c.ConnectEnabled = false
|
|
})
|
|
defer os.RemoveAll(dir1)
|
|
defer srv.Shutdown()
|
|
codec := rpcClient(t, srv)
|
|
defer codec.Close()
|
|
|
|
testrpc.WaitForLeader(t, srv.RPC, "dc1")
|
|
|
|
state := srv.fsm.State()
|
|
|
|
// Initially empty
|
|
_, entries, err := state.SystemMetadataList(nil)
|
|
require.NoError(t, err)
|
|
require.Len(t, entries, 0)
|
|
|
|
// Create 3
|
|
require.NoError(t, srv.setSystemMetadataKey("key1", "val1"))
|
|
require.NoError(t, srv.setSystemMetadataKey("key2", "val2"))
|
|
require.NoError(t, srv.setSystemMetadataKey("key3", ""))
|
|
|
|
mapify := func(entries []*structs.SystemMetadataEntry) map[string]string {
|
|
m := make(map[string]string)
|
|
for _, entry := range entries {
|
|
m[entry.Key] = entry.Value
|
|
}
|
|
return m
|
|
}
|
|
|
|
_, entries, err = state.SystemMetadataList(nil)
|
|
require.NoError(t, err)
|
|
require.Len(t, entries, 3)
|
|
|
|
require.Equal(t, map[string]string{
|
|
"key1": "val1",
|
|
"key2": "val2",
|
|
"key3": "",
|
|
}, mapify(entries))
|
|
|
|
// Update one and delete one.
|
|
require.NoError(t, srv.setSystemMetadataKey("key3", "val3"))
|
|
require.NoError(t, srv.deleteSystemMetadataKey("key1"))
|
|
|
|
_, entries, err = state.SystemMetadataList(nil)
|
|
require.NoError(t, err)
|
|
require.Len(t, entries, 2)
|
|
|
|
require.Equal(t, map[string]string{
|
|
"key2": "val2",
|
|
"key3": "val3",
|
|
}, mapify(entries))
|
|
}
|