"errors"
"net/http"
. "github.com/cloudfoundry/bosh-agent/agentclient/http"
"github.com/cloudfoundry/bosh-agent/agentclient"
"github.com/cloudfoundry/bosh-agent/agentclient/applyspec"
fakehttpclient "github.com/cloudfoundry/bosh-utils/httpclient/fakes"
boshlog "github.com/cloudfoundry/bosh-utils/logger"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("AgentClient", func() {
var (
fakeHTTPClient *fakehttpclient.FakeHTTPClient
agentClient agentclient.AgentClient
)
BeforeEach(func() {
logger := boshlog.NewLogger(boshlog.LevelNone)
fakeHTTPClient = fakehttpclient.NewFakeHTTPClient()
toleratedErrorCount := 2
agentClient = NewAgentClient("http://localhost:6305", "fake-uuid", 0, toleratedErrorCount, fakeHTTPClient, logger)
})
Describe("get_task", func() {
Context("when the http client errors", func() {
It("should retry", func() {
fakeHTTPClient.SetPostBehavior(`{"value":{"agent_task_id":"fake-agent-task-id","state":"running"}}`, 200, nil)
fakeHTTPClient.SetPostBehavior("", 0, errors.New("connection reset by peer"))
fakeHTTPClient.SetPostBehavior("", 0, errors.New("connection reset by peer"))
fakebicrypto "github.com/cloudfoundry/bosh-init/crypto/fakes"
. "github.com/cloudfoundry/bosh-init/installation/tarball"
fakebiui "github.com/cloudfoundry/bosh-init/ui/fakes"
fakebihttpclient "github.com/cloudfoundry/bosh-utils/httpclient/fakes"
boshlog "github.com/cloudfoundry/bosh-utils/logger"
fakesys "github.com/cloudfoundry/bosh-utils/system/fakes"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("Provider", func() {
var (
provider Provider
cache Cache
fs *fakesys.FakeFileSystem
httpClient *fakebihttpclient.FakeHTTPClient
sha1Calculator *fakebicrypto.FakeSha1Calculator
source *fakeSource
fakeStage *fakebiui.FakeStage
)
BeforeEach(func() {
fs = fakesys.NewFakeFileSystem()
logger := boshlog.NewLogger(boshlog.LevelNone)
cache = NewCache("/fake-base-path", fs, logger)
sha1Calculator = fakebicrypto.NewFakeSha1Calculator()
httpClient = fakebihttpclient.NewFakeHTTPClient()
provider = NewProvider(cache, fs, httpClient, sha1Calculator, 3, 0, logger)
fakeStage = fakebiui.NewFakeStage()
})
package cert_test
import (
. "github.com/cloudfoundry/bosh-agent/agentclient/http"
"github.com/cloudfoundry/bosh-agent/agentclient"
"github.com/cloudfoundry/bosh-agent/settings"
fakehttpclient "github.com/cloudfoundry/bosh-utils/httpclient/fakes"
boshlog "github.com/cloudfoundry/bosh-utils/logger"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
)
var _ = Describe("CertManager", func() {
var (
fakeHTTPClient *fakehttpclient.FakeHTTPClient
agentClient agentclient.AgentClient
)
BeforeEach(func() {
logger := boshlog.NewLogger(boshlog.LevelNone)
fakeHTTPClient = fakehttpclient.NewFakeHTTPClient()
agentClient = NewAgentClient("http://localhost:6305", "fake-uuid", 0, 10, fakeHTTPClient, logger)
fakeHTTPClient.SetPostBehavior(`{"value":{"agent_task_id":"fake-agent-task-id","state":"running"}}`, 200, nil)
fakeHTTPClient.SetPostBehavior(`{"value":"updated"}`, 200, nil)
})
It("adds and registers new certs on a fresh machine", func() {
var cert string = "This certificate is the first one. It's more awesome than the other one.\n-----BEGIN CERTIFICATE-----\nMIIEJDCCAwygAwIBAgIJAO+CqgiJnCgpMA0GCSqGSIb3DQEBBQUAMGkxCzAJBgNV\nBAYTAkNBMRMwEQYDVQQIEwpTb21lLVN0YXRlMSIBAgIJAO+CqgiJnCgpMA0GCSqGSIb3DQEBBQUAMGkxCzAJBgNV\nBAYTAkNBMRMwEQYDVQQIEwpTb21lLVN0YXRlMSEwHwYDVQQKExhJbnRlcm5ldCBX\naWRnaXRzIFB0eSBMdGQxIjAgBgNVBAMTGWR4MTkwLnRvci5waXZvdGFsbGFicy5EwHwYDVQQKExhJbnRlcm5ldCBX\naWRnaXRzIFB0eSBMdGQxIjAgBgNVBAMTGWR4MTkwLnRvci5waXZvdGFsbGFicy5j\nb20wHhcNMTUwNTEzMTM1NjA2WhcNMjUwNTEwMTM1NjA2WjBpMQswCQYDVQQGEwJD\nQTETMBEGA1UECBMKU29tZGackAF\nqokoSBXzJCJTt2P681gyqBDr/hUYzqpoXUsOTRisScbEbaSv8hTiTeFJUMyNQAqn\nDtmvI8bXKxU=\n-----END CERTIFICATE-----\n"
updateSettings := settings.UpdateSettings{TrustedCerts: cert}
err := agentClient.UpdateSettings(updateSettings)