test/test_scan_cms.rb in yawast-0.2.0.beta2 vs test/test_scan_cms.rb in yawast-0.2.0.beta3

@@ -1,23 +1,23 @@
 require 'minitest/autorun'
-require './lib/yawast'
-require './test/base'
+require File.dirname(__FILE__) + '/../lib/yawast'
+require File.dirname(__FILE__) + '/base'
 
 class TestScannerCms < Minitest::Test
   include TestBase
 
   def test_generator_tag_valid
-    body = File.read('test/data/cms_wordpress_body.txt')
+    body = File.read(File.dirname(__FILE__) + '/data/cms_wordpress_body.txt')
     override_stdout
     Yawast::Scanner::Cms.get_generator body
 
     assert stdout_value.include?('WordPress'), "Unexpected generator tag: #{stdout_value}"
 
     restore_stdout
   end
 
   def test_generator_tag_invalid
-    body = File.read('test/data/cms_none_body.txt')
+    body = File.read(File.dirname(__FILE__) + '/data/cms_none_body.txt')
     override_stdout
     Yawast::Scanner::Cms.get_generator body
 
     assert stdout_value == '', "Unexpected generator tag: #{stdout_value}"