mirror of
https://github.com/hanzoai/deploy.git
synced 2026-10-07 08:44:50 +00:00
zeekayandHanzo Dev
1ab6e799c1
go.mod still declared `module github.com/hanzoai/deploy` after the repo was renamed. A Go module path is not cosmetic — it is the resolvable location. So `go get github.com/hanzoai/cd` could not work, and anything importing this module had to spell a name that no longer exists at that address. The build log of v3.6.0 is where it surfaced: the compiler was still emitting github.com/hanzoai/deploy/cmd/cd/commands. 3541 import paths across 690 .go, 19 .proto, 4 .yaml, 3 .sh, 2 .mod, Procfile and Makefile. Plus the 62 ghcr.io/hanzoai/deploy image references, moved to ghcr.io/hanzoai/cd to match hanzo.yml's build target — a distinct string from the module path, replaced separately so neither could clobber the other. go mod tidy clean, CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build ./... clean. Co-authored-by: Hanzo Dev <[email protected]>
172 lines
4.8 KiB
Go
172 lines
4.8 KiB
Go
package kubetest
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
|
"k8s.io/apimachinery/pkg/runtime/schema"
|
|
"k8s.io/apimachinery/pkg/watch"
|
|
"k8s.io/client-go/dynamic"
|
|
"k8s.io/client-go/rest"
|
|
cmdutil "k8s.io/kubectl/pkg/cmd/util"
|
|
|
|
"github.com/hanzoai/cd/gitops-engine/pkg/utils/kube"
|
|
)
|
|
|
|
type MockResourceOps struct {
|
|
ExecuteForDryRun bool
|
|
Commands map[string]KubectlOutput
|
|
Events chan watch.Event
|
|
DynamicClient dynamic.Interface
|
|
|
|
lastCommandPerResource map[kube.ResourceKey]string
|
|
lastValidate bool
|
|
serverSideApply bool
|
|
serverSideApplyManager string
|
|
lastForce bool
|
|
|
|
recordLock sync.RWMutex
|
|
|
|
getResourceFunc *func(ctx context.Context, config *rest.Config, gvk schema.GroupVersionKind, name string, namespace string) (*unstructured.Unstructured, error)
|
|
}
|
|
|
|
// WithGetResourceFunc overrides the default ConvertToVersion behavior.
|
|
func (r *MockResourceOps) WithGetResourceFunc(getResourcefunc func(context.Context, *rest.Config, schema.GroupVersionKind, string, string) (*unstructured.Unstructured, error)) *MockResourceOps {
|
|
r.getResourceFunc = &getResourcefunc
|
|
return r
|
|
}
|
|
|
|
func (r *MockResourceOps) SetLastValidate(validate bool) {
|
|
r.recordLock.Lock()
|
|
r.lastValidate = validate
|
|
r.recordLock.Unlock()
|
|
}
|
|
|
|
func (r *MockResourceOps) GetLastValidate() bool {
|
|
r.recordLock.RLock()
|
|
validate := r.lastValidate
|
|
r.recordLock.RUnlock()
|
|
return validate
|
|
}
|
|
|
|
func (r *MockResourceOps) SetLastServerSideApply(serverSideApply bool) {
|
|
r.recordLock.Lock()
|
|
r.serverSideApply = serverSideApply
|
|
r.recordLock.Unlock()
|
|
}
|
|
|
|
func (r *MockResourceOps) GetLastServerSideApplyManager() string {
|
|
r.recordLock.Lock()
|
|
manager := r.serverSideApplyManager
|
|
r.recordLock.Unlock()
|
|
return manager
|
|
}
|
|
|
|
func (r *MockResourceOps) GetLastServerSideApply() bool {
|
|
r.recordLock.RLock()
|
|
serverSideApply := r.serverSideApply
|
|
r.recordLock.RUnlock()
|
|
return serverSideApply
|
|
}
|
|
|
|
func (r *MockResourceOps) SetLastServerSideApplyManager(manager string) {
|
|
r.recordLock.Lock()
|
|
r.serverSideApplyManager = manager
|
|
r.recordLock.Unlock()
|
|
}
|
|
|
|
func (r *MockResourceOps) SetLastForce(force bool) {
|
|
r.recordLock.Lock()
|
|
r.lastForce = force
|
|
r.recordLock.Unlock()
|
|
}
|
|
|
|
func (r *MockResourceOps) GetLastForce() bool {
|
|
r.recordLock.RLock()
|
|
force := r.lastForce
|
|
r.recordLock.RUnlock()
|
|
return force
|
|
}
|
|
|
|
func (r *MockResourceOps) SetLastResourceCommand(key kube.ResourceKey, cmd string) {
|
|
r.recordLock.Lock()
|
|
if r.lastCommandPerResource == nil {
|
|
r.lastCommandPerResource = map[kube.ResourceKey]string{}
|
|
}
|
|
r.lastCommandPerResource[key] = cmd
|
|
r.recordLock.Unlock()
|
|
}
|
|
|
|
func (r *MockResourceOps) GetLastResourceCommand(key kube.ResourceKey) string {
|
|
r.recordLock.Lock()
|
|
defer r.recordLock.Unlock()
|
|
if r.lastCommandPerResource == nil {
|
|
return ""
|
|
}
|
|
return r.lastCommandPerResource[key]
|
|
}
|
|
|
|
func (r *MockResourceOps) ApplyResource(_ context.Context, obj *unstructured.Unstructured, dryRun cmdutil.DryRunStrategy, force, validate, serverSideApply bool, manager string) (string, error) {
|
|
if dryRun != cmdutil.DryRunNone && !r.ExecuteForDryRun {
|
|
return "", nil
|
|
}
|
|
r.SetLastValidate(validate)
|
|
r.SetLastServerSideApply(serverSideApply)
|
|
r.SetLastServerSideApplyManager(manager)
|
|
r.SetLastForce(force)
|
|
r.SetLastResourceCommand(kube.GetResourceKey(obj), "apply")
|
|
command, ok := r.Commands[obj.GetName()]
|
|
if !ok {
|
|
return "", nil
|
|
}
|
|
|
|
return command.Output, command.Err
|
|
}
|
|
|
|
func (r *MockResourceOps) ReplaceResource(_ context.Context, obj *unstructured.Unstructured, dryRun cmdutil.DryRunStrategy, force bool) (string, error) {
|
|
if dryRun != cmdutil.DryRunNone && !r.ExecuteForDryRun {
|
|
return "", nil
|
|
}
|
|
r.SetLastForce(force)
|
|
command, ok := r.Commands[obj.GetName()]
|
|
r.SetLastResourceCommand(kube.GetResourceKey(obj), "replace")
|
|
if !ok {
|
|
return "", nil
|
|
}
|
|
|
|
return command.Output, command.Err
|
|
}
|
|
|
|
func (r *MockResourceOps) UpdateResource(_ context.Context, obj *unstructured.Unstructured, dryRun cmdutil.DryRunStrategy) (*unstructured.Unstructured, error) {
|
|
if dryRun != cmdutil.DryRunNone && !r.ExecuteForDryRun {
|
|
return obj, nil
|
|
}
|
|
r.SetLastResourceCommand(kube.GetResourceKey(obj), "update")
|
|
command, ok := r.Commands[obj.GetName()]
|
|
if !ok {
|
|
return obj, nil
|
|
}
|
|
return obj, command.Err
|
|
}
|
|
|
|
func (r *MockResourceOps) CreateResource(_ context.Context, obj *unstructured.Unstructured, dryRun cmdutil.DryRunStrategy, _ bool) (string, error) {
|
|
if dryRun != cmdutil.DryRunNone && !r.ExecuteForDryRun {
|
|
return "", nil
|
|
}
|
|
r.SetLastResourceCommand(kube.GetResourceKey(obj), "create")
|
|
command, ok := r.Commands[obj.GetName()]
|
|
if !ok {
|
|
return "", nil
|
|
}
|
|
return command.Output, command.Err
|
|
}
|
|
|
|
/*func (r *MockResourceOps) ConvertToVersion(obj *unstructured.Unstructured, group, version string) (*unstructured.Unstructured, error) {
|
|
if r.convertToVersionFunc != nil {
|
|
return (*r.convertToVersionFunc)(obj, group, version)
|
|
}
|
|
|
|
return obj, nil
|
|
}*/
|