test/rbbt/util/test_data_module.rb in rbbt-util-1.1.0 vs test/rbbt/util/test_data_module.rb in rbbt-util-1.2.1

@@ -2,10 +2,16 @@
 require 'rbbt'
 require 'rbbt/util/data_module'
 require 'test/unit'
 require 'fileutils'
 
+SHAREDIR = File.join(PKGData.sharedir_for_file(__FILE__), 'install/DataTest')
+FileUtils.mkdir_p SHAREDIR
+File.open(File.join(SHAREDIR, 'Rakefile'), 'w') do |f|
+  f.puts "file :file1 do |t| File.open(t.name, 'w') do |f| f.write 'File 1' end end"
+end
+
 module DataTest  
   extend DataModule
 
   def self.salute(name)
     "Hello #{name}"
@@ -14,29 +20,23 @@
   World = with_key("world")
 end
 
 class TestDataModule < Test::Unit::TestCase
 
-  SHAREDIR = File.join(PKGData.sharedir_for_file(__FILE__), 'install', 'DataTest')
-
   def setup
+
     FileUtils.mkdir_p SHAREDIR
     File.open(File.join(SHAREDIR, 'Rakefile'), 'w') do |f|
-      f.puts "task :file1 do |t| File.open(t.name, 'w') do |f| f.write 'File 1' end end"
+      f.puts "file :file1 do |t| File.open(t.name, 'w') do |f| f.write 'File 1' end end"
     end
   end
 
-  def test_module
-    assert_equal "File 1", Open.read(DataTest.file1).chomp
+  def test_rakefile
+    assert_equal "File 1", Rbbt.files.DataTest.file1.read
     assert_equal "Hello world", DataTest.salute("world")
-    assert_equal "Hello world", DataTest.salute("world")
     assert_equal "Hello world", DataTest::with_key("world").salute
     assert_equal "Hello world", DataTest::World.salute
     FileUtils.rm_rf File.join(Rbbt.datadir, 'DataTest')
-  end
-
-  def test_method_missing
-    assert_raise NoMethodError do DataTest.missing end
   end
 
   def teardown
     FileUtils.rm_rf SHAREDIR
   end