mirror of
https://github.com/hanzoai/deploy.git
synced 2026-10-07 08:44:50 +00:00
41536738c8
* feat(api): hard-fork CRD API group argoproj.io -> deploy.hanzo.ai
Hanzo CD hard-fork, phase 1.1. Single source of truth is
pkg/apis/application/register.go's Group constant (Application/AppProject/
ApplicationSet all derive their FullName from it); every other Go call-site
that hardcoded the literal group string is updated to match: the 3 generated
informer GVRs, the app-of-apps nested-Application detector in types.go, the
resource-tree Application-kind cache populator, the 3 NotFound error
constructors in server/application, and the applicationset-controller's
leader-election Lease name/ID (kept in lockstep across Go source + RBAC
resourceNames, or leader election would get RBAC-denied on startup).
CRD YAMLs (application/appproject/applicationset) and every RBAC apiGroups
rule that actually governs our 3 CRDs are repointed to deploy.hanzo.ai, across
both manifests/base/* (namespaced Roles) and manifests/cluster-rbac/* (cluster
scope). Two adjacent things that share the "argoproj.io" string but are NOT
ours are deliberately left untouched: gitops-engine's health-check dispatch
for Argo Workflows' `Workflow` kind, and argocd-server-clusterrole's second
apiGroups block granting read on Argo Workflows' `workflows` resource — both
belong to a separate, unrelated Argoproj-family project we are not forking.
Also untouched, per explicit scope: annotation/label keys (argocd.argoproj.io/*)
and finalizers (*.argoproj.io) that track already-applied live resources, and
argocd-cm.yaml's resource.customizations.ignoreResourceUpdates keys (a narrow
app-of-apps diffing nicety, not build- or RBAC-relevant).
CGO_ENABLED=0 go build ./... verified green.
* ci: wire canonical hanzo.yml + cicd.yml (Hanzo CD image build)
Phase 1.3. Follows the same one-way pattern every hanzo/lux/zoo repo uses:
hanzo.yml declares the image (root Dockerfile -> ghcr.io/hanzoai/deploy) and a
fast go-build sanity test (the multi-stage Dockerfile build is the slow, real
gate); cicd.yml is the ~10-line trigger shim importing hanzoai/ci's reusable
build workflow on the arc pool, secrets from KMS only.
No `deploy:` block yet, matching the hanzoai/cms build-only-first-cutover
pattern: this repo's own live hanzo-cd namespace runs the whole Hanzo fleet's
GitOps, so the prod image/CRD swap is a deliberate separate action (see the
[HELD] universe PR), never an automatic rollout on merge. Every PR/push still
builds and pushes an immutable ghcr.io/hanzoai/deploy:sha-<7>-amd64 image.
* feat(ui): rebrand Argo CD -> Hanzo CD (title, favicon, logo, sidebar)
Phase 1.2. User-facing "Argo CD"/"ArgoCD" text replaced with "Hanzo CD"
across the login page, help modal, settings screens, version-info panel,
sync-options warning, fullscreen-logs title, app.tsx page titles, and
index.html's <title>/noscript text (dropped the now-wrong upstream CLI
doc link rather than invent a hanzo.ai URL that doesn't exist).
Logo/favicon assets replaced with the canonical Hanzo mark (never
hand-drawn, copied from ~/work/hanzo/logo/dist): the two real usages of
Argo's logo collapse onto one Hanzo glyph (mono + white variants) at the
same 5 call sites (login icon + footer, login.scss $logoPath/
$logoNegativePath, sidebar text-logo), preserving the existing CSS
invert() filters so both the light login panel and dark sidebar still
render correctly. Old argologo.svg/argo_o.svg/argo.png deleted (no
longer referenced). favicon-{16x16,32x32}.png added at
ui/src/assets/favicon/ (index.html's links to that path were previously
dead - the directory didn't exist in this checkout). package.json name
argo-cd-ui -> hanzo-cd-ui.
Left alone, deliberately: the two Kubernetes resource-kind icon sets
under ui/src/assets/images/resources/ (unrelated to branding), and
upstream doc/project links (argoproj.io, argo-cd.readthedocs.io) since
we have no equivalent hanzo.ai docs to point to yet - except the login
footer logo's outbound link, which now points to hanzo.ai instead of
argoproj.io (a Hanzo-branded mark linking to the upstream project it
was forked from read as an obvious leftover).
* feat(module): rename Go module github.com/argoproj/argo-cd/v3 -> github.com/hanzoai/deploy
Phase 2. No version suffix on the new path (org policy: never publish our
own Go modules above v1.x.x / with a v2+ suffix) — this fork starts a fresh
v0/v1 module identity rather than carrying the v3 the upstream fork point
had reached.
Mechanical, symmetric rewrite: 704 .go files (import statements only; two
disjoint substitutions applied specific-first so the nested gitops-engine
module path never double-matches), plus go.mod (module line, the
gitops-engine require+replace pair, version reset to v0.0.0-... since a
suffix-less path can't carry a v3 semver), gitops-engine/go.mod, Makefile
(ldflags PACKAGE + 2 TEST_MODULE filters), .golangci.yaml, .goreleaser.yaml,
.mockery.yaml (+ gitops-engine's), Procfile, the codegen/proto generation
scripts (hack/generate-proto.sh, update-codegen.sh, update-openapi.sh), and
20 .proto files' go_package options. Also one OpenTelemetry tracer name
string in gitops-engine/pkg/sync/sync_context.go (a cosmestic label, not an
import, but worth keeping consistent).
CGO_ENABLED=0 go build ./... verified green for both the root module and
the nested gitops-engine module; go vet clean on both; go mod tidy run
(network available) to reconcile go.sum, dependency set unchanged.
Left alone, deliberately: doc-only mentions (docs/snyk/*.html scan reports,
gitops-engine/README.md), a comment-only reference in
.github/workflows/release.yaml, and one Renovate bot config line — none are
build- or codegen-load-bearing. Also left alone: comments citing upstream
GitHub issue URLs (github.com/argoproj/argo-cd/issues/NNNNN) and a handful
of test fixtures using "https://github.com/argoproj/argo-cd.git" as a
representative example repo URL — both are accurate historical/test-data
references, not stale self-references.
---------
Co-authored-by: Hanzo Dev <[email protected]>
581 lines
18 KiB
Go
581 lines
18 KiB
Go
package db
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
"google.golang.org/grpc/codes"
|
|
"google.golang.org/grpc/status"
|
|
appsv1 "k8s.io/api/apps/v1"
|
|
corev1 "k8s.io/api/core/v1"
|
|
"k8s.io/apimachinery/pkg/api/errors"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/client-go/kubernetes/fake"
|
|
|
|
"github.com/hanzoai/deploy/common"
|
|
"github.com/hanzoai/deploy/pkg/apis/application/v1alpha1"
|
|
"github.com/hanzoai/deploy/util/settings"
|
|
)
|
|
|
|
const (
|
|
testNamespace = "default"
|
|
)
|
|
|
|
func getClientset(objects ...runtime.Object) *fake.Clientset {
|
|
secret := corev1.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "argocd-secret",
|
|
Namespace: testNamespace,
|
|
},
|
|
Data: map[string][]byte{
|
|
"admin.password": []byte("test"),
|
|
"server.secretkey": []byte("test"),
|
|
},
|
|
}
|
|
cm := corev1.ConfigMap{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: "argocd-cm",
|
|
Namespace: testNamespace,
|
|
Labels: map[string]string{
|
|
"app.kubernetes.io/part-of": "argocd",
|
|
},
|
|
},
|
|
}
|
|
return fake.NewClientset(append(objects, &cm, &secret)...)
|
|
}
|
|
|
|
func TestCreateRepository(t *testing.T) {
|
|
clientset := getClientset()
|
|
db := NewDB(testNamespace, settings.NewSettingsManager(t.Context(), clientset, testNamespace), clientset)
|
|
|
|
repo, err := db.CreateRepository(t.Context(), &v1alpha1.Repository{
|
|
Repo: "https://github.com/argoproj/argocd-example-apps",
|
|
Username: "test-username",
|
|
Password: "test-password",
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "https://github.com/argoproj/argocd-example-apps", repo.Repo)
|
|
|
|
secret, err := clientset.CoreV1().Secrets(testNamespace).Get(t.Context(), RepoURLToSecretName(repoSecretPrefix, repo.Repo, ""), metav1.GetOptions{})
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, common.AnnotationValueManagedByArgoCD, secret.Annotations[common.AnnotationKeyManagedBy])
|
|
assert.Equal(t, "test-username", string(secret.Data[username]))
|
|
assert.Equal(t, "test-password", string(secret.Data[password]))
|
|
assert.Empty(t, secret.Data[sshPrivateKey])
|
|
}
|
|
|
|
func TestCreateProjectScopedRepository(t *testing.T) {
|
|
clientset := getClientset()
|
|
db := NewDB(testNamespace, settings.NewSettingsManager(t.Context(), clientset, testNamespace), clientset)
|
|
|
|
repo, err := db.CreateRepository(t.Context(), &v1alpha1.Repository{
|
|
Repo: "https://github.com/argoproj/argocd-example-apps",
|
|
Username: "test-username",
|
|
Password: "test-password",
|
|
Project: "test-project",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
otherRepo, err := db.CreateRepository(t.Context(), &v1alpha1.Repository{
|
|
Repo: "https://github.com/argoproj/argocd-example-apps",
|
|
Username: "other-username",
|
|
Password: "other-password",
|
|
Project: "other-project",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
_, err = db.CreateRepository(t.Context(), &v1alpha1.Repository{
|
|
Repo: "https://github.com/argoproj/argocd-example-apps",
|
|
Username: "wrong-username",
|
|
Password: "wrong-password",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, "https://github.com/argoproj/argocd-example-apps", repo.Repo)
|
|
|
|
secret, err := clientset.CoreV1().Secrets(testNamespace).Get(t.Context(), RepoURLToSecretName(repoSecretPrefix, repo.Repo, "test-project"), metav1.GetOptions{})
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, common.AnnotationValueManagedByArgoCD, secret.Annotations[common.AnnotationKeyManagedBy])
|
|
assert.Equal(t, "test-username", string(secret.Data[username]))
|
|
assert.Equal(t, "test-password", string(secret.Data[password]))
|
|
assert.Equal(t, "test-project", string(secret.Data[project]))
|
|
assert.Empty(t, secret.Data[sshPrivateKey])
|
|
|
|
secret, err = clientset.CoreV1().Secrets(testNamespace).Get(t.Context(), RepoURLToSecretName(repoSecretPrefix, otherRepo.Repo, "other-project"), metav1.GetOptions{})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, common.AnnotationValueManagedByArgoCD, secret.Annotations[common.AnnotationKeyManagedBy])
|
|
assert.Equal(t, "other-username", string(secret.Data[username]))
|
|
assert.Equal(t, "other-password", string(secret.Data[password]))
|
|
assert.Equal(t, "other-project", string(secret.Data[project]))
|
|
assert.Empty(t, secret.Data[sshPrivateKey])
|
|
}
|
|
|
|
func TestCreateRepoCredentials(t *testing.T) {
|
|
clientset := getClientset()
|
|
db := NewDB(testNamespace, settings.NewSettingsManager(t.Context(), clientset, testNamespace), clientset)
|
|
|
|
creds, err := db.CreateRepositoryCredentials(t.Context(), &v1alpha1.RepoCreds{
|
|
URL: "https://github.com/argoproj/",
|
|
Username: "test-username",
|
|
Password: "test-password",
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "https://github.com/argoproj/", creds.URL)
|
|
|
|
secret, err := clientset.CoreV1().Secrets(testNamespace).Get(t.Context(), RepoURLToSecretName(credSecretPrefix, creds.URL, ""), metav1.GetOptions{})
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, common.AnnotationValueManagedByArgoCD, secret.Annotations[common.AnnotationKeyManagedBy])
|
|
assert.Equal(t, "test-username", string(secret.Data[username]))
|
|
assert.Equal(t, "test-password", string(secret.Data[password]))
|
|
assert.Empty(t, secret.Data[sshPrivateKey])
|
|
|
|
created, err := db.CreateRepository(t.Context(), &v1alpha1.Repository{
|
|
Repo: "https://github.com/argoproj/argo-cd",
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "https://github.com/argoproj/argo-cd", created.Repo)
|
|
|
|
// There seems to be a race or some other hiccup in the fake K8s clientset used for this test.
|
|
// Just give it a little time to settle.
|
|
time.Sleep(1 * time.Second)
|
|
|
|
repo, err := db.GetRepository(t.Context(), created.Repo, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "test-username", repo.Username)
|
|
assert.Equal(t, "test-password", repo.Password)
|
|
}
|
|
|
|
func TestCreateWriteRepoCredentials(t *testing.T) {
|
|
clientset := getClientset()
|
|
db := NewDB(testNamespace, settings.NewSettingsManager(t.Context(), clientset, testNamespace), clientset)
|
|
|
|
creds, err := db.CreateWriteRepositoryCredentials(t.Context(), &v1alpha1.RepoCreds{
|
|
URL: "https://github.com/argoproj/",
|
|
Username: "test-username",
|
|
Password: "test-password",
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "https://github.com/argoproj/", creds.URL)
|
|
|
|
secret, err := clientset.CoreV1().Secrets(testNamespace).Get(t.Context(), RepoURLToSecretName(credWriteSecretPrefix, creds.URL, ""), metav1.GetOptions{})
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, common.AnnotationValueManagedByArgoCD, secret.Annotations[common.AnnotationKeyManagedBy])
|
|
assert.Equal(t, "test-username", string(secret.Data[username]))
|
|
assert.Equal(t, "test-password", string(secret.Data[password]))
|
|
assert.Empty(t, secret.Data[sshPrivateKey])
|
|
|
|
created, err := db.CreateWriteRepository(t.Context(), &v1alpha1.Repository{
|
|
Repo: "https://github.com/argoproj/argo-cd",
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "https://github.com/argoproj/argo-cd", created.Repo)
|
|
|
|
repo, err := db.GetWriteRepository(t.Context(), created.Repo, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "test-username", repo.Username)
|
|
assert.Equal(t, "test-password", repo.Password)
|
|
}
|
|
|
|
func TestGetRepositoryCredentials(t *testing.T) {
|
|
clientset := getClientset()
|
|
db := NewDB(testNamespace, settings.NewSettingsManager(t.Context(), clientset, testNamespace), clientset)
|
|
_, err := db.CreateRepositoryCredentials(t.Context(), &v1alpha1.RepoCreds{
|
|
URL: "https://secured",
|
|
Username: "test-username",
|
|
Password: "test-password",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
tests := []struct {
|
|
name string
|
|
repoURL string
|
|
want *v1alpha1.RepoCreds
|
|
}{
|
|
{
|
|
name: "TestUnknownRepo",
|
|
repoURL: "https://unknown/repo",
|
|
want: nil,
|
|
},
|
|
{
|
|
name: "TestKnownRepo",
|
|
repoURL: "https://known/repo",
|
|
want: nil,
|
|
},
|
|
{
|
|
name: "TestSecuredRepo",
|
|
repoURL: "https://secured/repo",
|
|
want: &v1alpha1.RepoCreds{URL: "https://secured", Username: "test-username", Password: "test-password"},
|
|
},
|
|
{
|
|
name: "TestMissingRepo",
|
|
repoURL: "https://missing/repo",
|
|
want: nil,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := db.GetRepositoryCredentials(t.Context(), tt.repoURL)
|
|
|
|
require.NoError(t, err)
|
|
assert.Equal(t, tt.want, got)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestGetWriteRepositoryCredentials(t *testing.T) {
|
|
clientset := getClientset()
|
|
db := NewDB(testNamespace, settings.NewSettingsManager(t.Context(), clientset, testNamespace), clientset)
|
|
_, err := db.CreateWriteRepositoryCredentials(t.Context(), &v1alpha1.RepoCreds{
|
|
URL: "https://secured",
|
|
Username: "test-username",
|
|
Password: "test-password",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
tests := []struct {
|
|
name string
|
|
repoURL string
|
|
want *v1alpha1.RepoCreds
|
|
}{
|
|
{
|
|
name: "TestUnknownRepo",
|
|
repoURL: "https://unknown/repo",
|
|
want: nil,
|
|
},
|
|
{
|
|
name: "TestKnownRepo",
|
|
repoURL: "https://known/repo",
|
|
want: nil,
|
|
},
|
|
{
|
|
name: "TestSecuredRepo",
|
|
repoURL: "https://secured/repo",
|
|
want: &v1alpha1.RepoCreds{URL: "https://secured", Username: "test-username", Password: "test-password"},
|
|
},
|
|
{
|
|
name: "TestMissingRepo",
|
|
repoURL: "https://missing/repo",
|
|
want: nil,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := db.GetWriteRepositoryCredentials(t.Context(), tt.repoURL)
|
|
|
|
require.NoError(t, err)
|
|
assert.Equal(t, tt.want, got)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestCreateExistingRepository(t *testing.T) {
|
|
clientset := getClientset()
|
|
db := NewDB(testNamespace, settings.NewSettingsManager(t.Context(), clientset, testNamespace), clientset)
|
|
|
|
_, err := db.CreateRepository(t.Context(), &v1alpha1.Repository{
|
|
Repo: "https://github.com/argoproj/argocd-example-apps",
|
|
Username: "test-username",
|
|
Password: "test-password",
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
_, err = db.CreateRepository(t.Context(), &v1alpha1.Repository{
|
|
Repo: "https://github.com/argoproj/argocd-example-apps",
|
|
Username: "test-username",
|
|
Password: "test-password",
|
|
})
|
|
require.Error(t, err)
|
|
assert.Equal(t, codes.AlreadyExists, status.Convert(err).Code())
|
|
}
|
|
|
|
func TestGetRepository(t *testing.T) {
|
|
clientset := getClientset(&corev1.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Namespace: testNamespace,
|
|
Name: "known-repo-secret",
|
|
Annotations: map[string]string{
|
|
common.AnnotationKeyManagedBy: common.AnnotationValueManagedByArgoCD,
|
|
},
|
|
Labels: map[string]string{
|
|
common.LabelKeySecretType: common.LabelValueSecretTypeRepository,
|
|
},
|
|
},
|
|
Data: map[string][]byte{
|
|
"url": []byte("https://known/repo"),
|
|
},
|
|
}, &corev1.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Namespace: testNamespace,
|
|
Name: "secured-repo-secret",
|
|
Annotations: map[string]string{
|
|
common.AnnotationKeyManagedBy: common.AnnotationValueManagedByArgoCD,
|
|
},
|
|
Labels: map[string]string{
|
|
common.LabelKeySecretType: common.LabelValueSecretTypeRepository,
|
|
},
|
|
},
|
|
Data: map[string][]byte{
|
|
"url": []byte("https://secured/repo"),
|
|
},
|
|
}, &corev1.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Namespace: testNamespace,
|
|
Name: "secured-repo-creds-secret",
|
|
Annotations: map[string]string{
|
|
common.AnnotationKeyManagedBy: common.AnnotationValueManagedByArgoCD,
|
|
},
|
|
Labels: map[string]string{
|
|
common.LabelKeySecretType: common.LabelValueSecretTypeRepoCreds,
|
|
},
|
|
},
|
|
Data: map[string][]byte{
|
|
"url": []byte("https://secured"),
|
|
"username": []byte("test-username"),
|
|
"password": []byte("test-password"),
|
|
},
|
|
})
|
|
db := NewDB(testNamespace, settings.NewSettingsManager(t.Context(), clientset, testNamespace), clientset)
|
|
|
|
tests := []struct {
|
|
name string
|
|
repoURL string
|
|
want *v1alpha1.Repository
|
|
}{
|
|
{
|
|
name: "TestUnknownRepo",
|
|
repoURL: "https://unknown/repo",
|
|
want: &v1alpha1.Repository{Repo: "https://unknown/repo"},
|
|
},
|
|
{
|
|
name: "TestKnownRepo",
|
|
repoURL: "https://known/repo",
|
|
want: &v1alpha1.Repository{Repo: "https://known/repo"},
|
|
},
|
|
{
|
|
name: "TestSecuredRepo",
|
|
repoURL: "https://secured/repo",
|
|
want: &v1alpha1.Repository{Repo: "https://secured/repo", Username: "test-username", Password: "test-password", InheritedCreds: true},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := db.GetRepository(t.Context(), tt.repoURL, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, tt.want, got)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestGetWriteRepository(t *testing.T) {
|
|
clientset := getClientset(&corev1.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Namespace: testNamespace,
|
|
Name: "known-repo-secret",
|
|
Annotations: map[string]string{
|
|
common.AnnotationKeyManagedBy: common.AnnotationValueManagedByArgoCD,
|
|
},
|
|
Labels: map[string]string{
|
|
common.LabelKeySecretType: common.LabelValueSecretTypeRepositoryWrite,
|
|
},
|
|
},
|
|
Data: map[string][]byte{
|
|
"url": []byte("https://known/repo"),
|
|
},
|
|
}, &corev1.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Namespace: testNamespace,
|
|
Name: "secured-repo-secret",
|
|
Annotations: map[string]string{
|
|
common.AnnotationKeyManagedBy: common.AnnotationValueManagedByArgoCD,
|
|
},
|
|
Labels: map[string]string{
|
|
common.LabelKeySecretType: common.LabelValueSecretTypeRepositoryWrite,
|
|
},
|
|
},
|
|
Data: map[string][]byte{
|
|
"url": []byte("https://secured/repo"),
|
|
},
|
|
}, &corev1.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Namespace: testNamespace,
|
|
Name: "secured-repo-creds-secret",
|
|
Annotations: map[string]string{
|
|
common.AnnotationKeyManagedBy: common.AnnotationValueManagedByArgoCD,
|
|
},
|
|
Labels: map[string]string{
|
|
common.LabelKeySecretType: common.LabelValueSecretTypeRepoCredsWrite,
|
|
},
|
|
},
|
|
Data: map[string][]byte{
|
|
"url": []byte("https://secured"),
|
|
"username": []byte("test-username"),
|
|
"password": []byte("test-password"),
|
|
},
|
|
})
|
|
db := NewDB(testNamespace, settings.NewSettingsManager(t.Context(), clientset, testNamespace), clientset)
|
|
|
|
tests := []struct {
|
|
name string
|
|
repoURL string
|
|
want *v1alpha1.Repository
|
|
}{
|
|
{
|
|
name: "TestUnknownRepo",
|
|
repoURL: "https://unknown/repo",
|
|
want: &v1alpha1.Repository{Repo: "https://unknown/repo"},
|
|
},
|
|
{
|
|
name: "TestKnownRepo",
|
|
repoURL: "https://known/repo",
|
|
want: &v1alpha1.Repository{Repo: "https://known/repo"},
|
|
},
|
|
{
|
|
name: "TestSecuredRepo",
|
|
repoURL: "https://secured/repo",
|
|
want: &v1alpha1.Repository{Repo: "https://secured/repo", Username: "test-username", Password: "test-password", InheritedCreds: true},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
got, err := db.GetWriteRepository(t.Context(), tt.repoURL, "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, tt.want, got)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestCreateClusterSuccessful(t *testing.T) {
|
|
server := "https://mycluster"
|
|
clientset := getClientset()
|
|
db := NewDB(testNamespace, settings.NewSettingsManager(t.Context(), clientset, testNamespace), clientset)
|
|
|
|
_, err := db.CreateCluster(t.Context(), &v1alpha1.Cluster{
|
|
Server: server,
|
|
})
|
|
require.NoError(t, err)
|
|
|
|
secret, err := clientset.CoreV1().Secrets(testNamespace).Get(t.Context(), "cluster-mycluster-3274446258", metav1.GetOptions{})
|
|
require.NoError(t, err)
|
|
|
|
assert.Equal(t, server, string(secret.Data["server"]))
|
|
assert.Equal(t, common.AnnotationValueManagedByArgoCD, secret.Annotations[common.AnnotationKeyManagedBy])
|
|
}
|
|
|
|
func TestDeleteClusterWithManagedSecret(t *testing.T) {
|
|
clusterURL := "https://mycluster"
|
|
clusterName := "cluster-mycluster-3274446258"
|
|
|
|
clientset := getClientset(&corev1.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: clusterName,
|
|
Namespace: testNamespace,
|
|
Labels: map[string]string{
|
|
common.LabelKeySecretType: common.LabelValueSecretTypeCluster,
|
|
},
|
|
Annotations: map[string]string{
|
|
common.AnnotationKeyManagedBy: common.AnnotationValueManagedByArgoCD,
|
|
},
|
|
},
|
|
Data: map[string][]byte{
|
|
"server": []byte(clusterURL),
|
|
"config": []byte("{}"),
|
|
},
|
|
})
|
|
|
|
db := NewDB(testNamespace, settings.NewSettingsManager(t.Context(), clientset, testNamespace), clientset)
|
|
err := db.DeleteCluster(t.Context(), clusterURL)
|
|
require.NoError(t, err)
|
|
|
|
_, err = clientset.CoreV1().Secrets(testNamespace).Get(t.Context(), clusterName, metav1.GetOptions{})
|
|
require.Error(t, err)
|
|
|
|
assert.True(t, errors.IsNotFound(err))
|
|
}
|
|
|
|
func TestDeleteClusterWithUnmanagedSecret(t *testing.T) {
|
|
clusterURL := "https://mycluster"
|
|
clusterName := "mycluster-443"
|
|
|
|
clientset := getClientset(&corev1.Secret{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: clusterName,
|
|
Namespace: testNamespace,
|
|
Labels: map[string]string{
|
|
common.LabelKeySecretType: common.LabelValueSecretTypeCluster,
|
|
},
|
|
},
|
|
Data: map[string][]byte{
|
|
"server": []byte(clusterURL),
|
|
"config": []byte("{}"),
|
|
},
|
|
})
|
|
|
|
db := NewDB(testNamespace, settings.NewSettingsManager(t.Context(), clientset, testNamespace), clientset)
|
|
err := db.DeleteCluster(t.Context(), clusterURL)
|
|
require.NoError(t, err)
|
|
|
|
secret, err := clientset.CoreV1().Secrets(testNamespace).Get(t.Context(), clusterName, metav1.GetOptions{})
|
|
require.NoError(t, err)
|
|
|
|
assert.Empty(t, secret.Labels)
|
|
}
|
|
|
|
func TestFuzzyEquivalence(t *testing.T) {
|
|
clientset := getClientset()
|
|
ctx := t.Context()
|
|
db := NewDB(testNamespace, settings.NewSettingsManager(t.Context(), clientset, testNamespace), clientset)
|
|
|
|
repo, err := db.CreateRepository(ctx, &v1alpha1.Repository{
|
|
Repo: "https://github.com/argoproj/argocd-example-apps",
|
|
})
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "https://github.com/argoproj/argocd-example-apps", repo.Repo)
|
|
|
|
repo, err = db.CreateRepository(ctx, &v1alpha1.Repository{
|
|
Repo: "https://github.com/argoproj/argocd-example-apps.git",
|
|
})
|
|
require.ErrorContains(t, err, "already exists")
|
|
assert.Nil(t, repo)
|
|
|
|
repo, err = db.CreateRepository(ctx, &v1alpha1.Repository{
|
|
Repo: "https://github.com/argoproj/argocd-example-APPS",
|
|
})
|
|
require.ErrorContains(t, err, "already exists")
|
|
assert.Nil(t, repo)
|
|
|
|
repo, err = db.GetRepository(ctx, "https://github.com/argoproj/argocd-example-APPS", "")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "https://github.com/argoproj/argocd-example-apps", repo.Repo)
|
|
}
|
|
|
|
func TestGetApplicationControllerReplicas(t *testing.T) {
|
|
clientset := getClientset()
|
|
expectedReplicas := int32(2)
|
|
t.Setenv(common.EnvControllerReplicas, "2")
|
|
db := NewDB(testNamespace, settings.NewSettingsManager(t.Context(), clientset, testNamespace), clientset)
|
|
replicas := db.GetApplicationControllerReplicas()
|
|
assert.Equal(t, int(expectedReplicas), replicas)
|
|
|
|
expectedReplicas = int32(3)
|
|
clientset = getClientset(&appsv1.Deployment{
|
|
ObjectMeta: metav1.ObjectMeta{
|
|
Name: common.CommandApplicationController,
|
|
Namespace: testNamespace,
|
|
},
|
|
Spec: appsv1.DeploymentSpec{
|
|
Replicas: &expectedReplicas,
|
|
},
|
|
})
|
|
t.Setenv(common.EnvControllerReplicas, "2")
|
|
db = NewDB(testNamespace, settings.NewSettingsManager(t.Context(), clientset, testNamespace), clientset)
|
|
replicas = db.GetApplicationControllerReplicas()
|
|
assert.Equal(t, int(expectedReplicas), replicas)
|
|
}
|