{{- $root := . }}
{{- $namespace := .Release.Namespace }}
apiVersion: v1
kind: Pod
metadata:
  name: "{{ include "qdrant.fullname" . }}-test-db-interaction"
  labels:
    {{- include "qdrant.labels" . | nindent 4 }}
    {{- include "qdrant.additionalLabels" . | nindent 4 }}
  annotations:
    "helm.sh/hook": test
{{- with .Values.additionalAnnotations }}
    {{- toYaml . | nindent 4 }}
{{- end }}
spec:
  containers:
    - name: test-script
      image: {{ .Values.chartTests.dbInteraction.image | quote }}
      args: ['bash', '/app/entrypoint.sh']
      volumeMounts:
        - mountPath: /app
          name: test-script
        {{- if .Values.additionalVolumeMounts }}
{{- toYaml .Values.additionalVolumeMounts | default "" | nindent 8 }}
        {{- end}}
      resources:
        {{- toYaml .Values.chartTests.dbInteraction.resources | nindent 8 }}
  {{- with .Values.imagePullSecrets }}
  imagePullSecrets:
    {{- toYaml . | nindent 8 }}
  {{- end }}
  {{- with .Values.nodeSelector }}
  nodeSelector:
    {{- toYaml . | nindent 4 }}
  {{- end }}
  {{- with .Values.tolerations }}
  tolerations:
    {{- toYaml . | nindent 4 }}
  {{- end }}
  volumes:
    - name: test-script
      configMap:
        name: "{{ include "qdrant.fullname" . }}-test-db-interaction"
    {{- if .Values.additionalVolumes }}
{{- toYaml .Values.additionalVolumes | default "" | nindent 4 }}
    {{- end}}
  restartPolicy: Never
  serviceAccountName: {{ include "qdrant.fullname" . }}
---
apiVersion: v1
kind: ConfigMap
metadata:
  name: "{{ include "qdrant.fullname" . }}-test-db-interaction"
  labels:
    {{- include "qdrant.labels" . | nindent 4 }}
    {{- include "qdrant.additionalLabels" . | nindent 4 }}
  annotations:
    "helm.sh/hook": test
{{- with .Values.additionalAnnotations }}
    {{- toYaml . | nindent 4 }}
{{- end }}
data:
  entrypoint.sh: |
    #!/bin/bash
    set -xe
    # Kind's networking is very flaky
    echo 'connect-timeout = 5' > $HOME/.curlrc
    echo 'retry = 60' >> $HOME/.curlrc
    echo 'retry-delay = 5' >> $HOME/.curlrc
    echo 'retry-all-errors' >> $HOME/.curlrc
    # Don't clutter the logs with progress bars
    echo 'no-progress-meter' >> $HOME/.curlrc
    # Ensure errors cause the script to fail, but show the response body
    echo 'fail-with-body' >> $HOME/.curlrc

    if [ -d /mnt/secrets/certs ]; then
      cp /mnt/secrets/certs/ca.pem /usr/share/pki/trust/anchors/private-ca.pem
      update-ca-certificates
    fi

    QDRANT_COLLECTION="test_collection"
    {{- range .Values.service.ports }}
    {{- if eq .name "http" }}
    echo "Connecting to {{ include "qdrant.fullname" $root }}.{{ $namespace }}:{{ .port }}"
    QDRANT_URL="http://{{ include "qdrant.fullname" $root }}.{{ $namespace }}:{{ .port }}"
    {{- if and $root.Values.config.service $root.Values.config.service.enable_tls }}
    echo "Using https"
    QDRANT_URL="https://{{ include "qdrant.fullname" $root }}.{{ $namespace }}:{{ .port }}"
    {{- end }}
    {{- end }}
    {{- end }}
    API_KEY_HEADER=""
    {{- if .Values.apiKey }}
    API_KEY_HEADER="Api-key: {{ .Values.apiKey }}"
    {{- else if .Values.readOnlyApiKey }}
    API_KEY_HEADER="Api-key: {{ .Values.readOnlyApiKey }}"
    {{- end }}

    # Delete collection if exists
    curl -X DELETE -H "${API_KEY_HEADER}" $QDRANT_URL/collections/${QDRANT_COLLECTION}

    # Create collection
    curl -X PUT \
    -H 'Content-Type: application-json' \
    -d '{"vectors":{"size":4,"distance":"Dot"}}' \
    -H  "${API_KEY_HEADER}" \
    $QDRANT_URL/collections/${QDRANT_COLLECTION}

    # Insert points
    curl -X PUT \
    -H 'Content-Type: application-json' \
    -d '{"points":[
      {"id":1,"vector":[0.05, 0.61, 0.76, 0.74],"payload":{"city":"Berlin"}},
      {"id":2,"vector":[0.19, 0.81, 0.75, 0.11],"payload":{"city":"London"}},
      {"id":3,"vector":[0.36, 0.55, 0.47, 0.94],"payload":{"city":"Moscow"}},
      {"id":4,"vector":[0.18, 0.01, 0.85, 0.80],"payload":{"city":"New York"}},
      {"id":5,"vector":[0.24, 0.18, 0.22, 0.44],"payload":{"city":"Beijing"}},
      {"id":6,"vector":[0.35, 0.08, 0.11, 0.44],"payload":{"city":"Mumbai"}}
    ]}' \
    -H  "${API_KEY_HEADER}" \
    $QDRANT_URL/collections/${QDRANT_COLLECTION}/points

    # Run query
    curl -X POST \
    -H 'Content-Type: application-json' \
    -d '{"vector":[0.2, 0.1, 0.9, 0.7],"limit":3}' \
    -H  "${API_KEY_HEADER}" \
    $QDRANT_URL/collections/${QDRANT_COLLECTION}/points/search