mirror of
https://github.com/BerriAI/litellm.git
synced 2026-08-28 05:25:59 +00:00
3f2e0badb4
The chart shipped ghcr.io/berriai/litellm-database as its image default, with a comment offering it as the "optimized image with database". That distinction no longer exists: Dockerfile and docker/Dockerfile.database differ only in comment text and one builder-stage apk package, and both published images bake the prisma CLI, engines, schema.prisma and prisma_migration.py into /app, so either one runs the migrations job. Point the default at the canonical image the release notes, the cosign verification instructions and the chart's own README already name, and update the chart's tests and README so nothing still refers to the legacy repository.
344 lines
9.3 KiB
YAML
344 lines
9.3 KiB
YAML
suite: test migrations job
|
|
templates:
|
|
- migrations-job.yaml
|
|
tests:
|
|
- it: should work with envVars
|
|
template: migrations-job.yaml
|
|
set:
|
|
envVars:
|
|
TEST_ENV_VAR: "test_value"
|
|
ANOTHER_VAR: "another_value"
|
|
migrationJob:
|
|
enabled: true
|
|
asserts:
|
|
- contains:
|
|
path: spec.template.spec.containers[0].env
|
|
content:
|
|
name: TEST_ENV_VAR
|
|
value: "test_value"
|
|
- contains:
|
|
path: spec.template.spec.containers[0].env
|
|
content:
|
|
name: ANOTHER_VAR
|
|
value: "another_value"
|
|
|
|
- it: should work with extraEnvVars
|
|
template: migrations-job.yaml
|
|
set:
|
|
extraEnvVars:
|
|
- name: EXTRA_ENV_VAR
|
|
valueFrom:
|
|
fieldRef:
|
|
fieldPath: metadata.labels['env']
|
|
- name: SIMPLE_EXTRA_VAR
|
|
value: "simple_value"
|
|
migrationJob:
|
|
enabled: true
|
|
asserts:
|
|
- contains:
|
|
path: spec.template.spec.containers[0].env
|
|
content:
|
|
name: EXTRA_ENV_VAR
|
|
valueFrom:
|
|
fieldRef:
|
|
fieldPath: metadata.labels['env']
|
|
- contains:
|
|
path: spec.template.spec.containers[0].env
|
|
content:
|
|
name: SIMPLE_EXTRA_VAR
|
|
value: "simple_value"
|
|
|
|
- it: should work with both envVars and extraEnvVars
|
|
template: migrations-job.yaml
|
|
set:
|
|
envVars:
|
|
ENV_VAR: "env_var_value"
|
|
extraEnvVars:
|
|
- name: EXTRA_ENV_VAR
|
|
value: "extra_env_var_value"
|
|
migrationJob:
|
|
enabled: true
|
|
asserts:
|
|
- contains:
|
|
path: spec.template.spec.containers[0].env
|
|
content:
|
|
name: ENV_VAR
|
|
value: "env_var_value"
|
|
- contains:
|
|
path: spec.template.spec.containers[0].env
|
|
content:
|
|
name: EXTRA_ENV_VAR
|
|
value: "extra_env_var_value"
|
|
|
|
- it: should not render when migrations job is disabled
|
|
template: migrations-job.yaml
|
|
set:
|
|
migrationJob:
|
|
enabled: false
|
|
asserts:
|
|
- hasDocuments:
|
|
count: 0
|
|
|
|
- it: should still include default env vars
|
|
template: migrations-job.yaml
|
|
set:
|
|
envVars:
|
|
CUSTOM_VAR: "custom_value"
|
|
migrationJob:
|
|
enabled: true
|
|
db:
|
|
useExisting: true
|
|
endpoint: "test-db"
|
|
database: "testdb"
|
|
url: "postgresql://user:pass@test-db:5432/testdb"
|
|
secret:
|
|
name: "test-secret"
|
|
usernameKey: "username"
|
|
passwordKey: "password"
|
|
asserts:
|
|
- contains:
|
|
path: spec.template.spec.containers[0].env
|
|
content:
|
|
name: DISABLE_SCHEMA_UPDATE
|
|
value: "false"
|
|
- contains:
|
|
path: spec.template.spec.containers[0].env
|
|
content:
|
|
name: DATABASE_HOST
|
|
value: "test-db"
|
|
- contains:
|
|
path: spec.template.spec.containers[0].env
|
|
content:
|
|
name: CUSTOM_VAR
|
|
value: "custom_value"
|
|
|
|
- it: should not include DATABASE_URL when deployStandalone is false
|
|
template: migrations-job.yaml
|
|
set:
|
|
migrationJob:
|
|
enabled: true
|
|
db:
|
|
deployStandalone: false
|
|
useExisting: false
|
|
asserts:
|
|
- notContains:
|
|
path: spec.template.spec.containers[0].env
|
|
content:
|
|
name: DATABASE_URL
|
|
|
|
- it: should use default service account for helm hooks when serviceAccount.create is true
|
|
template: migrations-job.yaml
|
|
set:
|
|
migrationJob:
|
|
enabled: true
|
|
hooks:
|
|
helm:
|
|
enabled: true
|
|
serviceAccount:
|
|
create: true
|
|
asserts:
|
|
- equal:
|
|
path: spec.template.spec.serviceAccountName
|
|
value: default
|
|
|
|
- it: should use migrationJob.serviceAccountName override for helm hooks when serviceAccount.create is true
|
|
template: migrations-job.yaml
|
|
set:
|
|
migrationJob:
|
|
enabled: true
|
|
serviceAccountName: migration-sa
|
|
hooks:
|
|
helm:
|
|
enabled: true
|
|
serviceAccount:
|
|
create: true
|
|
asserts:
|
|
- equal:
|
|
path: spec.template.spec.serviceAccountName
|
|
value: migration-sa
|
|
|
|
- it: should use chart service account when helm hooks are disabled
|
|
template: migrations-job.yaml
|
|
set:
|
|
migrationJob:
|
|
enabled: true
|
|
hooks:
|
|
helm:
|
|
enabled: false
|
|
serviceAccount:
|
|
create: true
|
|
name: my-custom-sa
|
|
asserts:
|
|
- equal:
|
|
path: spec.template.spec.serviceAccountName
|
|
value: my-custom-sa
|
|
|
|
- it: should use pre-existing service account when helm hooks are enabled but serviceAccount.create is false
|
|
template: migrations-job.yaml
|
|
set:
|
|
migrationJob:
|
|
enabled: true
|
|
hooks:
|
|
helm:
|
|
enabled: true
|
|
serviceAccount:
|
|
create: false
|
|
name: pre-existing-sa
|
|
asserts:
|
|
- equal:
|
|
path: spec.template.spec.serviceAccountName
|
|
value: pre-existing-sa
|
|
- it: should work with extraInitContainers
|
|
template: migrations-job.yaml
|
|
set:
|
|
migrationJob:
|
|
enabled: true
|
|
extraInitContainers:
|
|
- name: init-test
|
|
image: busybox:latest
|
|
command: ["echo", "hello"]
|
|
asserts:
|
|
- contains:
|
|
path: spec.template.spec.initContainers
|
|
content:
|
|
name: init-test
|
|
image: busybox:latest
|
|
command: ["echo", "hello"]
|
|
- it: should support tpl in extraInitContainers
|
|
template: migrations-job.yaml
|
|
set:
|
|
image:
|
|
repository: ghcr.io/berriai/litellm
|
|
tag: test
|
|
migrationJob:
|
|
enabled: true
|
|
extraInitContainers:
|
|
- name: init-tpl
|
|
image: "{{ .Values.image.repository }}:{{ .Values.image.tag }}"
|
|
command: ["echo", "hello"]
|
|
asserts:
|
|
- contains:
|
|
path: spec.template.spec.initContainers
|
|
content:
|
|
name: init-tpl
|
|
image: "ghcr.io/berriai/litellm:test"
|
|
command: ["echo", "hello"]
|
|
- it: should work with extraContainers
|
|
template: migrations-job.yaml
|
|
set:
|
|
migrationJob:
|
|
enabled: true
|
|
extraContainers:
|
|
- name: sidecar
|
|
image: busybox:latest
|
|
asserts:
|
|
- contains:
|
|
path: spec.template.spec.containers
|
|
content:
|
|
name: sidecar
|
|
image: busybox:latest
|
|
- it: should support tpl in extraContainers
|
|
template: migrations-job.yaml
|
|
set:
|
|
image:
|
|
repository: ghcr.io/berriai/litellm
|
|
tag: test
|
|
migrationJob:
|
|
enabled: true
|
|
extraContainers:
|
|
- name: sidecar-tpl
|
|
image: "{{ .Values.image.repository }}:{{ .Values.image.tag }}"
|
|
asserts:
|
|
- contains:
|
|
path: spec.template.spec.containers
|
|
content:
|
|
name: sidecar-tpl
|
|
image: "ghcr.io/berriai/litellm:test"
|
|
- it: should render the pod-level securityContext from podSecurityContext
|
|
template: migrations-job.yaml
|
|
set:
|
|
migrationJob:
|
|
enabled: true
|
|
podSecurityContext:
|
|
fsGroup: 10000
|
|
runAsUser: 10000
|
|
runAsNonRoot: true
|
|
asserts:
|
|
- equal:
|
|
path: spec.template.spec.securityContext
|
|
value:
|
|
fsGroup: 10000
|
|
runAsUser: 10000
|
|
runAsNonRoot: true
|
|
- it: should keep the pod-level and container-level securityContext separate
|
|
template: migrations-job.yaml
|
|
set:
|
|
migrationJob:
|
|
enabled: true
|
|
podSecurityContext:
|
|
fsGroup: 10000
|
|
securityContext:
|
|
allowPrivilegeEscalation: false
|
|
readOnlyRootFilesystem: true
|
|
asserts:
|
|
- equal:
|
|
path: spec.template.spec.securityContext
|
|
value:
|
|
fsGroup: 10000
|
|
- equal:
|
|
path: spec.template.spec.containers[0].securityContext
|
|
value:
|
|
allowPrivilegeEscalation: false
|
|
readOnlyRootFilesystem: true
|
|
- it: should schedule onto the same nodes as the gateway
|
|
template: migrations-job.yaml
|
|
set:
|
|
migrationJob:
|
|
enabled: true
|
|
nodeSelector:
|
|
karpenter.sh/nodepool: litellm-e2e
|
|
tolerations:
|
|
- key: workload
|
|
operator: Equal
|
|
value: litellm-e2e
|
|
effect: NoSchedule
|
|
asserts:
|
|
- equal:
|
|
path: spec.template.spec.nodeSelector
|
|
value:
|
|
karpenter.sh/nodepool: litellm-e2e
|
|
- equal:
|
|
path: spec.template.spec.tolerations
|
|
value:
|
|
- key: workload
|
|
operator: Equal
|
|
value: litellm-e2e
|
|
effect: NoSchedule
|
|
|
|
- it: bounds the Job with a deadline by default, so a blocked migration cannot stall the release forever
|
|
set:
|
|
migrationJob:
|
|
enabled: true
|
|
asserts:
|
|
- equal:
|
|
path: spec.activeDeadlineSeconds
|
|
value: 1800
|
|
|
|
- it: honours an operator-supplied deadline
|
|
set:
|
|
migrationJob:
|
|
enabled: true
|
|
activeDeadlineSeconds: 600
|
|
asserts:
|
|
- equal:
|
|
path: spec.activeDeadlineSeconds
|
|
value: 600
|
|
|
|
- it: omits the deadline entirely when it is nulled out, restoring the unbounded behaviour
|
|
set:
|
|
migrationJob:
|
|
enabled: true
|
|
activeDeadlineSeconds: null
|
|
asserts:
|
|
- notExists:
|
|
path: spec.activeDeadlineSeconds
|