test/test_helper.rb in stomp-1.2.7 vs test/test_helper.rb in stomp-1.2.8

@@ -1,12 +1,17 @@
 # -*- encoding: utf-8 -*-
 
-$:.unshift(File.join(File.dirname(__FILE__), "..", "lib"))
-
 require 'test/unit'
 require 'timeout'
-require 'stomp'
-require 'tlogger'
+
+if Kernel.respond_to?(:require_relative)
+  require_relative("../lib/stomp")
+  require_relative("tlogger")
+else
+  $:.unshift(File.join(File.dirname(__FILE__), "..", "lib"))
+  require 'stomp'
+  require 'tlogger'
+end
 
 begin
   dummy = RUBY_ENGINE
 rescue NameError => ne
   RUBY_ENGINE = "unknown"