153 lines
3.8 KiB
Go
153 lines
3.8 KiB
Go
package tests_test
|
|
|
|
import (
|
|
"sigs.k8s.io/kustomize/v3/k8sdeps/kunstruct"
|
|
"sigs.k8s.io/kustomize/v3/k8sdeps/transformer"
|
|
"sigs.k8s.io/kustomize/v3/pkg/fs"
|
|
"sigs.k8s.io/kustomize/v3/pkg/loader"
|
|
"sigs.k8s.io/kustomize/v3/pkg/plugins"
|
|
"sigs.k8s.io/kustomize/v3/pkg/resmap"
|
|
"sigs.k8s.io/kustomize/v3/pkg/resource"
|
|
"sigs.k8s.io/kustomize/v3/pkg/target"
|
|
"sigs.k8s.io/kustomize/v3/pkg/validators"
|
|
"testing"
|
|
)
|
|
|
|
func writeVizierDbBase(th *KustTestHarness) {
|
|
th.writeF("/manifests/katib-v1alpha1/vizier-db/base/vizier-db-deployment.yaml", `
|
|
apiVersion: apps/v1
|
|
kind: Deployment
|
|
metadata:
|
|
name: vizier-db
|
|
labels:
|
|
component: db
|
|
spec:
|
|
replicas: 1
|
|
selector:
|
|
matchLabels:
|
|
component: db
|
|
template:
|
|
metadata:
|
|
name: vizier-db
|
|
labels:
|
|
component: db
|
|
spec:
|
|
containers:
|
|
- name: vizier-db
|
|
image: mysql:8.0.3
|
|
args:
|
|
- --datadir
|
|
- /var/lib/mysql/datadir
|
|
env:
|
|
- name: MYSQL_ROOT_PASSWORD
|
|
valueFrom:
|
|
secretKeyRef:
|
|
name: vizier-db-secrets
|
|
key: MYSQL_ROOT_PASSWORD
|
|
- name: MYSQL_ALLOW_EMPTY_PASSWORD
|
|
value: "true"
|
|
- name: MYSQL_DATABASE
|
|
value: "vizier"
|
|
ports:
|
|
- name: dbapi
|
|
containerPort: 3306
|
|
readinessProbe:
|
|
exec:
|
|
command:
|
|
- "/bin/bash"
|
|
- "-c"
|
|
- "mysql -D $$MYSQL_DATABASE -p$$MYSQL_ROOT_PASSWORD -e 'SELECT 1'"
|
|
initialDelaySeconds: 5
|
|
periodSeconds: 2
|
|
timeoutSeconds: 1
|
|
volumeMounts:
|
|
- name: katib-mysql
|
|
mountPath: /var/lib/mysql
|
|
volumes:
|
|
- name: katib-mysql
|
|
persistentVolumeClaim:
|
|
claimName: katib-mysql
|
|
`)
|
|
th.writeF("/manifests/katib-v1alpha1/vizier-db/base/vizier-db-pvc.yaml", `
|
|
apiVersion: v1
|
|
kind: PersistentVolumeClaim
|
|
metadata:
|
|
name: katib-mysql
|
|
spec:
|
|
accessModes:
|
|
- ReadWriteOnce
|
|
resources:
|
|
requests:
|
|
storage: 10Gi
|
|
`)
|
|
th.writeF("/manifests/katib-v1alpha1/vizier-db/base/vizier-db-secret.yaml", `
|
|
apiVersion: v1
|
|
kind: Secret
|
|
type: Opaque
|
|
metadata:
|
|
name: vizier-db-secrets
|
|
data:
|
|
MYSQL_ROOT_PASSWORD: dGVzdA== # "test"
|
|
`)
|
|
th.writeF("/manifests/katib-v1alpha1/vizier-db/base/vizier-db-service.yaml", `
|
|
apiVersion: v1
|
|
kind: Service
|
|
metadata:
|
|
name: vizier-db
|
|
labels:
|
|
component: db
|
|
spec:
|
|
type: ClusterIP
|
|
ports:
|
|
- port: 3306
|
|
protocol: TCP
|
|
name: dbapi
|
|
selector:
|
|
component: db
|
|
`)
|
|
th.writeK("/manifests/katib-v1alpha1/vizier-db/base", `
|
|
namespace: kubeflow
|
|
resources:
|
|
- vizier-db-deployment.yaml
|
|
- vizier-db-pvc.yaml
|
|
- vizier-db-secret.yaml
|
|
- vizier-db-service.yaml
|
|
generatorOptions:
|
|
disableNameSuffixHash: true
|
|
images:
|
|
- name: mysql
|
|
newTag: 8.0.3
|
|
`)
|
|
}
|
|
|
|
func TestVizierDbBase(t *testing.T) {
|
|
th := NewKustTestHarness(t, "/manifests/katib-v1alpha1/vizier-db/base")
|
|
writeVizierDbBase(th)
|
|
m, err := th.makeKustTarget().MakeCustomizedResMap()
|
|
if err != nil {
|
|
t.Fatalf("Err: %v", err)
|
|
}
|
|
expected, err := m.AsYaml()
|
|
if err != nil {
|
|
t.Fatalf("Err: %v", err)
|
|
}
|
|
targetPath := "../katib-v1alpha1/vizier-db/base"
|
|
fsys := fs.MakeRealFS()
|
|
lrc := loader.RestrictionRootOnly
|
|
_loader, loaderErr := loader.NewLoader(lrc, validators.MakeFakeValidator(), targetPath, fsys)
|
|
if loaderErr != nil {
|
|
t.Fatalf("could not load kustomize loader: %v", loaderErr)
|
|
}
|
|
rf := resmap.NewFactory(resource.NewFactory(kunstruct.NewKunstructuredFactoryImpl()), transformer.NewFactoryImpl())
|
|
pc := plugins.DefaultPluginConfig()
|
|
kt, err := target.NewKustTarget(_loader, rf, transformer.NewFactoryImpl(), plugins.NewLoader(pc, rf))
|
|
if err != nil {
|
|
th.t.Fatalf("Unexpected construction error %v", err)
|
|
}
|
|
actual, err := kt.MakeCustomizedResMap()
|
|
if err != nil {
|
|
t.Fatalf("Err: %v", err)
|
|
}
|
|
th.assertActualEqualsExpected(actual, string(expected))
|
|
}
|