Files
my-rook-config/pkg/operator/ceph/cluster/osd/spec_test.go
T
Sébastien Han 087d71106b ceph: add a new CR property to force upgrades
On rare occasion, the ceph checks might not be 100% correct and the user
might decide to force an upgrade anyway.
Correctly, if we are in an upgrade, we perform `ok-to-stop` check for
each daemon, if one of them fails, we retry and eventually fail.
They are corner-cases where users would like to force the upgrade anyway.

Enhance the new CR property: "skipUpgradeChecks: true".

Closes: https://github.com/rook/rook/issues/3872
Signed-off-by: Sébastien Han <seb@redhat.com>
2019-09-19 10:55:56 +02:00

383 lines
14 KiB
Go

/*
Copyright 2016 The Rook Authors. All rights reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
*/
// Package osd for the Ceph OSDs.
package osd
import (
"testing"
cephv1 "github.com/rook/rook/pkg/apis/ceph.rook.io/v1"
rookalpha "github.com/rook/rook/pkg/apis/rook.io/v1alpha2"
"github.com/rook/rook/pkg/clusterd"
cephconfig "github.com/rook/rook/pkg/daemon/ceph/config"
"github.com/rook/rook/pkg/operator/ceph/cluster/osd/config"
cephver "github.com/rook/rook/pkg/operator/ceph/version"
"github.com/rook/rook/pkg/operator/k8sutil"
exectest "github.com/rook/rook/pkg/util/exec/test"
v1 "k8s.io/api/core/v1"
"k8s.io/apimachinery/pkg/api/resource"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/client-go/kubernetes/fake"
"github.com/stretchr/testify/assert"
"github.com/stretchr/testify/require"
)
func TestPodContainer(t *testing.T) {
cluster := &Cluster{Namespace: "myosd", rookVersion: "23", cephVersion: cephv1.CephVersionSpec{}, clusterInfo: &cephconfig.ClusterInfo{}}
osdProps := osdProperties{
crushHostname: "node",
devices: []rookalpha.Device{},
resources: v1.ResourceRequirements{},
storeConfig: config.StoreConfig{},
}
c, err := cluster.provisionPodTemplateSpec(osdProps, v1.RestartPolicyAlways)
assert.NotNil(t, c)
assert.Nil(t, err)
assert.Equal(t, 1, len(c.Spec.InitContainers))
assert.Equal(t, 1, len(c.Spec.Containers))
container := c.Spec.InitContainers[0]
logger.Infof("container: %+v", container)
assert.Equal(t, "copy-binaries", container.Args[0])
container = c.Spec.Containers[0]
assert.Equal(t, "/rook/tini", container.Command[0])
assert.Equal(t, "--", container.Args[0])
assert.Equal(t, "/rook/rook", container.Args[1])
assert.Equal(t, "ceph", container.Args[2])
assert.Equal(t, "osd", container.Args[3])
assert.Equal(t, "provision", container.Args[4])
}
func TestDaemonset(t *testing.T) {
testPodDevices(t, "", "sda", true)
testPodDevices(t, "/var/lib/mydatadir", "sdb", false)
testPodDevices(t, "", "", true)
testPodDevices(t, "", "", false)
}
func testPodDevices(t *testing.T, dataDir, deviceName string, allDevices bool) {
storageSpec := rookalpha.StorageScopeSpec{
Selection: rookalpha.Selection{UseAllDevices: &allDevices, DeviceFilter: deviceName},
Nodes: []rookalpha.Node{{Name: "node1"}},
}
devices := []rookalpha.Device{
{Name: deviceName},
}
clientset := fake.NewSimpleClientset()
cephVersion := cephv1.CephVersionSpec{Image: "ceph/ceph:v12.2.8"}
clusterInfo := &cephconfig.ClusterInfo{
CephVersion: cephver.Nautilus,
}
c := New(clusterInfo, &clusterd.Context{Clientset: clientset, ConfigDir: "/var/lib/rook", Executor: &exectest.MockExecutor{}}, "ns", "rook/rook:myversion", cephVersion,
storageSpec, dataDir, rookalpha.Placement{}, rookalpha.Annotations{}, cephv1.NetworkSpec{}, v1.ResourceRequirements{}, metav1.OwnerReference{}, false, false)
devMountNeeded := deviceName != "" || allDevices
n := c.DesiredStorage.ResolveNode(storageSpec.Nodes[0].Name)
if len(devices) == 0 && len(dataDir) == 0 {
return
}
osd := OSDInfo{
ID: 0,
}
osdProp := osdProperties{
crushHostname: n.Name,
selection: n.Selection,
resources: v1.ResourceRequirements{},
storeConfig: config.StoreConfig{},
location: n.Location,
}
deployment, err := c.makeDeployment(osdProp, osd)
assert.Nil(t, err)
assert.NotNil(t, deployment)
assert.Equal(t, "rook-ceph-osd-0", deployment.Name)
assert.Equal(t, c.Namespace, deployment.Namespace)
assert.Equal(t, serviceAccountName, deployment.Spec.Template.Spec.ServiceAccountName)
assert.Equal(t, int32(1), *(deployment.Spec.Replicas))
assert.Equal(t, "node1", deployment.Spec.Template.Spec.NodeSelector[v1.LabelHostname])
assert.Equal(t, v1.RestartPolicyAlways, deployment.Spec.Template.Spec.RestartPolicy)
if devMountNeeded && len(dataDir) > 0 {
assert.Equal(t, 5, len(deployment.Spec.Template.Spec.Volumes))
}
if devMountNeeded && len(dataDir) == 0 {
assert.Equal(t, 5, len(deployment.Spec.Template.Spec.Volumes))
}
if !devMountNeeded && len(dataDir) > 0 {
assert.Equal(t, 1, len(deployment.Spec.Template.Spec.Volumes))
}
assert.Equal(t, "rook-data", deployment.Spec.Template.Spec.Volumes[0].Name)
assert.Equal(t, AppName, deployment.Spec.Template.ObjectMeta.Name)
assert.Equal(t, AppName, deployment.Spec.Template.ObjectMeta.Labels["app"])
assert.Equal(t, c.Namespace, deployment.Spec.Template.ObjectMeta.Labels["rook_cluster"])
assert.Equal(t, 0, len(deployment.Spec.Template.ObjectMeta.Annotations))
assert.Equal(t, 2, len(deployment.Spec.Template.Spec.InitContainers))
initCont := deployment.Spec.Template.Spec.InitContainers[0]
assert.Equal(t, "rook/rook:myversion", initCont.Image)
assert.Equal(t, "config-init", initCont.Name)
assert.Equal(t, 3, len(initCont.VolumeMounts))
assert.Equal(t, 1, len(deployment.Spec.Template.Spec.Containers))
cont := deployment.Spec.Template.Spec.Containers[0]
assert.Equal(t, cephVersion.Image, cont.Image)
assert.Equal(t, 5, len(cont.VolumeMounts))
assert.Equal(t, "ceph-osd", cont.Command[0])
}
func verifyEnvVar(t *testing.T, envVars []v1.EnvVar, expectedName, expectedValue string, expectedFound bool) {
found := false
for _, envVar := range envVars {
if envVar.Name == expectedName {
assert.Equal(t, expectedValue, envVar.Value)
found = true
break
}
}
assert.Equal(t, expectedFound, found)
}
func TestStorageSpecDevicesAndDirectories(t *testing.T) {
storageSpec := rookalpha.StorageScopeSpec{
Selection: rookalpha.Selection{
Directories: []rookalpha.Directory{{Path: "/rook/dir2"}},
},
Nodes: []rookalpha.Node{
{
Name: "node1",
Selection: rookalpha.Selection{
Devices: []rookalpha.Device{{Name: "sda"}},
Directories: []rookalpha.Directory{{Path: "/rook/dir1"}},
},
},
},
}
clientset := fake.NewSimpleClientset()
clusterInfo := &cephconfig.ClusterInfo{
CephVersion: cephver.Nautilus,
}
c := New(clusterInfo, &clusterd.Context{Clientset: clientset, ConfigDir: "/var/lib/rook", Executor: &exectest.MockExecutor{}}, "ns", "rook/rook:myversion", cephv1.CephVersionSpec{},
storageSpec, "/var/lib/rook", rookalpha.Placement{}, rookalpha.Annotations{}, cephv1.NetworkSpec{}, v1.ResourceRequirements{}, metav1.OwnerReference{}, false, false)
n := c.DesiredStorage.ResolveNode(storageSpec.Nodes[0].Name)
osd := OSDInfo{
ID: 0,
IsDirectory: true,
DataPath: "/my/root/path/osd1",
}
osdProp := osdProperties{
crushHostname: n.Name,
selection: n.Selection,
resources: v1.ResourceRequirements{},
storeConfig: config.StoreConfig{},
location: n.Location,
}
deployment, err := c.makeDeployment(osdProp, osd)
assert.NotNil(t, deployment)
assert.Nil(t, err)
// pod spec should have a volume for the given dir in the main container and the init container
podSpec := deployment.Spec.Template.Spec
assert.Equal(t, 5, len(podSpec.Volumes))
require.Equal(t, 1, len(podSpec.Containers))
cont := podSpec.Containers[0]
assert.Equal(t, 5, len(cont.VolumeMounts))
assert.Equal(t, "/var/lib/rook", cont.VolumeMounts[0].MountPath)
assert.Equal(t, "/etc/ceph", cont.VolumeMounts[1].MountPath)
assert.Equal(t, "/var/log/ceph", cont.VolumeMounts[2].MountPath)
assert.Equal(t, "/my/root/path", cont.VolumeMounts[3].MountPath)
require.Equal(t, 2, len(podSpec.InitContainers))
initCont := podSpec.InitContainers[0]
assert.Equal(t, 4, len(initCont.VolumeMounts))
assert.Equal(t, "/var/lib/rook", initCont.VolumeMounts[0].MountPath)
assert.Equal(t, "/etc/ceph", initCont.VolumeMounts[1].MountPath)
assert.Equal(t, "/var/log/ceph", initCont.VolumeMounts[2].MountPath)
assert.Equal(t, "/my/root/path", initCont.VolumeMounts[3].MountPath)
// the default osd created on a node will be under /var/lib/rook, which won't need an extra mount
osd = OSDInfo{
ID: 1,
IsDirectory: true,
DataPath: "/var/lib/rook/osd1",
}
deployment, err = c.makeDeployment(osdProp, osd)
assert.NotNil(t, deployment)
assert.Nil(t, err)
// pod spec should have a volume for the given dir in the main container and the init container
podSpec = deployment.Spec.Template.Spec
assert.Equal(t, 4, len(podSpec.Volumes))
require.Equal(t, 1, len(podSpec.Containers))
cont = podSpec.Containers[0]
require.Equal(t, 4, len(cont.VolumeMounts))
assert.Equal(t, "/var/lib/rook", cont.VolumeMounts[0].MountPath)
assert.Equal(t, "/etc/ceph", cont.VolumeMounts[1].MountPath)
assert.Equal(t, (7 + len(k8sutil.ClusterDaemonEnvVars(c.cephVersion.Image))), len(cont.Env))
require.Equal(t, 2, len(podSpec.InitContainers))
initCont = podSpec.InitContainers[0]
require.Equal(t, 3, len(initCont.VolumeMounts))
assert.Equal(t, "/var/lib/rook", initCont.VolumeMounts[0].MountPath)
assert.Equal(t, "/etc/ceph", initCont.VolumeMounts[1].MountPath)
assert.Equal(t, "/var/log/ceph", initCont.VolumeMounts[2].MountPath)
}
func TestStorageSpecConfig(t *testing.T) {
storageSpec := rookalpha.StorageScopeSpec{
Nodes: []rookalpha.Node{
{
Name: "node1",
Location: "rack=foo",
Config: map[string]string{
"storeType": "bluestore",
"databaseSizeMB": "10",
"walSizeMB": "20",
"journalSizeMB": "30",
"metadataDevice": "nvme093",
},
Selection: rookalpha.Selection{
Directories: []rookalpha.Directory{{Path: "/rook/storageDir472"}},
},
Resources: v1.ResourceRequirements{
Limits: v1.ResourceList{
v1.ResourceCPU: *resource.NewQuantity(1024.0, resource.BinarySI),
v1.ResourceMemory: *resource.NewQuantity(4096.0, resource.BinarySI),
},
Requests: v1.ResourceList{
v1.ResourceCPU: *resource.NewQuantity(500.0, resource.BinarySI),
v1.ResourceMemory: *resource.NewQuantity(2048.0, resource.BinarySI),
},
},
},
},
}
clientset := fake.NewSimpleClientset()
clusterInfo := &cephconfig.ClusterInfo{
CephVersion: cephver.Nautilus,
}
c := New(clusterInfo, &clusterd.Context{Clientset: clientset, ConfigDir: "/var/lib/rook", Executor: &exectest.MockExecutor{}}, "ns", "rook/rook:myversion", cephv1.CephVersionSpec{},
storageSpec, "", rookalpha.Placement{}, rookalpha.Annotations{}, cephv1.NetworkSpec{}, v1.ResourceRequirements{}, metav1.OwnerReference{}, false, false)
n := c.DesiredStorage.ResolveNode(storageSpec.Nodes[0].Name)
storeConfig := config.ToStoreConfig(storageSpec.Nodes[0].Config)
metadataDevice := config.MetadataDevice(storageSpec.Nodes[0].Config)
osdProp := osdProperties{
crushHostname: n.Name,
devices: n.Devices,
selection: n.Selection,
resources: c.DesiredStorage.Nodes[0].Resources,
storeConfig: storeConfig,
metadataDevice: metadataDevice,
location: n.Location,
}
job, err := c.makeJob(osdProp)
assert.NotNil(t, job)
assert.Nil(t, err)
assert.Equal(t, "rook-ceph-osd-prepare-node1", job.ObjectMeta.Name)
container := job.Spec.Template.Spec.InitContainers[0]
assert.NotNil(t, container)
container = job.Spec.Template.Spec.Containers[0]
assert.NotNil(t, container)
verifyEnvVar(t, container.Env, "ROOK_OSD_STORE", "bluestore", true)
verifyEnvVar(t, container.Env, "ROOK_OSD_DATABASE_SIZE", "10", true)
verifyEnvVar(t, container.Env, "ROOK_OSD_WAL_SIZE", "20", true)
verifyEnvVar(t, container.Env, "ROOK_OSD_JOURNAL_SIZE", "30", true)
verifyEnvVar(t, container.Env, "ROOK_LOCATION", "rack=foo", true)
verifyEnvVar(t, container.Env, "ROOK_METADATA_DEVICE", "nvme093", true)
assert.Equal(t, "1Ki", container.Resources.Limits.Cpu().String(), "limit cpu is: %s", container.Resources.Limits.Cpu().String())
assert.Equal(t, "500", container.Resources.Requests.Cpu().String())
assert.Equal(t, "4Ki", container.Resources.Limits.Memory().String())
assert.Equal(t, "2Ki", container.Resources.Requests.Memory().String())
// verify that osd config can be discovered from the container and matches the original config from the spec
discoveredConfig := getConfigFromContainer(container)
assert.Equal(t, n.Config, discoveredConfig)
discoveredDirs := getDirectoriesFromContainer(container)
assert.Equal(t, n.Directories, discoveredDirs)
}
func TestHostNetwork(t *testing.T) {
storageSpec := rookalpha.StorageScopeSpec{
Nodes: []rookalpha.Node{
{
Name: "node1",
Location: "rack=foo",
Config: map[string]string{
"storeType": "bluestore",
"databaseSizeMB": "10",
"walSizeMB": "20",
"journalSizeMB": "30",
},
},
},
}
clientset := fake.NewSimpleClientset()
clusterInfo := &cephconfig.ClusterInfo{
CephVersion: cephver.Nautilus,
}
c := New(clusterInfo, &clusterd.Context{Clientset: clientset, ConfigDir: "/var/lib/rook", Executor: &exectest.MockExecutor{}}, "ns", "myversion", cephv1.CephVersionSpec{},
storageSpec, "", rookalpha.Placement{}, rookalpha.Annotations{}, cephv1.NetworkSpec{HostNetwork: true}, v1.ResourceRequirements{}, metav1.OwnerReference{}, false, false)
n := c.DesiredStorage.ResolveNode(storageSpec.Nodes[0].Name)
osd := OSDInfo{
ID: 0,
}
osdProp := osdProperties{
crushHostname: n.Name,
devices: n.Devices,
selection: n.Selection,
resources: c.DesiredStorage.Nodes[0].Resources,
storeConfig: config.StoreConfig{},
location: n.Location,
}
r, err := c.makeDeployment(osdProp, osd)
assert.NotNil(t, r)
assert.Nil(t, err)
assert.Equal(t, "rook-ceph-osd-0", r.ObjectMeta.Name)
assert.Equal(t, true, r.Spec.Template.Spec.HostNetwork)
assert.Equal(t, v1.DNSClusterFirstWithHostNet, r.Spec.Template.Spec.DNSPolicy)
}
func TestOsdOnSDNFlag(t *testing.T) {
network := cephv1.NetworkSpec{}
v := cephver.Mimic
args := osdOnSDNFlag(network, v)
assert.Empty(t, args)
v = cephver.CephVersion{Major: 14, Minor: 2, Extra: 2}
args = osdOnSDNFlag(network, v)
assert.NotEmpty(t, args)
v = cephver.Octopus
args = osdOnSDNFlag(network, v)
assert.NotEmpty(t, args)
}