schleyfox / gigantron

Framework for Data Processing

This URL has Read+Write access

gigantron / test / test_mapreduce_task_generator.rb
100644 51 lines (42 sloc) 1.731 kb
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
require File.join(File.dirname(__FILE__), "test_generator_helper.rb")
 
 
class TestMapreduceTaskGenerator < Test::Unit::TestCase
  include RubiGen::GeneratorTestHelper
 
  def setup
    bare_setup
  end
 
  def teardown
    bare_teardown
  end
 
  # Some generator-related assertions:
  # assert_generated_file(name, &block) # block passed the file contents
  # assert_directory_exists(name)
  # assert_generated_class(name, &block)
  # assert_generated_module(name, &block)
  # assert_generated_test_for(name, &block)
  # The assert_generated_(class|module|test_for) &block is passed the body of the class/module within the file
  # assert_has_method(body, *methods) # check that the body has a list of methods (methods with parentheses not supported yet)
  #
  # Other helper methods are:
  # app_root_files - put this in teardown to show files generated by the test method (e.g. p app_root_files)
  # bare_setup - place this in setup method to create the APP_ROOT folder for each test
  # bare_teardown - place this in teardown method to destroy the TMP_ROOT or APP_ROOT folder after each test
 
  def test_generator_without_options
    name = "myapp"
    run_generator('mapreduce_task', [name], sources)
    assert_directory_exists "mapreduce/"
    assert_generated_file "mapreduce/mr_#{name}.rb"
    assert_directory_exists "tasks/"
    assert_generated_file "tasks/#{name}.rake"
    assert_directory_exists "test/"
    assert_directory_exists "test/tasks/"
    assert_generated_file "test/tasks/test_#{name}.rb"
  end
 
  private
  def sources
    [RubiGen::PathSource.new(:test, File.join(File.dirname(__FILE__),"..", generator_path))
    ]
  end
 
  def generator_path
    "gigantron_generators"
  end
end