Elasticsearch tests

k8s/elasticsearch/apptest/deployer/manifest/tester.yaml.template

@@ -26,5 +26,7 @@ spec:
         fieldRef:
           fieldPath: metadata.namespace
     - name: APP_INSTANCE_NAME
-      value: ${APP_INSTANCE_NAME}
+      value: "${APP_INSTANCE_NAME}"
+    - name: REPLICAS
+      value: "${REPLICAS}"
   restartPolicy: Never

k8s/elasticsearch/apptest/tester/Dockerfile

@@ -6,6 +6,7 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     jq \
     uuid-runtime \
     wget \
+    curl \
     && rm -rf /var/lib/apt/lists/*
 
 RUN wget -q -O /bin/kubectl \

k8s/elasticsearch/apptest/tester/tester.sh

@@ -17,6 +17,9 @@
 set -xeo pipefail
 shopt -s nullglob
 
+export ELASTIC_URL="http://${APP_INSTANCE_NAME}-elasticsearch-svc:9200"
+export HEALTH_URL="${ELASTIC_URL}/_cluster/health"
+
 for test in /tests/*; do
   testrunner -logtostderr "--test_spec=${test}"
 done

k8s/elasticsearch/apptest/tester/tests/basic-suite.yaml

@@ -1,8 +1,89 @@
 actions:
-- name: dummy test
+- name: Wait until all expected nodes are connected to the cluster
   bashTest:
     script: |-
-      exit 0
+      function get_nodes_count() {
+        curl -s "${HEALTH_URL}" | jq '.number_of_nodes'
+      }
+
+      expected_nodes="${REPLICAS}"
+
+      timeout=300
+      timeout_time=$(( $(date +%s) + timeout ))
+
+      # Wait for all nodes to join the cluster
+      nodes_count="$(get_nodes_count)"
+      while [[ "${nodes_count}" -ne "${expected_nodes}" ]] \
+          && [[ "$(date +%s)" -le "${timeout_time}" ]]; do
+        sleep 3
+        nodes_count="$(get_nodes_count)"
+      done
+
+      if [[ "${nodes_count}" -eq "${expected_nodes}" ]]; then
+        echo "OK - all nodes joined"
+      else
+        echo "FAIL - nodes count: ${nodes_count}"
+      fi
+    expect:
+      stdout:
+        contains: OK - all nodes joined
+
+- name: Wait for green status in the cluster
+  bashTest:
+    script: |-
+      function get_cluster_status() {
+        curl -s "${HEALTH_URL}" | jq '.status'
+      }
+
+      timeout=300
+      timeout_time=$(( $(date +%s) + timeout ))
+
+      # Wait for green status
+      status="$(get_cluster_status)"
+      while [[ "${status}" != '"green"' ]] \
+          && [[ "$(date +%s)" -le "${timeout_time}" ]]; do
+        sleep 3
+        status="$(get_cluster_status)"
+      done
+
+      if [[ "${status}" == '"green"' ]]; then
+        echo "OK - green status"
+      else
+        echo "FAIL - status: ${status}"
+      fi
+    expect:
+      stdout:
+        contains: OK - green status
+
+- name: Add a document
+  bashTest:
+    script: |-
+      curl -s -f -X PUT -H 'Content-Type:application/json' -d '{
+        "name": "John",
+        "surname": "Smith"
+      }' "${ELASTIC_URL}/employees/person/1"
+    expect:
+      exitCode:
+        equals: 0
+
+- name: Update the document
+  bashTest:
+    script: |-
+      curl -s -f -X POST -H 'Content-Type:application/json' -d '{
+        "doc": {
+          "lastname": "Jones"
+        }
+      }' "${ELASTIC_URL}/employees/person/1/_update"
     expect:
       exitCode:
         equals: 0
+
+- name: Search for the document
+  bashTest:
+    script: |-
+      # Wait to allow the object to be propagated between all nodes
+      sleep 5
+      curl -s -f "${ELASTIC_URL}/_search?q=jones"
+    expect:
+      stdout:
+        contains: "\"name\":\"John\""

k8s/elasticsearch/deployer/Dockerfile

@@ -8,6 +8,10 @@ COPY apptest/deployer /data-test/
 # Provide registry prefix and tag for default values for images.
 ARG REGISTRY
 ARG TAG
+
+ENV WAIT_FOR_READY_TIMEOUT 600
+ENV TESTER_TIMEOUT 600
+
 RUN cat /data/schema.yaml \
     | env -i "REGISTRY=$REGISTRY" "TAG=$TAG" envsubst \
     > /data/schema.yaml.new \