mirror of
https://github.com/cozystack/cozystack.git
synced 2026-11-03 10:37:21 +09:00
Aleksei Sviridkin
d07dcb0b78
The GPU reconciler always runs, so GPU finalizers and usage records are handled on every cluster, including ones that later turn GPU off; --enable-gpu adds placement and the KubeVirt watches. GPU admission webhooks are registered only with it, because asking the manager for its webhook server starts it and the server needs a serving certificate. The hoster's resource-name list and raw-name policy come in as flags and are written to the settings ConfigMap for the admission policies to read. GPU admission has to answer from warm caches, so with GPU enabled every replica starts the informers it needs before the manager rather than at the first admission, and readiness waits for exactly those to sync. It does not wait for KubeVirt, whose kinds a fresh install serves only after this controller is ready; those informers start once the kinds are served. The leader releases its lease on shutdown so a rollout does not leave starting VMs gated for a lease duration. The ConfigMap cache holds only the settings object. Non-positive grace periods are refused, since a zero attachment grace would clear every attachment before its VMI appears. Assisted-by: LLM Signed-off-by: Aleksei Sviridkin <f@lex.la>
75 lines
2.9 KiB
Go
75 lines
2.9 KiB
Go
/*
|
|
Copyright 2026 The Cozystack Authors.
|
|
|
|
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 main
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"sigs.k8s.io/controller-runtime/pkg/client/apiutil"
|
|
|
|
gpuv1alpha1 "github.com/cozystack/cozystack/api/gpu/v1alpha1"
|
|
internalv1alpha1 "github.com/cozystack/cozystack/api/internalapi/v1alpha1"
|
|
"github.com/cozystack/cozystack/internal/gpu"
|
|
)
|
|
|
|
// TestSchemeRegistersReconciledKinds pins that the manager scheme init() builds
|
|
// resolves the kinds the CA-extraction and GPU controllers reconcile. A kind
|
|
// missing here is not a build error — it fails at runtime: SetupWithManager
|
|
// cannot resolve the GVK for its field index or its For() source, returns an
|
|
// error, and main() calls os.Exit(1), taking the whole binary (and every other
|
|
// controller) down with it. The scheme is the package-level var init() populates.
|
|
func TestSchemeRegistersReconciledKinds(t *testing.T) {
|
|
for name, obj := range map[string]runtime.Object{
|
|
"TenantProjection": &internalv1alpha1.TenantProjection{},
|
|
"TenantProjectionList": &internalv1alpha1.TenantProjectionList{},
|
|
"GPUClass": &gpuv1alpha1.GPUClass{},
|
|
"GPUClaimList": &gpuv1alpha1.GPUClaimList{},
|
|
"GPUNode": &gpuv1alpha1.GPUNode{},
|
|
} {
|
|
if _, err := apiutil.GVKForObject(obj, scheme); err != nil {
|
|
t.Errorf("manager scheme does not register %s, so a controller that reconciles it cannot start: %v", name, err)
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestValidateGPUConfig(t *testing.T) {
|
|
valid := gpu.Config{LostCapacityGrace: time.Minute, AttachmentGrace: time.Minute}
|
|
for name, tc := range map[string]struct {
|
|
edit func(*gpu.Config)
|
|
ok bool
|
|
raw gpu.RawNamesPolicy
|
|
}{
|
|
"empty policy means allowed": {func(*gpu.Config) {}, true, gpu.RawNamesAllowed},
|
|
"forbidden": {func(c *gpu.Config) { c.RawResourceNames = gpu.RawNamesForbidden }, true, gpu.RawNamesForbidden},
|
|
"unknown policy": {func(c *gpu.Config) { c.RawResourceNames = "maybe" }, false, ""},
|
|
"zero lost-capacity grace": {func(c *gpu.Config) { c.LostCapacityGrace = 0 }, false, ""},
|
|
"negative attachment grace": {func(c *gpu.Config) { c.AttachmentGrace = -time.Second }, false, ""},
|
|
} {
|
|
cfg := valid
|
|
tc.edit(&cfg)
|
|
err := validateGPUConfig(&cfg)
|
|
if (err == nil) != tc.ok {
|
|
t.Errorf("%s: err = %v, want ok %v", name, err, tc.ok)
|
|
}
|
|
if tc.ok && cfg.RawResourceNames != tc.raw {
|
|
t.Errorf("%s: policy = %q, want %q", name, cfg.RawResourceNames, tc.raw)
|
|
}
|
|
}
|
|
}
|