test/helper.rb in embulk-input-elasticsearch-0.3.5 vs test/helper.rb in embulk-input-elasticsearch-0.3.6

@@ -2,10 +2,16 @@
 
 require 'test/unit'
 
 # require 'embulk/java/bootstrap'
 require 'embulk'
-Embulk.setup
+begin
+  # Embulk ~> 0.8.x
+  Embulk.setup
+rescue NotImplementedError
+  # Embulk ~> 0.9.x
+  require 'embulk/java/bootstrap'
+end
 Embulk.logger = Embulk::Logger.new('/dev/null')
 
 APP_ROOT = File.expand_path('../', __dir__)
 TEST_ROOT = File.expand_path(File.dirname(__FILE__))