"fmt"
"net/http"
"net/http/httptest"
"strings"
. "github.com/onsi/ginkgo"
. "github.com/onsi/gomega"
. "bosh/infrastructure"
fakeinf "bosh/infrastructure/fakes"
boshsettings "bosh/settings"
)
var _ = Describe("concreteRegistry", func() {
var (
metadataService *fakeinf.FakeMetadataService
registry Registry
)
BeforeEach(func() {
metadataService = &fakeinf.FakeMetadataService{}
registry = NewConcreteRegistry(metadataService, false)
})
Describe("GetSettings", func() {
var (
ts *httptest.Server
settingsJSON string
)
BeforeEach(func() {
boshRegistryHandler := http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
func init() {
Describe("awsInfrastructure", func() {
var (
metadataService *fakeinf.FakeMetadataService
registry *fakeinf.FakeRegistry
platform *fakeplatform.FakePlatform
devicePathResolver *fakedpresolv.FakeDevicePathResolver
aws Infrastructure
)
BeforeEach(func() {
metadataService = &fakeinf.FakeMetadataService{}
registry = &fakeinf.FakeRegistry{}
platform = fakeplatform.NewFakePlatform()
devicePathResolver = fakedpresolv.NewFakeDevicePathResolver()
logger := boshlog.NewLogger(boshlog.LevelNone)
aws = NewAwsInfrastructure(metadataService, registry, platform, devicePathResolver, logger)
})
Describe("SetupSSH", func() {
It("gets the public key and sets up ssh via the platform", func() {
metadataService.PublicKey = "fake-public-key"
err := aws.SetupSSH("vcap")
Expect(err).NotTo(HaveOccurred())
Expect(platform.SetupSSHPublicKey).To(Equal("fake-public-key"))
Expect(platform.SetupSSHUsername).To(Equal("vcap"))
})
It("returns error without configuring ssh on the platform if getting public key fails", func() {
metadataService.GetPublicKeyErr = errors.New("fake-get-public-key-err")
err := aws.SetupSSH("vcap")
Expect(err).To(HaveOccurred())
Expect(err.Error()).To(ContainSubstring("fake-get-public-key-err"))
Expect(platform.SetupSSHCalled).To(BeFalse())
})
It("returns error if configuring ssh on the platform fails", func() {
platform.SetupSSHErr = errors.New("fake-setup-ssh-err")
err := aws.SetupSSH("vcap")
Expect(err).To(HaveOccurred())
Expect(err.Error()).To(ContainSubstring("fake-setup-ssh-err"))
})
})
Describe("GetSettings", func() {
It("gets settings", func() {
settings := boshsettings.Settings{AgentID: "fake-agent-id"}
registry.Settings = settings
settings, err := aws.GetSettings()
Expect(err).ToNot(HaveOccurred())
Expect(settings).To(Equal(settings))
})
It("returns an error when registry fails to get settings", func() {
registry.GetSettingsErr = errors.New("fake-get-settings-err")
settings, err := aws.GetSettings()
Expect(err).To(HaveOccurred())
Expect(err.Error()).To(ContainSubstring("fake-get-settings-err"))
Expect(settings).To(Equal(boshsettings.Settings{}))
})
})
Describe("SetupNetworking", func() {
It("sets up DHCP on the platform", func() {
networks := boshsettings.Networks{"bosh": boshsettings.Network{}}
err := aws.SetupNetworking(networks)
Expect(err).ToNot(HaveOccurred())
Expect(platform.SetupDhcpNetworks).To(Equal(networks))
})
It("returns error if configuring DHCP fails", func() {
platform.SetupDhcpErr = errors.New("fake-setup-dhcp-err")
err := aws.SetupNetworking(boshsettings.Networks{})
Expect(err).To(HaveOccurred())
Expect(err.Error()).To(ContainSubstring("fake-setup-dhcp-err"))
})
})
Describe("GetEphemeralDiskPath", func() {
It("returns the real disk path given an AWS EBS hint", func() {
platform.NormalizeDiskPathRealPath = "/dev/xvdb"
platform.NormalizeDiskPathFound = true
realPath, found := aws.GetEphemeralDiskPath("/dev/sdb")
Expect(found).To(Equal(true))
Expect(realPath).To(Equal("/dev/xvdb"))
Expect(platform.NormalizeDiskPathPath).To(Equal("/dev/sdb"))
})
It("returns false if path cannot be normalized", func() {
platform.NormalizeDiskPathRealPath = ""
platform.NormalizeDiskPathFound = false
realPath, found := aws.GetEphemeralDiskPath("/dev/sdb")
Expect(found).To(BeFalse())
Expect(realPath).To(Equal(""))
Expect(platform.NormalizeDiskPathPath).To(Equal("/dev/sdb"))
})
It("returns false if device path is empty because ephemeral storage should not be on root partition", func() {
realPath, found := aws.GetEphemeralDiskPath("")
Expect(found).To(BeFalse())
Expect(realPath).To(BeEmpty())
Expect(platform.NormalizeDiskPathCalled).To(BeFalse())
})
})
})
}