Ejemplo n.º 1
0
				Expect(err).To(MatchError("members error"))
				Expect(consulAPIAgent.MembersCallCount()).To(Equal(1))
			})
		})
	})

	Describe("SetKeys", func() {
		BeforeEach(func() {
			consulRPCClient.InstallKeyReturns(nil)
			consulRPCClient.UseKeyReturns(nil)
			consulRPCClient.ListKeysReturns([]string{"key3", "key4"}, nil)
			consulRPCClient.RemoveKeyReturns(nil)
		})

		It("installs the given keys", func() {
			Expect(client.SetKeys([]string{"key1", "key2"})).To(Succeed())
			Expect(consulRPCClient.InstallKeyCallCount()).To(Equal(2))

			key := consulRPCClient.InstallKeyArgsForCall(0)
			Expect(key).To(Equal("key1"))

			key = consulRPCClient.InstallKeyArgsForCall(1)
			Expect(key).To(Equal("key2"))

			Expect(consulRPCClient.UseKeyCallCount()).To(Equal(1))

			key = consulRPCClient.UseKeyArgsForCall(0)
			Expect(key).To(Equal("key1"))
		})

		Context("when there are extra keys", func() {