From cfa436dc16646d4a076775e000181613f8d25817 Mon Sep 17 00:00:00 2001 From: Siva Prasad Date: Tue, 7 Aug 2018 08:29:48 -0400 Subject: [PATCH] Revert "CA initialization while boostrapping and TestLeader_ChangeServerID fix." (#4497) * Revert "BUGFIX: Unit test relying on WaitForLeader() did not work due to wrong test (#4472)" This reverts commit cec5d7239621e0732b3f70158addb1899442acb3. * Revert "CA initialization while boostrapping and TestLeader_ChangeServerID fix. (#4493)" This reverts commit 589b589b53e56af38de25db9b56967bdf1f2c069. --- agent/connect/ca/provider_consul.go | 17 +++++++++++------ agent/consul/leader_test.go | 22 +++++++++++++--------- 2 files changed, 24 insertions(+), 15 deletions(-) diff --git a/agent/connect/ca/provider_consul.go b/agent/connect/ca/provider_consul.go index f4c1ab4d9..5cbf744ab 100644 --- a/agent/connect/ca/provider_consul.go +++ b/agent/connect/ca/provider_consul.go @@ -45,18 +45,23 @@ func NewConsulProvider(rawConfig map[string]interface{}, delegate ConsulProvider // Check if this configuration of the provider has already been // initialized in the state store. state := delegate.State() - idx, providerState, err := state.CAProviderState(provider.id) + _, providerState, err := state.CAProviderState(provider.id) if err != nil { return nil, err } + // Exit early if the state store has already been populated for this config. + if providerState != nil { + return provider, nil + } + newState := structs.CAConsulProviderState{ ID: provider.id, } // Write the initial provider state to get the index to use for the // CA serial number. - if providerState == nil { + { args := &structs.CARequest{ Op: structs.CAOpSetProviderState, ProviderState: &newState, @@ -64,11 +69,11 @@ func NewConsulProvider(rawConfig map[string]interface{}, delegate ConsulProvider if err := delegate.ApplyCARequest(args); err != nil { return nil, err } + } - idx, _, err = state.CAProviderState(provider.id) - if err != nil { - return nil, err - } + idx, _, err := state.CAProviderState(provider.id) + if err != nil { + return nil, err } // Generate a private key if needed diff --git a/agent/consul/leader_test.go b/agent/consul/leader_test.go index 9bf8a4f84..c56477abf 100644 --- a/agent/consul/leader_test.go +++ b/agent/consul/leader_test.go @@ -864,16 +864,21 @@ func TestLeader_RollRaftServer(t *testing.T) { func TestLeader_ChangeServerID(t *testing.T) { t.Parallel() - - dir1, s1 := testServerDCBootstrap(t, "dc1", true) + conf := func(c *Config) { + c.Bootstrap = false + c.BootstrapExpect = 3 + c.Datacenter = "dc1" + c.RaftConfig.ProtocolVersion = 3 + } + dir1, s1 := testServerWithConfig(t, conf) defer os.RemoveAll(dir1) defer s1.Shutdown() - dir2, s2 := testServerDCBootstrap(t, "dc1", false) + dir2, s2 := testServerWithConfig(t, conf) defer os.RemoveAll(dir2) defer s2.Shutdown() - dir3, s3 := testServerDCBootstrap(t, "dc1", false) + dir3, s3 := testServerWithConfig(t, conf) defer os.RemoveAll(dir3) defer s3.Shutdown() @@ -886,12 +891,8 @@ func TestLeader_ChangeServerID(t *testing.T) { retry.Run(t, func(r *retry.R) { r.Check(wantPeers(s, 3)) }) } - testrpc.WaitForLeader(t, s1.RPC, "dc1") - // Shut down a server, freeing up its address/port - if err := s3.Shutdown(); err != nil { - t.Fatal(err) - } + s3.Shutdown() retry.Run(t, func(r *retry.R) { alive := 0 @@ -908,6 +909,9 @@ func TestLeader_ChangeServerID(t *testing.T) { // Bring up a new server with s3's address that will get a different ID dir4, s4 := testServerWithConfig(t, func(c *Config) { c.Bootstrap = false + c.BootstrapExpect = 3 + c.Datacenter = "dc1" + c.RaftConfig.ProtocolVersion = 3 c.SerfLANConfig.MemberlistConfig = s3.config.SerfLANConfig.MemberlistConfig c.RPCAddr = s3.config.RPCAddr c.RPCAdvertise = s3.config.RPCAdvertise