func NewProvider(logger boshlog.Logger, dirProvider boshdirs.DirectoriesProvider) (p provider) { runner := boshsys.NewExecCmdRunner(logger) fs := boshsys.NewOsFileSystem(logger, runner) sigarCollector := boshstats.NewSigarStatsCollector() linuxDiskManager := boshdisk.NewLinuxDiskManager(logger, runner, fs) udev := boshudev.NewConcreteUdevDevice(runner) linuxCdrom := boshcdrom.NewLinuxCdrom("/dev/sr0", udev, runner) linuxCdutil := boshcd.NewCdUtil(dirProvider.SettingsDir(), fs, linuxCdrom) compressor := boshcmd.NewTarballCompressor(runner, fs) copier := boshcmd.NewCpCopier(runner, fs) vitalsService := boshvitals.NewService(sigarCollector, dirProvider) centosNetManager := boshnet.NewCentosNetManager(fs, runner, 10*time.Second) ubuntuNetManager := boshnet.NewUbuntuNetManager(fs, runner, 10*time.Second) centos := NewLinuxPlatform( fs, runner, sigarCollector, compressor, copier, dirProvider, vitalsService, linuxCdutil, linuxDiskManager, centosNetManager, 500*time.Millisecond, logger, ) ubuntu := NewLinuxPlatform( fs, runner, sigarCollector, compressor, copier, dirProvider, vitalsService, linuxCdutil, linuxDiskManager, ubuntuNetManager, 500*time.Millisecond, logger, ) p.platforms = map[string]Platform{ "ubuntu": ubuntu, "centos": centos, "dummy": NewDummyPlatform(sigarCollector, fs, runner, dirProvider, linuxDiskManager), } return }
func NewProvider(logger boshlog.Logger, dirProvider boshdirs.DirectoriesProvider, options ProviderOptions) (p provider) { runner := boshsys.NewExecCmdRunner(logger) fs := boshsys.NewOsFileSystem(logger) linuxDiskManager := boshdisk.NewLinuxDiskManager(logger, runner, fs, options.Linux.BindMountPersistentDisk) udev := boshudev.NewConcreteUdevDevice(runner) linuxCdrom := boshcdrom.NewLinuxCdrom("/dev/sr0", udev, runner) linuxCdutil := boshcd.NewCdUtil(dirProvider.SettingsDir(), fs, linuxCdrom) compressor := boshcmd.NewTarballCompressor(runner, fs) copier := boshcmd.NewCpCopier(runner, fs, logger) sigarCollector := boshstats.NewSigarStatsCollector(&sigar.ConcreteSigar{}) // Kick of stats collection as soon as possible go sigarCollector.StartCollecting(SigarStatsCollectionInterval, nil) vitalsService := boshvitals.NewService(sigarCollector, dirProvider) routesSearcher := boshnet.NewCmdRoutesSearcher(runner) ipResolver := boship.NewIPResolver(boship.NetworkInterfaceToAddrsFunc) defaultNetworkResolver := boshnet.NewDefaultNetworkResolver(routesSearcher, ipResolver) arping := bosharp.NewArping(runner, fs, logger, ArpIterations, ArpIterationDelay, ArpInterfaceCheckDelay) centosNetManager := boshnet.NewCentosNetManager(fs, runner, defaultNetworkResolver, ipResolver, arping, logger) ubuntuNetManager := boshnet.NewUbuntuNetManager(fs, runner, defaultNetworkResolver, ipResolver, arping, logger) centos := NewLinuxPlatform( fs, runner, sigarCollector, compressor, copier, dirProvider, vitalsService, linuxCdutil, linuxDiskManager, centosNetManager, 500*time.Millisecond, options.Linux, logger, ) ubuntu := NewLinuxPlatform( fs, runner, sigarCollector, compressor, copier, dirProvider, vitalsService, linuxCdutil, linuxDiskManager, ubuntuNetManager, 500*time.Millisecond, options.Linux, logger, ) p.platforms = map[string]Platform{ "ubuntu": ubuntu, "centos": centos, "dummy": NewDummyPlatform(sigarCollector, fs, runner, dirProvider, logger), } return }
func init() { const expectedCentosDHCPConfig = `# Generated by bosh-agent option rfc3442-classless-static-routes code 121 = array of unsigned integer 8; send host-name "<hostname>"; request subnet-mask, broadcast-address, time-offset, routers, domain-name, domain-name-servers, domain-search, host-name, netbios-name-servers, netbios-scope, interface-mtu, rfc3442-classless-static-routes, ntp-servers; prepend domain-name-servers zz.zz.zz.zz; prepend domain-name-servers yy.yy.yy.yy; prepend domain-name-servers xx.xx.xx.xx; ` const expectedCentosIfcfg = `DEVICE=eth0 BOOTPROTO=static IPADDR=192.168.195.6 NETMASK=255.255.255.0 BROADCAST=192.168.195.255 GATEWAY=192.168.195.1 ONBOOT=yes` Describe("centos", func() { var ( collector *fakestats.FakeStatsCollector fs *fakesys.FakeFileSystem cmdRunner *fakesys.FakeCmdRunner diskManager *fakedisk.FakeDiskManager dirProvider boshdirs.DirectoriesProvider platform Platform cdutil *fakecd.FakeCdUtil compressor boshcmd.Compressor copier boshcmd.Copier vitalsService boshvitals.Service logger boshlog.Logger ) BeforeEach(func() { collector = &fakestats.FakeStatsCollector{} fs = fakesys.NewFakeFileSystem() cmdRunner = fakesys.NewFakeCmdRunner() diskManager = fakedisk.NewFakeDiskManager() dirProvider = boshdirs.NewDirectoriesProvider("/fake-dir") cdutil = fakecd.NewFakeCdUtil() compressor = boshcmd.NewTarballCompressor(cmdRunner, fs) copier = boshcmd.NewCpCopier(cmdRunner, fs) vitalsService = boshvitals.NewService(collector, dirProvider) logger = boshlog.NewLogger(boshlog.LevelNone) }) JustBeforeEach(func() { netManager := boshnet.NewCentosNetManager(fs, cmdRunner, 1*time.Millisecond) platform = NewLinuxPlatform( fs, cmdRunner, collector, compressor, copier, dirProvider, vitalsService, cdutil, diskManager, netManager, 1*time.Millisecond, logger, ) }) Describe("SetupDhcp", func() { networks := boshsettings.Networks{ "bosh": boshsettings.Network{ Default: []string{"dns"}, DNS: []string{"xx.xx.xx.xx", "yy.yy.yy.yy", "zz.zz.zz.zz"}, }, "vip": boshsettings.Network{ Default: []string{}, DNS: []string{"aa.aa.aa.aa"}, }, } Context("when dhcp was not previously configured", func() { It("writes dhcp configuration", func() { err := platform.SetupDhcp(networks) Expect(err).ToNot(HaveOccurred()) dhcpConfig := fs.GetFileTestStat("/etc/dhcp/dhclient.conf") Expect(dhcpConfig).ToNot(BeNil()) Expect(dhcpConfig.StringContents()).To(Equal(expectedCentosDHCPConfig)) }) It("restarts network", func() { err := platform.SetupDhcp(networks) Expect(err).ToNot(HaveOccurred()) Expect(len(cmdRunner.RunCommands)).To(Equal(1)) Expect(cmdRunner.RunCommands[0]).To(Equal([]string{"service", "network", "restart"})) }) }) Context("when dhcp was previously configured with different configuration", func() { BeforeEach(func() { fs.WriteFileString("/etc/dhcp/dhclient.conf", "fake-other-configuration") }) It("updates dhcp configuration", func() { err := platform.SetupDhcp(networks) Expect(err).ToNot(HaveOccurred()) dhcpConfig := fs.GetFileTestStat("/etc/dhcp/dhclient.conf") Expect(dhcpConfig).ToNot(BeNil()) Expect(dhcpConfig.StringContents()).To(Equal(expectedCentosDHCPConfig)) }) It("restarts network", func() { err := platform.SetupDhcp(networks) Expect(err).ToNot(HaveOccurred()) Expect(len(cmdRunner.RunCommands)).To(Equal(1)) Expect(cmdRunner.RunCommands[0]).To(Equal([]string{"service", "network", "restart"})) }) }) Context("when dhcp was previously configured with same configuration", func() { BeforeEach(func() { fs.WriteFileString("/etc/dhcp/dhclient.conf", expectedCentosDHCPConfig) }) It("keeps dhcp configuration", func() { err := platform.SetupDhcp(networks) Expect(err).ToNot(HaveOccurred()) dhcpConfig := fs.GetFileTestStat("/etc/dhcp/dhclient.conf") Expect(dhcpConfig).ToNot(BeNil()) Expect(dhcpConfig.StringContents()).To(Equal(expectedCentosDHCPConfig)) }) It("does not restart network", func() { err := platform.SetupDhcp(networks) Expect(err).ToNot(HaveOccurred()) Expect(len(cmdRunner.RunCommands)).To(Equal(0)) }) }) }) Describe("SetupManualNetworking", func() { networks := boshsettings.Networks{ "bosh": boshsettings.Network{ Default: []string{"dns", "gateway"}, IP: "192.168.195.6", Netmask: "255.255.255.0", Gateway: "192.168.195.1", Mac: "22:00:0a:1f:ac:2a", DNS: []string{"10.80.130.2", "10.80.130.1"}, }, } BeforeEach(func() { fs.WriteFile("/sys/class/net/eth0", []byte{}) fs.WriteFileString("/sys/class/net/eth0/address", "22:00:0a:1f:ac:2a\n") fs.SetGlob("/sys/class/net/*", []string{"/sys/class/net/eth0"}) }) It("sets up centos expected ifconfig", func() { err := platform.SetupManualNetworking(networks) Expect(err).ToNot(HaveOccurred()) networkConfig := fs.GetFileTestStat("/etc/sysconfig/network-scripts/ifcfg-eth0") Expect(networkConfig).ToNot(BeNil()) Expect(networkConfig.StringContents()).To(Equal(expectedCentosIfcfg)) }) It("sets up centos /etc/resolv.conf", func() { err := platform.SetupManualNetworking(networks) Expect(err).ToNot(HaveOccurred()) resolvConf := fs.GetFileTestStat("/etc/resolv.conf") Expect(resolvConf).ToNot(BeNil()) Expect(resolvConf.StringContents()).To(Equal(expectedCentosResolvConf)) }) It("restarts networking", func() { err := platform.SetupManualNetworking(networks) Expect(err).ToNot(HaveOccurred()) fs.GetFileTestStat("/etc/sysconfig/network-scripts/ifcfg-eth0") fs.GetFileTestStat("/etc/resolv.conf") time.Sleep(100 * time.Millisecond) Expect(len(cmdRunner.RunCommands)).To(Equal(7)) Expect(cmdRunner.RunCommands[0]).To(Equal([]string{"service", "network", "restart"})) }) It("runs arping commands", func() { err := platform.SetupManualNetworking(networks) Expect(err).ToNot(HaveOccurred()) fs.GetFileTestStat("/etc/sysconfig/network-scripts/ifcfg-eth0") fs.GetFileTestStat("/etc/resolv.conf") time.Sleep(100 * time.Millisecond) Expect(cmdRunner.RunCommands[1]).To(Equal([]string{"arping", "-c", "1", "-U", "-I", "eth0", "192.168.195.6"})) Expect(cmdRunner.RunCommands[6]).To(Equal([]string{"arping", "-c", "1", "-U", "-I", "eth0", "192.168.195.6"})) }) }) }) }
cdutil = fakecd.NewFakeCdUtil() compressor = boshcmd.NewTarballCompressor(cmdRunner, fs) copier = boshcmd.NewCpCopier(cmdRunner, fs) vitalsService = boshvitals.NewService(collector, dirProvider) fs.SetGlob("/sys/bus/scsi/devices/*:0:0:0/block/*", []string{ "/sys/bus/scsi/devices/0:0:0:0/block/sr0", "/sys/bus/scsi/devices/6:0:0:0/block/sdd", "/sys/bus/scsi/devices/fake-host-id:0:0:0/block/sda", }) fs.SetGlob("/sys/bus/scsi/devices/fake-host-id:0:fake-disk-id:0/block/*", []string{"/sys/bus/scsi/devices/fake-host-id:0:fake-disk-id:0/block/sdf"}) logger = boshlog.NewLogger(boshlog.LevelNone) }) JustBeforeEach(func() { netManager := boshnet.NewCentosNetManager(fs, cmdRunner, 1*time.Millisecond) platform = NewLinuxPlatform( fs, cmdRunner, collector, compressor, copier, dirProvider, vitalsService, cdutil, diskManager, netManager, sleepInterval, logger, )
func NewProvider(logger boshlog.Logger, dirProvider boshdirs.DirectoriesProvider, options ProviderOptions) (p provider) { runner := boshsys.NewExecCmdRunner(logger) fs := boshsys.NewOsFileSystem(logger) linuxDiskManager := boshdisk.NewLinuxDiskManager(logger, runner, fs, options.Linux.BindMountPersistentDisk) udev := boshudev.NewConcreteUdevDevice(runner) linuxCdrom := boshcdrom.NewLinuxCdrom("/dev/sr0", udev, runner) linuxCdutil := boshcd.NewCdUtil(dirProvider.SettingsDir(), fs, linuxCdrom) compressor := boshcmd.NewTarballCompressor(runner, fs) copier := boshcmd.NewCpCopier(runner, fs, logger) sigarCollector := boshstats.NewSigarStatsCollector() vitalsService := boshvitals.NewService(sigarCollector, dirProvider) routesSearcher := boshnet.NewCmdRoutesSearcher(runner) defaultNetworkResolver := boshnet.NewDefaultNetworkResolver( routesSearcher, boshnet.DefaultInterfaceToAddrsFunc, ) centosNetManager := boshnet.NewCentosNetManager(fs, runner, defaultNetworkResolver, 10*time.Second, logger) ubuntuNetManager := boshnet.NewUbuntuNetManager(fs, runner, defaultNetworkResolver, 10*time.Second, logger) centos := NewLinuxPlatform( fs, runner, sigarCollector, compressor, copier, dirProvider, vitalsService, linuxCdutil, linuxDiskManager, centosNetManager, 500*time.Millisecond, options.Linux, logger, ) ubuntu := NewLinuxPlatform( fs, runner, sigarCollector, compressor, copier, dirProvider, vitalsService, linuxCdutil, linuxDiskManager, ubuntuNetManager, 500*time.Millisecond, options.Linux, logger, ) p.platforms = map[string]Platform{ "ubuntu": ubuntu, "centos": centos, "dummy": NewDummyPlatform(sigarCollector, fs, runner, dirProvider, linuxDiskManager, logger), } return }
func init() { Describe("Testing with Ginkgo", func() { var ( collector *fakestats.FakeStatsCollector fs *fakesys.FakeFileSystem cmdRunner *fakesys.FakeCmdRunner diskManager fakedisk.FakeDiskManager dirProvider boshdirs.DirectoriesProvider platform Platform cdutil *fakecd.FakeCdUtil compressor boshcmd.Compressor copier boshcmd.Copier vitalsService boshvitals.Service ) BeforeEach(func() { collector = &fakestats.FakeStatsCollector{} fs = &fakesys.FakeFileSystem{} cmdRunner = &fakesys.FakeCmdRunner{} diskManager = fakedisk.NewFakeDiskManager(cmdRunner) dirProvider = boshdirs.NewDirectoriesProvider("/fake-dir") cdutil = fakecd.NewFakeCdUtil() compressor = boshcmd.NewTarballCompressor(cmdRunner, fs) copier = boshcmd.NewCpCopier(cmdRunner, fs) vitalsService = boshvitals.NewService(collector, dirProvider) }) JustBeforeEach(func() { netManager := boshnet.NewCentosNetManager(fs, cmdRunner, 1*time.Millisecond) platform = NewLinuxPlatform( fs, cmdRunner, collector, compressor, copier, dirProvider, vitalsService, cdutil, diskManager, 1*time.Millisecond, netManager, ) }) It("centos setup runtime configuration", func() { err := platform.SetupRuntimeConfiguration() assert.NoError(GinkgoT(), err) assert.Equal(GinkgoT(), 1, len(cmdRunner.RunCommands)) assert.Equal(GinkgoT(), []string{"bosh-agent-rc"}, cmdRunner.RunCommands[0]) }) It("centos create user", func() { expectedUseradd := []string{ "useradd", "-m", "-b", "/some/path/to/home", "-s", "/bin/bash", "-p", "bar-pwd", "foo-user", } password := "******" err := platform.CreateUser("foo-user", password, "/some/path/to/home") assert.NoError(GinkgoT(), err) basePathStat := fs.GetFileTestStat("/some/path/to/home") assert.Equal(GinkgoT(), fakesys.FakeFileTypeDir, basePathStat.FileType) assert.Equal(GinkgoT(), os.FileMode(0755), basePathStat.FileMode) assert.Equal(GinkgoT(), 1, len(cmdRunner.RunCommands)) assert.Equal(GinkgoT(), expectedUseradd, cmdRunner.RunCommands[0]) }) It("centos create user with an empty password", func() { expectedUseradd := []string{ "useradd", "-m", "-b", "/some/path/to/home", "-s", "/bin/bash", "foo-user", } password := "" err := platform.CreateUser("foo-user", password, "/some/path/to/home") assert.NoError(GinkgoT(), err) basePathStat := fs.GetFileTestStat("/some/path/to/home") assert.Equal(GinkgoT(), fakesys.FakeFileTypeDir, basePathStat.FileType) assert.Equal(GinkgoT(), os.FileMode(0755), basePathStat.FileMode) assert.Equal(GinkgoT(), 1, len(cmdRunner.RunCommands)) assert.Equal(GinkgoT(), expectedUseradd, cmdRunner.RunCommands[0]) }) It("centos add user to groups", func() { err := platform.AddUserToGroups("foo-user", []string{"group1", "group2", "group3"}) assert.NoError(GinkgoT(), err) assert.Equal(GinkgoT(), 1, len(cmdRunner.RunCommands)) usermod := []string{"usermod", "-G", "group1,group2,group3", "foo-user"} assert.Equal(GinkgoT(), usermod, cmdRunner.RunCommands[0]) }) It("centos delete users with prefix and regex", func() { passwdFile := fmt.Sprintf(`%sfoo:... %sbar:... foo:... bar:... foobar:... %sfoobar:...`, boshsettings.EPHEMERAL_USER_PREFIX, boshsettings.EPHEMERAL_USER_PREFIX, boshsettings.EPHEMERAL_USER_PREFIX, ) fs.WriteToFile("/etc/passwd", passwdFile) err := platform.DeleteEphemeralUsersMatching("bar$") assert.NoError(GinkgoT(), err) assert.Equal(GinkgoT(), 2, len(cmdRunner.RunCommands)) assert.Equal(GinkgoT(), []string{"userdel", "-r", "bosh_bar"}, cmdRunner.RunCommands[0]) assert.Equal(GinkgoT(), []string{"userdel", "-r", "bosh_foobar"}, cmdRunner.RunCommands[1]) }) It("centos setup ssh", func() { fs.HomeDirHomePath = "/some/home/dir" platform.SetupSsh("some public key", "vcap") sshDirPath := "/some/home/dir/.ssh" sshDirStat := fs.GetFileTestStat(sshDirPath) assert.Equal(GinkgoT(), fs.HomeDirUsername, "vcap") assert.NotNil(GinkgoT(), sshDirStat) assert.Equal(GinkgoT(), fakesys.FakeFileTypeDir, sshDirStat.FileType) assert.Equal(GinkgoT(), sshDirStat.FileMode, os.FileMode(0700)) assert.Equal(GinkgoT(), sshDirStat.Username, "vcap") authKeysStat := fs.GetFileTestStat(filepath.Join(sshDirPath, "authorized_keys")) assert.NotNil(GinkgoT(), authKeysStat) assert.Equal(GinkgoT(), authKeysStat.FileType, fakesys.FakeFileTypeFile) assert.Equal(GinkgoT(), authKeysStat.FileMode, os.FileMode(0600)) assert.Equal(GinkgoT(), authKeysStat.Username, "vcap") assert.Equal(GinkgoT(), authKeysStat.Content, "some public key") }) It("centos set user password", func() { platform.SetUserPassword("my-user", "my-encrypted-password") assert.Equal(GinkgoT(), 1, len(cmdRunner.RunCommands)) assert.Equal(GinkgoT(), []string{"usermod", "-p", "my-encrypted-password", "my-user"}, cmdRunner.RunCommands[0]) }) It("centos setup hostname", func() { platform.SetupHostname("foobar.local") assert.Equal(GinkgoT(), 1, len(cmdRunner.RunCommands)) assert.Equal(GinkgoT(), []string{"hostname", "foobar.local"}, cmdRunner.RunCommands[0]) hostnameFileContent, err := fs.ReadFile("/etc/hostname") assert.NoError(GinkgoT(), err) assert.Equal(GinkgoT(), "foobar.local", hostnameFileContent) hostsFileContent, err := fs.ReadFile("/etc/hosts") assert.NoError(GinkgoT(), err) assert.Equal(GinkgoT(), CENTOS_EXPECTED_ETC_HOSTS, hostsFileContent) }) It("centos setup dhcp", func() { networks := boshsettings.Networks{ "bosh": boshsettings.Network{ Default: []string{"dns"}, Dns: []string{"xx.xx.xx.xx", "yy.yy.yy.yy", "zz.zz.zz.zz"}, }, "vip": boshsettings.Network{ Default: []string{}, Dns: []string{"aa.aa.aa.aa"}, }, } platform.SetupDhcp(networks) dhcpConfig := fs.GetFileTestStat("/etc/dhcp/dhclient.conf") assert.NotNil(GinkgoT(), dhcpConfig) assert.Equal(GinkgoT(), dhcpConfig.Content, CENTOS_EXPECTED_DHCP_CONFIG) assert.Equal(GinkgoT(), len(cmdRunner.RunCommands), 1) assert.Equal(GinkgoT(), cmdRunner.RunCommands[0], []string{"service", "network", "restart"}) }) It("centos setup dhcp with pre existing configuration", func() { fs.WriteToFile("/etc/dhcp/dhclient.conf", CENTOS_EXPECTED_DHCP_CONFIG) networks := boshsettings.Networks{ "bosh": boshsettings.Network{ Default: []string{"dns"}, Dns: []string{"xx.xx.xx.xx", "yy.yy.yy.yy", "zz.zz.zz.zz"}, }, "vip": boshsettings.Network{ Default: []string{}, Dns: []string{"aa.aa.aa.aa"}, }, } platform.SetupDhcp(networks) dhcpConfig := fs.GetFileTestStat("/etc/dhcp/dhclient.conf") assert.NotNil(GinkgoT(), dhcpConfig) assert.Equal(GinkgoT(), dhcpConfig.Content, CENTOS_EXPECTED_DHCP_CONFIG) assert.Equal(GinkgoT(), len(cmdRunner.RunCommands), 0) }) It("centos setup manual networking", func() { networks := boshsettings.Networks{ "bosh": boshsettings.Network{ Default: []string{"dns", "gateway"}, Ip: "192.168.195.6", Netmask: "255.255.255.0", Gateway: "192.168.195.1", Mac: "22:00:0a:1f:ac:2a", Dns: []string{"10.80.130.2", "10.80.130.1"}, }, } fs.WriteToFile("/sys/class/net/eth0", "") fs.WriteToFile("/sys/class/net/eth0/address", "22:00:0a:1f:ac:2a\n") fs.GlobPaths = []string{"/sys/class/net/eth0"} platform.SetupManualNetworking(networks) networkConfig := fs.GetFileTestStat("/etc/sysconfig/network-scripts/ifcfg-eth0") assert.NotNil(GinkgoT(), networkConfig) assert.Equal(GinkgoT(), networkConfig.Content, CENTOS_EXPECTED_IFCFG) resolvConf := fs.GetFileTestStat("/etc/resolv.conf") assert.NotNil(GinkgoT(), resolvConf) assert.Equal(GinkgoT(), resolvConf.Content, CENTOS_EXPECTED_RESOLV_CONF) time.Sleep(100 * time.Millisecond) assert.Equal(GinkgoT(), len(cmdRunner.RunCommands), 7) assert.Equal(GinkgoT(), cmdRunner.RunCommands[0], []string{"service", "network", "restart"}) assert.Equal(GinkgoT(), cmdRunner.RunCommands[1], []string{"arping", "-c", "1", "-U", "-I", "eth0", "192.168.195.6"}) assert.Equal(GinkgoT(), cmdRunner.RunCommands[6], []string{"arping", "-c", "1", "-U", "-I", "eth0", "192.168.195.6"}) }) It("centos setup logrotate", func() { platform.SetupLogrotate("fake-group-name", "fake-base-path", "fake-size") logrotateFileContent, err := fs.ReadFile("/etc/logrotate.d/fake-group-name") assert.NoError(GinkgoT(), err) assert.Equal(GinkgoT(), CENTOS_EXPECTED_ETC_LOGROTATE, logrotateFileContent) }) It("centos set time with ntp servers", func() { platform.SetTimeWithNtpServers([]string{"0.north-america.pool.ntp.org", "1.north-america.pool.ntp.org"}) ntpConfig := fs.GetFileTestStat("/fake-dir/bosh/etc/ntpserver") assert.Equal(GinkgoT(), "0.north-america.pool.ntp.org 1.north-america.pool.ntp.org", ntpConfig.Content) assert.Equal(GinkgoT(), fakesys.FakeFileTypeFile, ntpConfig.FileType) assert.Equal(GinkgoT(), 1, len(cmdRunner.RunCommands)) assert.Equal(GinkgoT(), []string{"ntpdate"}, cmdRunner.RunCommands[0]) }) It("centos set time with ntp servers is noop when no ntp server provided", func() { platform.SetTimeWithNtpServers([]string{}) assert.Equal(GinkgoT(), 0, len(cmdRunner.RunCommands)) ntpConfig := fs.GetFileTestStat("/fake-dir/bosh/etc/ntpserver") assert.Nil(GinkgoT(), ntpConfig) }) It("centos setup ephemeral disk with path", func() { fakeFormatter := diskManager.FakeFormatter fakePartitioner := diskManager.FakePartitioner fakeMounter := diskManager.FakeMounter fakePartitioner.GetDeviceSizeInMbSizes = map[string]uint64{"/dev/xvda": uint64(1024 * 1024 * 1024)} fs.WriteToFile("/dev/xvda", "") err := platform.SetupEphemeralDiskWithPath("/dev/xvda") assert.NoError(GinkgoT(), err) dataDir := fs.GetFileTestStat("/fake-dir/data") assert.Equal(GinkgoT(), fakesys.FakeFileTypeDir, dataDir.FileType) assert.Equal(GinkgoT(), os.FileMode(0750), dataDir.FileMode) assert.Equal(GinkgoT(), "/dev/xvda", fakePartitioner.PartitionDevicePath) assert.Equal(GinkgoT(), 2, len(fakePartitioner.PartitionPartitions)) swapPartition := fakePartitioner.PartitionPartitions[0] ext4Partition := fakePartitioner.PartitionPartitions[1] assert.Equal(GinkgoT(), "swap", swapPartition.Type) assert.Equal(GinkgoT(), "linux", ext4Partition.Type) assert.Equal(GinkgoT(), 2, len(fakeFormatter.FormatPartitionPaths)) assert.Equal(GinkgoT(), "/dev/xvda1", fakeFormatter.FormatPartitionPaths[0]) assert.Equal(GinkgoT(), "/dev/xvda2", fakeFormatter.FormatPartitionPaths[1]) assert.Equal(GinkgoT(), 2, len(fakeFormatter.FormatFsTypes)) assert.Equal(GinkgoT(), boshdisk.FileSystemSwap, fakeFormatter.FormatFsTypes[0]) assert.Equal(GinkgoT(), boshdisk.FileSystemExt4, fakeFormatter.FormatFsTypes[1]) assert.Equal(GinkgoT(), 1, len(fakeMounter.MountMountPoints)) assert.Equal(GinkgoT(), "/fake-dir/data", fakeMounter.MountMountPoints[0]) assert.Equal(GinkgoT(), 1, len(fakeMounter.MountPartitionPaths)) assert.Equal(GinkgoT(), "/dev/xvda2", fakeMounter.MountPartitionPaths[0]) assert.Equal(GinkgoT(), 1, len(fakeMounter.SwapOnPartitionPaths)) assert.Equal(GinkgoT(), "/dev/xvda1", fakeMounter.SwapOnPartitionPaths[0]) sysLogStats := fs.GetFileTestStat("/fake-dir/data/sys/log") assert.NotNil(GinkgoT(), sysLogStats) assert.Equal(GinkgoT(), fakesys.FakeFileTypeDir, sysLogStats.FileType) assert.Equal(GinkgoT(), os.FileMode(0750), sysLogStats.FileMode) assert.Equal(GinkgoT(), []string{"chown", "root:vcap", "/fake-dir/data/sys"}, cmdRunner.RunCommands[0]) assert.Equal(GinkgoT(), []string{"chown", "root:vcap", "/fake-dir/data/sys/log"}, cmdRunner.RunCommands[1]) sysRunStats := fs.GetFileTestStat("/fake-dir/data/sys/run") assert.NotNil(GinkgoT(), sysRunStats) assert.Equal(GinkgoT(), fakesys.FakeFileTypeDir, sysRunStats.FileType) assert.Equal(GinkgoT(), os.FileMode(0750), sysRunStats.FileMode) assert.Equal(GinkgoT(), []string{"chown", "root:vcap", "/fake-dir/data/sys/run"}, cmdRunner.RunCommands[2]) }) It("centos setup tmp dir", func() { err := platform.SetupTmpDir() assert.NoError(GinkgoT(), err) assert.Equal(GinkgoT(), 2, len(cmdRunner.RunCommands)) assert.Equal(GinkgoT(), []string{"chown", "root:vcap", "/tmp"}, cmdRunner.RunCommands[0]) assert.Equal(GinkgoT(), []string{"chmod", "0770", "/tmp"}, cmdRunner.RunCommands[1]) }) It("centos mount persistent disk", func() { fakeFormatter := diskManager.FakeFormatter fakePartitioner := diskManager.FakePartitioner fakeMounter := diskManager.FakeMounter fs.WriteToFile("/dev/vdf", "") err := platform.MountPersistentDisk("/dev/sdf", "/mnt/point") assert.NoError(GinkgoT(), err) mountPoint := fs.GetFileTestStat("/mnt/point") assert.Equal(GinkgoT(), fakesys.FakeFileTypeDir, mountPoint.FileType) assert.Equal(GinkgoT(), os.FileMode(0700), mountPoint.FileMode) partition := fakePartitioner.PartitionPartitions[0] assert.Equal(GinkgoT(), "/dev/vdf", fakePartitioner.PartitionDevicePath) assert.Equal(GinkgoT(), 1, len(fakePartitioner.PartitionPartitions)) assert.Equal(GinkgoT(), "linux", partition.Type) assert.Equal(GinkgoT(), 1, len(fakeFormatter.FormatPartitionPaths)) assert.Equal(GinkgoT(), "/dev/vdf1", fakeFormatter.FormatPartitionPaths[0]) assert.Equal(GinkgoT(), 1, len(fakeFormatter.FormatFsTypes)) assert.Equal(GinkgoT(), boshdisk.FileSystemExt4, fakeFormatter.FormatFsTypes[0]) assert.Equal(GinkgoT(), 1, len(fakeMounter.MountMountPoints)) assert.Equal(GinkgoT(), "/mnt/point", fakeMounter.MountMountPoints[0]) assert.Equal(GinkgoT(), 1, len(fakeMounter.MountPartitionPaths)) assert.Equal(GinkgoT(), "/dev/vdf1", fakeMounter.MountPartitionPaths[0]) }) Context("when not mounted", func() { It("does not unmount persistent disk ", func() { fakeMounter := diskManager.FakeMounter fakeMounter.UnmountDidUnmount = false fs.WriteToFile("/dev/vdx", "") didUnmount, err := platform.UnmountPersistentDisk("/dev/sdx") assert.NoError(GinkgoT(), err) assert.Equal(GinkgoT(), didUnmount, false) assert.Equal(GinkgoT(), "/dev/vdx1", fakeMounter.UnmountPartitionPath) }) }) Context("when already mounted", func() { It("unmounts persistent disk ", func() { fakeMounter := diskManager.FakeMounter fakeMounter.UnmountDidUnmount = true fs.WriteToFile("/dev/vdx", "") didUnmount, err := platform.UnmountPersistentDisk("/dev/sdx") assert.NoError(GinkgoT(), err) assert.Equal(GinkgoT(), didUnmount, true) assert.Equal(GinkgoT(), "/dev/vdx1", fakeMounter.UnmountPartitionPath) }) }) It("centos get real device path with multiple possible devices", func() { fs.WriteToFile("/dev/xvda", "") fs.WriteToFile("/dev/vda", "") realPath, found := platform.NormalizeDiskPath("/dev/sda") assert.True(GinkgoT(), found) assert.Equal(GinkgoT(), "/dev/xvda", realPath) }) It("centos get real device path with delay within timeout", func() { time.AfterFunc(time.Millisecond, func() { fs.WriteToFile("/dev/xvda", "") }) realPath, found := platform.NormalizeDiskPath("/dev/sda") assert.True(GinkgoT(), found) assert.Equal(GinkgoT(), "/dev/xvda", realPath) }) It("centos get real device path with delay beyond timeout", func() { time.AfterFunc(2*time.Second, func() { fs.WriteToFile("/dev/xvda", "") }) _, found := platform.NormalizeDiskPath("/dev/sda") assert.False(GinkgoT(), found) }) It("centos calculate ephemeral disk partition sizes when disk is bigger than twice the memory", func() { totalMemInMb := uint64(1024) diskSizeInMb := totalMemInMb*2 + 64 expectedSwap := totalMemInMb collector.MemStats.Total = totalMemInMb * uint64(1024*1024) fakePartitioner := diskManager.FakePartitioner fakePartitioner.GetDeviceSizeInMbSizes = map[string]uint64{ "/dev/hda": diskSizeInMb, } err := platform.SetupEphemeralDiskWithPath("/dev/hda") assert.NoError(GinkgoT(), err) expectedPartitions := []boshdisk.Partition{ {SizeInMb: expectedSwap, Type: boshdisk.PartitionTypeSwap}, {SizeInMb: diskSizeInMb - expectedSwap, Type: boshdisk.PartitionTypeLinux}, } assert.Equal(GinkgoT(), fakePartitioner.PartitionPartitions, expectedPartitions) }) It("centos calculate ephemeral disk partition sizes when disk twice the memory or smaller", func() { totalMemInMb := uint64(1024) diskSizeInMb := totalMemInMb*2 - 64 expectedSwap := diskSizeInMb / 2 collector.MemStats.Total = totalMemInMb * uint64(1024*1024) fakePartitioner := diskManager.FakePartitioner fakePartitioner.GetDeviceSizeInMbSizes = map[string]uint64{ "/dev/hda": diskSizeInMb, } err := platform.SetupEphemeralDiskWithPath("/dev/hda") assert.NoError(GinkgoT(), err) expectedPartitions := []boshdisk.Partition{ {SizeInMb: expectedSwap, Type: boshdisk.PartitionTypeSwap}, {SizeInMb: diskSizeInMb - expectedSwap, Type: boshdisk.PartitionTypeLinux}, } assert.Equal(GinkgoT(), fakePartitioner.PartitionPartitions, expectedPartitions) }) It("centos migrate persistent disk", func() { fakeMounter := diskManager.FakeMounter platform.MigratePersistentDisk("/from/path", "/to/path") assert.Equal(GinkgoT(), fakeMounter.RemountAsReadonlyPath, "/from/path") assert.Equal(GinkgoT(), 1, len(cmdRunner.RunCommands)) assert.Equal(GinkgoT(), []string{"sh", "-c", "(tar -C /from/path -cf - .) | (tar -C /to/path -xpf -)"}, cmdRunner.RunCommands[0]) assert.Equal(GinkgoT(), fakeMounter.UnmountPartitionPath, "/from/path") assert.Equal(GinkgoT(), fakeMounter.RemountFromMountPoint, "/to/path") assert.Equal(GinkgoT(), fakeMounter.RemountToMountPoint, "/from/path") }) It("centos normalize disk path", func() { fs.WriteToFile("/dev/xvda", "") path, found := platform.NormalizeDiskPath("/dev/sda") assert.Equal(GinkgoT(), path, "/dev/xvda") assert.True(GinkgoT(), found) fs.RemoveAll("/dev/xvda") fs.WriteToFile("/dev/vda", "") path, found = platform.NormalizeDiskPath("/dev/sda") assert.Equal(GinkgoT(), path, "/dev/vda") assert.True(GinkgoT(), found) fs.RemoveAll("/dev/vda") fs.WriteToFile("/dev/sda", "") path, found = platform.NormalizeDiskPath("/dev/sda") assert.Equal(GinkgoT(), path, "/dev/sda") assert.True(GinkgoT(), found) }) Describe("GetFileContentsFromCDROM", func() { It("delegates to cdutil", func() { cdutil.GetFileContentsContents = []byte("fake-contents") filename := "fake-env" contents, err := platform.GetFileContentsFromCDROM(filename) Expect(err).NotTo(HaveOccurred()) Expect(cdutil.GetFileContentsFilename).To(Equal(filename)) Expect(contents).To(Equal(cdutil.GetFileContentsContents)) }) }) It("centos is device path mounted", func() { fs.WriteToFile("/dev/xvda", "") fakeMounter := diskManager.FakeMounter fakeMounter.IsMountedResult = true result, err := platform.IsDevicePathMounted("/dev/sda") assert.NoError(GinkgoT(), err) assert.True(GinkgoT(), result) assert.Equal(GinkgoT(), fakeMounter.IsMountedDevicePathOrMountPoint, "/dev/xvda1") }) It("centos start monit", func() { err := platform.StartMonit() assert.NoError(GinkgoT(), err) assert.Equal(GinkgoT(), 1, len(cmdRunner.RunCommands)) assert.Equal(GinkgoT(), []string{"sv", "up", "monit"}, cmdRunner.RunCommands[0]) }) It("centos setup monit user if file does not exist", func() { err := platform.SetupMonitUser() assert.NoError(GinkgoT(), err) monitUserFileStats := fs.GetFileTestStat("/fake-dir/monit/monit.user") assert.NotNil(GinkgoT(), monitUserFileStats) assert.Equal(GinkgoT(), "vcap:random-password", monitUserFileStats.Content) }) It("centos setup monit user if file does exist", func() { fs.WriteToFile("/fake-dir/monit/monit.user", "vcap:other-random-password") err := platform.SetupMonitUser() assert.NoError(GinkgoT(), err) monitUserFileStats := fs.GetFileTestStat("/fake-dir/monit/monit.user") assert.NotNil(GinkgoT(), monitUserFileStats) assert.Equal(GinkgoT(), "vcap:other-random-password", monitUserFileStats.Content) }) It("centos get monit credentials reads monit file from disk", func() { fs.WriteToFile("/fake-dir/monit/monit.user", "fake-user:fake-random-password") username, password, err := platform.GetMonitCredentials() assert.NoError(GinkgoT(), err) assert.Equal(GinkgoT(), "fake-user", username) assert.Equal(GinkgoT(), "fake-random-password", password) }) It("centos get monit credentials errs when invalid file format", func() { fs.WriteToFile("/fake-dir/monit/monit.user", "fake-user") _, _, err := platform.GetMonitCredentials() assert.Error(GinkgoT(), err) }) It("centos get monit credentials leaves colons in password intact", func() { fs.WriteToFile("/fake-dir/monit/monit.user", "fake-user:fake:random:password") username, password, err := platform.GetMonitCredentials() assert.NoError(GinkgoT(), err) assert.Equal(GinkgoT(), "fake-user", username) assert.Equal(GinkgoT(), "fake:random:password", password) }) }) }