Merge pull request #11957 from mikemccand/smoke_test_plugins

Add simple plugins smoke tester

Closes #11935

dev-tools/smoke_test_plugins.py

@@ -0,0 +1,170 @@
+# Licensed to Elasticsearch under one or more contributor
+# license agreements. See the NOTICE file distributed with
+# this work for additional information regarding copyright
+# ownership. Elasticsearch licenses this file to you under
+# the Apache License, Version 2.0 (the "License"); you may
+# not use this file except in compliance  with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on 
+# an 'AS IS' BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
+# either express or implied. See the License for the specific
+# language governing permissions and limitations under the License.
+
+import datetime
+import traceback
+import json
+import os
+import shutil
+import signal
+import socket
+import subprocess
+import tempfile
+import threading
+import time
+
+from http.client import HTTPConnection
+
+LOG = os.environ.get('ES_SMOKE_TEST_PLUGINS_LOG', '/tmp/elasticsearch_smoke_test_plugins.log')
+
+print('Logging to %s' % LOG)
+
+if os.path.exists(LOG):
+  raise RuntimeError('please remove old log %s first' % LOG)
+
+try:
+  JAVA_HOME = os.environ['JAVA7_HOME']
+except KeyError:
+  try:
+    JAVA_HOME = os.environ['JAVA_HOME']
+  except KeyError:
+    raise RuntimeError("""
+    Please set JAVA_HOME in the env before running release tool
+    On OSX use: export JAVA_HOME=`/usr/libexec/java_home -v '1.7*'`""")
+
+JAVA_ENV = 'export JAVA_HOME="%s" PATH="%s/bin:$PATH" JAVACMD="%s/bin/java"' % (JAVA_HOME, JAVA_HOME, JAVA_HOME)
+
+try:
+  # make sure mvn3 is used if mvn3 is available
+  # some systems use maven 2 as default
+  subprocess.check_output('mvn3 --version', shell=True, stderr=subprocess.STDOUT)
+  MVN = 'mvn3'
+except subprocess.CalledProcessError:
+  MVN = 'mvn'
+
+def log(msg):
+  f = open(LOG, mode='ab')
+  f.write(('\n'+msg).encode('utf-8'))
+  f.close()
+
+def run(command, quiet=False):
+  log('%s: RUN: %s\n' % (datetime.datetime.now(), command))
+  if os.system('%s >> %s 2>&1' % (command, LOG)):
+    msg = '    FAILED: %s [see log %s]' % (command, LOG)
+    if not quiet:
+      print(msg)
+    raise RuntimeError(msg)
+
+def readServerOutput(p, startupEvent, failureEvent):
+  try:
+    while True:
+      line = p.stdout.readline()
+      if len(line) == 0:
+        p.poll()
+        if not startupEvent.isSet():
+          failureEvent.set()
+          startupEvent.set()
+        print('ES: **process exit**\n')
+        break
+      line = line.decode('utf-8')
+      if line.endswith('started') and not startupEvent.isSet():
+        startupEvent.set()
+      print('ES: %s' % line.rstrip())
+  except:
+    print()
+    print('Exception reading Elasticsearch output:')
+    traceback.print_exc()
+    failureEvent.set()
+    startupEvent.set()
+
+if __name__ == '__main__':
+  print('Build release bits...')
+
+  run('%s; %s clean package -DskipTests' % (JAVA_ENV, MVN))
+
+  for f in os.listdir('core/target/releases/'):
+    if f.endswith('.tar.gz'):
+      artifact = f
+      break
+  else:
+    raise RuntimeError('could not find elasticsearch release under core/target/releases/')
+  
+  tmp_dir = tempfile.mkdtemp()
+  p = None
+  try:
+    # Extract artifact:
+    run('tar -xzf core/target/releases/%s -C %s' % (artifact, tmp_dir))
+    es_install_dir = os.path.join(tmp_dir, artifact[:-7])
+    es_plugin_path = os.path.join(es_install_dir, 'bin/plugin')
+    plugin_names = set()
+    print('Find plugins:')
+    for name in os.listdir('plugins'):
+      if name not in ('target', 'pom.xml'):
+        url = 'file://%s/plugins/%s/target/elasticsearch-%s-2.0.0-SNAPSHOT.jar' % (os.path.abspath('.'), name, name)
+        print('  install plugin %s...' % name)
+        run('%s; %s --url %s -install %s' % (JAVA_ENV, es_plugin_path, url, name))
+        plugin_names.add(name)
+
+    print('Start Elasticsearch')
+
+    env = os.environ.copy()
+    env['JAVA_HOME'] = JAVA_HOME
+    env['PATH'] = '%s/bin:%s' % (JAVA_HOME, env['PATH'])
+    env['JAVA_CMD'] = '%s/bin/java' % JAVA_HOME
+    
+    startupEvent = threading.Event()
+    failureEvent = threading.Event()
+    p = subprocess.Popen(('%s/bin/elasticsearch' % es_install_dir,
+                          '-Des.node.name=smoke_tester',
+                          '-Des.cluster.name=smoke_tester_cluster'
+                          '-Des.discovery.zen.ping.multicast.enabled=false',
+                          '-Des.script.inline=on',
+                          '-Des.script.indexed=on'),
+                         stdout = subprocess.PIPE,
+                         stderr = subprocess.STDOUT,
+                         env = env)
+    thread = threading.Thread(target=readServerOutput, args=(p, startupEvent, failureEvent))
+    thread.setDaemon(True)
+    thread.start()
+
+    startupEvent.wait(1200)
+    if failureEvent.isSet():
+      raise RuntimeError('ES failed to start')
+
+    print('Confirm plugins are installed')
+    conn = HTTPConnection('127.0.0.1', 9200, 20);
+    conn.request('GET', '/_nodes?plugin=true&pretty=true')
+    res = conn.getresponse()
+    if res.status == 200:
+      nodes = json.loads(res.read().decode("utf-8"))['nodes']
+      for _, node in nodes.items():
+        node_plugins = node['plugins']
+        for node_plugin in node_plugins:
+          if not plugin_names.get(node_plugin['name'], False):
+            raise RuntimeError('Unexpeced plugin %s' % node_plugin['name'])
+          del plugin_names[node_plugin['name']]
+      if plugin_names:
+        raise RuntimeError('Plugins not loaded %s' % list(plugin_names.keys()))
+    else:
+      raise RuntimeError('Expected HTTP 200 but got %s' % res.status)
+  finally:
+    if p is not None:
+      try:
+        os.kill(p.pid, signal.SIGKILL)
+      except ProcessLookupError:
+        pass
+    shutil.rmtree(tmp_dir)
+