Merge pull request #31 from elastic/rolly_polly

Generate node roles instead of hard coding them

elasticsearch/templates/statefulset.yaml

@@ -154,12 +154,10 @@ spec:
             value: "{{ .Values.networkHost }}"
           - name: ES_JAVA_OPTS
             value: "{{ .Values.esJavaOpts }}"
-          - name: node.master
-            value: "{{ .Values.roles.master }}"
-          - name: node.data
-            value: "{{ .Values.roles.data }}"
-          - name: node.ingest
-            value: "{{ .Values.roles.ingest }}"
+          {{- range $role, $enabled := .Values.roles }}
+          - name: node.{{ $role }}
+            value: "{{ $enabled }}"
+          {{- end }}
 {{- if .Values.extraEnvs }}
 {{ toYaml .Values.extraEnvs | indent 10 }}
 {{- end }}

elasticsearch/tests/elasticsearch_test.py

@@ -233,6 +233,17 @@ def test_dont_set_initial_master_nodes_if_not_master_when_using_es_version_7():
     for e in env:
         assert e['name'] != 'cluster.initial_master_nodes'
 
+def test_enabling_machine_learning_role():
+    config = '''
+roles:
+  ml: "true"
+'''
+    r = helm_template(config)
+    env = r['statefulset'][uname]['spec']['template']['spec']['containers'][0]['env']
+
+    assert {'name': 'node.ml',
+            'value': 'true'} in env
+
 
 def test_adding_extra_env_vars():
     config = '''