Skip to content

Commit 2177089

Browse files
yaroslavborbatyaroslavborbat
committed
add descheduler e2e
Signed-off-by: Yaroslav Borbat <[email protected]>
1 parent 8009ab9 commit 2177089

File tree

18 files changed

+943
-66
lines changed

18 files changed

+943
-66
lines changed

‎images/virtualization-artifact/pkg/builder/cvi/cvi.go

Lines changed: 50 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,50 @@
1+
/*
2+
Copyright 2025 Flant JSC
3+
4+
Licensed under the Apache License, Version 2.0 (the "License");
5+
you may not use this file except in compliance with the License.
6+
You may obtain a copy of the License at
7+
8+
http://www.apache.org/licenses/LICENSE-2.0
9+
10+
Unless required by applicable law or agreed to in writing, software
11+
distributed under the License is distributed on an "AS IS" BASIS,
12+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
See the License for the specific language governing permissions and
14+
limitations under the License.
15+
*/
16+
17+
package cvi
18+
19+
import (
20+
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
21+
22+
"github.com/deckhouse/virtualization/api/core/v1alpha2"
23+
)
24+
25+
func New(options ...Option) *v1alpha2.ClusterVirtualImage {
26+
cvi := NewEmpty("")
27+
ApplyOptions(cvi, options...)
28+
return cvi
29+
}
30+
31+
func ApplyOptions(cvi *v1alpha2.ClusterVirtualImage, opts ...Option) {
32+
if cvi == nil {
33+
return
34+
}
35+
for _, opt := range opts {
36+
opt(cvi)
37+
}
38+
}
39+
40+
func NewEmpty(name string) *v1alpha2.ClusterVirtualImage {
41+
return &v1alpha2.ClusterVirtualImage{
42+
TypeMeta: metav1.TypeMeta{
43+
APIVersion: v1alpha2.SchemeGroupVersion.String(),
44+
Kind: v1alpha2.ClusterVirtualImageKind,
45+
},
46+
ObjectMeta: metav1.ObjectMeta{
47+
Name: name,
48+
},
49+
}
50+
}

‎images/virtualization-artifact/pkg/builder/cvi/option.go

Lines changed: 93 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,93 @@
1+
/*
2+
Copyright 2025 Flant JSC
3+
4+
Licensed under the Apache License, Version 2.0 (the "License");
5+
you may not use this file except in compliance with the License.
6+
You may obtain a copy of the License at
7+
8+
http://www.apache.org/licenses/LICENSE-2.0
9+
10+
Unless required by applicable law or agreed to in writing, software
11+
distributed under the License is distributed on an "AS IS" BASIS,
12+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
See the License for the specific language governing permissions and
14+
limitations under the License.
15+
*/
16+
17+
package cvi
18+
19+
import (
20+
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
21+
22+
"github.com/deckhouse/virtualization-controller/pkg/builder/meta"
23+
"github.com/deckhouse/virtualization/api/core/v1alpha2"
24+
)
25+
26+
type Option func(cvi *v1alpha2.ClusterVirtualImage)
27+
28+
var (
29+
WithName = meta.WithName[*v1alpha2.ClusterVirtualImage]
30+
WithGenerateName = meta.WithGenerateName[*v1alpha2.ClusterVirtualImage]
31+
WithLabel = meta.WithLabel[*v1alpha2.ClusterVirtualImage]
32+
WithLabels = meta.WithLabels[*v1alpha2.ClusterVirtualImage]
33+
WithAnnotation = meta.WithAnnotation[*v1alpha2.ClusterVirtualImage]
34+
WithAnnotations = meta.WithAnnotations[*v1alpha2.ClusterVirtualImage]
35+
WithFinalizer = meta.WithFinalizer[*v1alpha2.ClusterVirtualImage]
36+
)
37+
38+
func WithDataSourceHTTP(url string, checksum *v1alpha2.Checksum, caBundle []byte) Option {
39+
return func(cvi *v1alpha2.ClusterVirtualImage) {
40+
cvi.Spec.DataSource = v1alpha2.ClusterVirtualImageDataSource{
41+
Type: v1alpha2.DataSourceTypeHTTP,
42+
HTTP: &v1alpha2.DataSourceHTTP{
43+
URL: url,
44+
Checksum: checksum,
45+
CABundle: caBundle,
46+
},
47+
}
48+
}
49+
}
50+
51+
func WithDataSourceContainerImage(image string, imagePullSecret v1alpha2.ImagePullSecret, caBundle []byte) Option {
52+
return func(cvi *v1alpha2.ClusterVirtualImage) {
53+
cvi.Spec.DataSource = v1alpha2.ClusterVirtualImageDataSource{
54+
Type: v1alpha2.DataSourceTypeContainerImage,
55+
ContainerImage: &v1alpha2.ClusterVirtualImageContainerImage{
56+
Image: image,
57+
ImagePullSecret: imagePullSecret,
58+
CABundle: caBundle,
59+
},
60+
}
61+
}
62+
}
63+
64+
func WithDataSourceObjectRef(kind v1alpha2.ClusterVirtualImageObjectRefKind, name, namespace string) Option {
65+
return func(cvi *v1alpha2.ClusterVirtualImage) {
66+
cvi.Spec.DataSource = v1alpha2.ClusterVirtualImageDataSource{
67+
Type: v1alpha2.DataSourceTypeObjectRef,
68+
ObjectRef: &v1alpha2.ClusterVirtualImageObjectRef{
69+
Kind: kind,
70+
Name: name,
71+
Namespace: namespace,
72+
},
73+
}
74+
}
75+
}
76+
77+
func WithDatasource(datasource v1alpha2.ClusterVirtualImageDataSource) func(cvi *v1alpha2.ClusterVirtualImage) {
78+
return func(cvi *v1alpha2.ClusterVirtualImage) {
79+
cvi.Spec.DataSource = datasource
80+
}
81+
}
82+
83+
func WithPhase(phase v1alpha2.ImagePhase) func(cvi *v1alpha2.ClusterVirtualImage) {
84+
return func(cvi *v1alpha2.ClusterVirtualImage) {
85+
cvi.Status.Phase = phase
86+
}
87+
}
88+
89+
func WithCondition(condition metav1.Condition) func(cvi *v1alpha2.ClusterVirtualImage) {
90+
return func(cvi *v1alpha2.ClusterVirtualImage) {
91+
cvi.Status.Conditions = append(cvi.Status.Conditions, condition)
92+
}
93+
}

‎images/virtualization-artifact/pkg/builder/vd/option.go

Lines changed: 112 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,112 @@
1+
/*
2+
Copyright 2025 Flant JSC
3+
4+
Licensed under the Apache License, Version 2.0 (the "License");
5+
you may not use this file except in compliance with the License.
6+
You may obtain a copy of the License at
7+
8+
http://www.apache.org/licenses/LICENSE-2.0
9+
10+
Unless required by applicable law or agreed to in writing, software
11+
distributed under the License is distributed on an "AS IS" BASIS,
12+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
See the License for the specific language governing permissions and
14+
limitations under the License.
15+
*/
16+
17+
package vd
18+
19+
import (
20+
"k8s.io/apimachinery/pkg/api/resource"
21+
"k8s.io/utils/ptr"
22+
23+
"github.com/deckhouse/virtualization-controller/pkg/builder/meta"
24+
"github.com/deckhouse/virtualization/api/core/v1alpha2"
25+
)
26+
27+
type Option func(vmop *v1alpha2.VirtualDisk)
28+
29+
var (
30+
WithName = meta.WithName[*v1alpha2.VirtualDisk]
31+
WithNamespace = meta.WithNamespace[*v1alpha2.VirtualDisk]
32+
WithGenerateName = meta.WithGenerateName[*v1alpha2.VirtualDisk]
33+
WithLabel = meta.WithLabel[*v1alpha2.VirtualDisk]
34+
WithLabels = meta.WithLabels[*v1alpha2.VirtualDisk]
35+
WithAnnotation = meta.WithAnnotation[*v1alpha2.VirtualDisk]
36+
WithAnnotations = meta.WithAnnotations[*v1alpha2.VirtualDisk]
37+
WithFinalizer = meta.WithFinalizer[*v1alpha2.VirtualDisk]
38+
)
39+
40+
func WithDatasource(datasource *v1alpha2.VirtualDiskDataSource) func(vd *v1alpha2.VirtualDisk) {
41+
return func(vd *v1alpha2.VirtualDisk) {
42+
vd.Spec.DataSource = datasource
43+
}
44+
}
45+
func WithDataSourceHTTP(url string, checksum *v1alpha2.Checksum, caBundle []byte) Option {
46+
return func(vd *v1alpha2.VirtualDisk) {
47+
vd.Spec.DataSource = &v1alpha2.VirtualDiskDataSource{
48+
Type: v1alpha2.DataSourceTypeHTTP,
49+
HTTP: &v1alpha2.DataSourceHTTP{
50+
URL: url,
51+
Checksum: checksum,
52+
CABundle: caBundle,
53+
},
54+
}
55+
}
56+
}
57+
58+
func WithDataSourceContainerImage(image, imagePullSecretName string, caBundle []byte) Option {
59+
return func(vd *v1alpha2.VirtualDisk) {
60+
vd.Spec.DataSource = &v1alpha2.VirtualDiskDataSource{
61+
Type: v1alpha2.DataSourceTypeContainerImage,
62+
ContainerImage: &v1alpha2.VirtualDiskContainerImage{
63+
Image: image,
64+
ImagePullSecret: v1alpha2.ImagePullSecretName{
65+
Name: imagePullSecretName,
66+
},
67+
CABundle: caBundle,
68+
},
69+
}
70+
}
71+
}
72+
73+
func WithDataSourceObjectRef(kind v1alpha2.VirtualDiskObjectRefKind, name string) Option {
74+
return func(vd *v1alpha2.VirtualDisk) {
75+
vd.Spec.DataSource = &v1alpha2.VirtualDiskDataSource{
76+
Type: v1alpha2.DataSourceTypeObjectRef,
77+
ObjectRef: &v1alpha2.VirtualDiskObjectRef{
78+
Kind: kind,
79+
Name: name,
80+
},
81+
}
82+
}
83+
}
84+
85+
func WithDataSourceObjectRefFromCVI(cvi *v1alpha2.ClusterVirtualImage) Option {
86+
return WithDataSourceObjectRef(v1alpha2.VirtualDiskObjectRefKindClusterVirtualImage, cvi.Name)
87+
}
88+
89+
func WithDataSourceObjectRefFromVI(vi *v1alpha2.VirtualImage) Option {
90+
return WithDataSourceObjectRef(v1alpha2.VirtualDiskObjectRefKindVirtualImage, vi.Name)
91+
}
92+
93+
func WithPersistentVolumeClaim(storageClass *string, size *resource.Quantity) Option {
94+
return func(vd *v1alpha2.VirtualDisk) {
95+
vd.Spec.PersistentVolumeClaim = v1alpha2.VirtualDiskPersistentVolumeClaim{
96+
StorageClass: storageClass,
97+
Size: size,
98+
}
99+
}
100+
}
101+
102+
func WithStorageClass(storageClass string) Option {
103+
return func(vd *v1alpha2.VirtualDisk) {
104+
vd.Spec.PersistentVolumeClaim.StorageClass = ptr.To(storageClass)
105+
}
106+
}
107+
108+
func WithSize(size *resource.Quantity) Option {
109+
return func(vd *v1alpha2.VirtualDisk) {
110+
vd.Spec.PersistentVolumeClaim.Size = size
111+
}
112+
}

‎images/virtualization-artifact/pkg/builder/vd/vd.go

Lines changed: 51 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,51 @@
1+
/*
2+
Copyright 2025 Flant JSC
3+
4+
Licensed under the Apache License, Version 2.0 (the "License");
5+
you may not use this file except in compliance with the License.
6+
You may obtain a copy of the License at
7+
8+
http://www.apache.org/licenses/LICENSE-2.0
9+
10+
Unless required by applicable law or agreed to in writing, software
11+
distributed under the License is distributed on an "AS IS" BASIS,
12+
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13+
See the License for the specific language governing permissions and
14+
limitations under the License.
15+
*/
16+
17+
package vd
18+
19+
import (
20+
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
21+
22+
"github.com/deckhouse/virtualization/api/core/v1alpha2"
23+
)
24+
25+
func New(options ...Option) *v1alpha2.VirtualDisk {
26+
vd := NewEmpty("", "")
27+
ApplyOptions(vd, options...)
28+
return vd
29+
}
30+
31+
func ApplyOptions(vd *v1alpha2.VirtualDisk, opts ...Option) {
32+
if vd == nil {
33+
return
34+
}
35+
for _, opt := range opts {
36+
opt(vd)
37+
}
38+
}
39+
40+
func NewEmpty(name, namespace string) *v1alpha2.VirtualDisk {
41+
return &v1alpha2.VirtualDisk{
42+
TypeMeta: metav1.TypeMeta{
43+
APIVersion: v1alpha2.SchemeGroupVersion.String(),
44+
Kind: v1alpha2.VirtualDiskKind,
45+
},
46+
ObjectMeta: metav1.ObjectMeta{
47+
Name: name,
48+
Namespace: namespace,
49+
},
50+
}
51+
}

‎images/virtualization-artifact/pkg/builder/vm/option.go

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,3 +49,40 @@ func WithMemory(size resource.Quantity) Option {
4949
vm.Spec.Memory.Size = size
5050
}
5151
}
52+
53+
func WithDisks(disks ...*v1alpha2.VirtualDisk) Option {
54+
return func(vm *v1alpha2.VirtualMachine) {
55+
blockDeviceRefs := make([]v1alpha2.BlockDeviceSpecRef, 0, len(disks))
56+
for _, disk := range disks {
57+
blockDeviceRefs = append(blockDeviceRefs, v1alpha2.BlockDeviceSpecRef{
58+
Kind: v1alpha2.VirtualDiskKind,
59+
Name: disk.Name,
60+
})
61+
}
62+
vm.Spec.BlockDeviceRefs = append(vm.Spec.BlockDeviceRefs, blockDeviceRefs...)
63+
}
64+
}
65+
66+
func WithBlockDeviceRefs(refs ...v1alpha2.BlockDeviceSpecRef) Option {
67+
return func(vm *v1alpha2.VirtualMachine) {
68+
vm.Spec.BlockDeviceRefs = append(vm.Spec.BlockDeviceRefs, refs...)
69+
}
70+
}
71+
72+
func WithNodeSelector(nodeSelector map[string]string) Option {
73+
return func(vm *v1alpha2.VirtualMachine) {
74+
vm.Spec.NodeSelector = nodeSelector
75+
}
76+
}
77+
78+
func WithLiveMigrationPolicy(liveMigrationPolicy v1alpha2.LiveMigrationPolicy) Option {
79+
return func(vm *v1alpha2.VirtualMachine) {
80+
vm.Spec.LiveMigrationPolicy = liveMigrationPolicy
81+
}
82+
}
83+
84+
func WithVirtualMachineClass(class string) Option {
85+
return func(vm *v1alpha2.VirtualMachine) {
86+
vm.Spec.VirtualMachineClassName = class
87+
}
88+
}

‎tests/e2e/config/config.go

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,8 @@ import (
2929

3030
yamlv3 "gopkg.in/yaml.v3"
3131
storagev1 "k8s.io/api/storage/v1"
32+
"k8s.io/cli-runtime/pkg/genericclioptions"
33+
"k8s.io/client-go/rest"
3234

3335
gt "github.com/deckhouse/virtualization/tests/e2e/git"
3436
kc "github.com/deckhouse/virtualization/tests/e2e/kubectl"
@@ -170,6 +172,26 @@ type ClusterTransport struct {
170172
InsecureTls bool `yaml:"insecureTls"`
171173
}
172174

175+
func (c ClusterTransport) RestConfig() (*rest.Config, error) {
176+
configFlags := genericclioptions.ConfigFlags{}
177+
if c.KubeConfig != "" {
178+
configFlags.KubeConfig = &c.KubeConfig
179+
}
180+
if c.Token != "" {
181+
configFlags.BearerToken = &c.Token
182+
}
183+
if c.InsecureTls {
184+
configFlags.Insecure = &c.InsecureTls
185+
}
186+
if c.CertificateAuthority != "" {
187+
configFlags.CAFile = &c.CertificateAuthority
188+
}
189+
if c.Endpoint != "" {
190+
configFlags.APIServer = &c.Endpoint
191+
}
192+
return configFlags.ToRESTConfig()
193+
}
194+
173195
type DisksConf struct {
174196
UploadHelperImage string `yaml:"uploadHelperImage"`
175197
CviTestDataDir string `yaml:"cviTestDataDir"`

0 commit comments

Comments
 (0)