gigantron 0.1.1 → 0.1.2

data/test/test_generator_helper.rb

@@ -1,29 +1,29 @@
-begin
-  require File.dirname(__FILE__) + '/test_helper'
-rescue LoadError
-  require 'test/unit'
-end
-require 'fileutils'
-
-# Must set before requiring generator libs.
-TMP_ROOT = File.dirname(__FILE__) + "/tmp" unless defined?(TMP_ROOT)
-PROJECT_NAME = "myproject" unless defined?(PROJECT_NAME)
-app_root = File.join(TMP_ROOT, PROJECT_NAME)
-if defined?(APP_ROOT)
-  APP_ROOT.replace(app_root)
-else
-  APP_ROOT = app_root
-end
-if defined?(RAILS_ROOT)
-  RAILS_ROOT.replace(app_root)
-else
-  RAILS_ROOT = app_root
-end
-
-begin
-  require 'rubigen'
-rescue LoadError
-  require 'rubygems'
-  require 'rubigen'
-end
-require 'rubigen/helpers/generator_test_helper'
+begin
+  require File.dirname(__FILE__) + '/test_helper'
+rescue LoadError
+  require 'test/unit'
+end
+require 'fileutils'
+
+# Must set before requiring generator libs.
+TMP_ROOT = File.dirname(__FILE__) + "/tmp" unless defined?(TMP_ROOT)
+PROJECT_NAME = "myproject" unless defined?(PROJECT_NAME)
+app_root = File.join(TMP_ROOT, PROJECT_NAME)
+if defined?(APP_ROOT)
+  APP_ROOT.replace(app_root)
+else
+  APP_ROOT = app_root
+end
+if defined?(RAILS_ROOT)
+  RAILS_ROOT.replace(app_root)
+else
+  RAILS_ROOT = app_root
+end
+
+begin
+  require 'rubigen'
+rescue LoadError
+  require 'rubygems'
+  require 'rubigen'
+end
+require 'rubigen/helpers/generator_test_helper'

data/test/test_gigantron_generator.rb

@@ -1,114 +1,118 @@
-require File.join(File.dirname(__FILE__), "test_generator_helper.rb")
-
-class TestGigantronGenerator < Test::Unit::TestCase
-  include RubiGen::GeneratorTestHelper
-
-  def setup
-    bare_setup
-  end
-  
-  def teardown
-    bare_teardown
-  end
-  
-  should "generate correct directory and file structure" do
-    run_generator('gigantron', [APP_ROOT], sources)
-    assert_directory_exists "tasks/"
-    assert_generated_file   "tasks/import.rake"
-    assert_directory_exists "test/"
-    assert_generated_file   "test/test_helper.rb"
-    assert_directory_exists "test/tasks/"
-    assert_generated_file   "test/tasks/test_import.rb"
-    assert_directory_exists "test/models/"
-    assert_directory_exists "lib/"
-    assert_directory_exists "models/"
-    assert_directory_exists "db/"
-    assert_directory_exists "script/"
-    assert_generated_file   "script/generate"
-    assert_generated_file   "database.yml"
-    assert_generated_file   "Rakefile"
-    assert_generated_file   "initialize.rb"
-  end
-
-  context "Generated project" do
-    setup do
-      run_generator('gigantron', [APP_ROOT], sources)
-
-      silence_warnings { GTRON_ENV = :test }
-      ENV["GTRON_ENV"] = "test"
-
-      initialize_gigantron
-      configure_gigantron_db
-      get_db_conn(GTRON_ENV)
-    end
-
-    context "with migration and model" do
-      setup do
-        run_generator('model', ['Foo'], sources)
-        
-        create_test_migration
-        get_db_conn(GTRON_ENV)
-
-        ENV["VERBOSE"] = "false"
-        Gigantron.migrate_dbs
-      end
-
-      should "create test db" do
-        assert File.exists?("#{APP_ROOT}/db/test.sqlite3")
-      end
-
-      should "create table foos and populate" do
-        assert_equal Foo.find(:all).size, 2
-      end
-    end
-
-    context "with Rake tasks" do
-      setup do
-        @rake = Rake::Application.new
-        Rake.application = @rake
-        load "#{APP_ROOT}/tasks/import.rake"
-      end
-
-      teardown { Rake.application = nil }
-
-      should "run empty import task" do
-        assert @rake['import'].invoke
-      end
-    end
-  end
-  
-  private
-  def sources
-    [RubiGen::PathSource.new(:test, File.join(File.dirname(__FILE__),"..", generator_path)),
-    RubiGen::PathSource.new(:test_components, File.join(File.dirname(__FILE__),"..", "gigantron_generators"))]
-  end
-  
-  def generator_path
-    "app_generators"
-  end
-
-  def initialize_gigantron
-    require File.join(APP_ROOT, "initialize.rb")
-    require 'gigantron/migrator'
-  end
-  
-  def configure_gigantron_db
-    if !File.exists? "test/template_database.yml"
-      flunk "No Database Configuration. Configure test DB in 'test/template_database.yml'!"
-    end
-    File.open("#{APP_ROOT}/database.yml", "w") do |f|
-      f.write ERB.new(
-        File.read("test/template_database.yml")).result(binding)
-    end
-  end
-
-  def create_test_migration
-    if !File.exists? "#{APP_ROOT}/db/migrate/001_create_foos.rb"
-      run_generator('migration', ['CreateFoos'], sources)
-    end
- 
-    assert File.exists?("#{APP_ROOT}/db/migrate/001_create_foos.rb")
-    FileUtils.cp("test/template_migration.rb", 
-      "#{APP_ROOT}/db/migrate/001_create_foos.rb")
-  end
-end
+require File.join(File.dirname(__FILE__), "test_generator_helper.rb")
+
+class TestGigantronGenerator < Test::Unit::TestCase
+  include RubiGen::GeneratorTestHelper
+
+  def setup
+    bare_setup
+  end
+  
+  def teardown
+    bare_teardown
+  end
+  
+  should "generate correct directory and file structure" do
+    run_generator('gigantron', [APP_ROOT], sources)
+    assert_directory_exists "tasks/"
+    assert_generated_file   "tasks/import.rake"
+    assert_directory_exists "test/"
+    assert_generated_file   "test/test_helper.rb"
+    assert_directory_exists "test/tasks/"
+    assert_generated_file   "test/tasks/test_import.rb"
+    assert_directory_exists "test/models/"
+    assert_directory_exists "lib/"
+    assert_directory_exists "lib/shoulda/"
+    assert_generated_file   "lib/shoulda/general.rb"
+    assert_generated_file   "lib/shoulda/private_helpers.rb"
+    assert_generated_file   "lib/shoulda/active_record_helpers.rb"
+    assert_directory_exists "models/"
+    assert_directory_exists "db/"
+    assert_directory_exists "script/"
+    assert_generated_file   "script/generate"
+    assert_generated_file   "database.yml.example"
+    assert_generated_file   "Rakefile"
+    assert_generated_file   "initialize.rb"
+  end
+
+  context "Generated project" do
+    setup do
+      run_generator('gigantron', [APP_ROOT], sources)
+
+      silence_warnings { GTRON_ENV = :test }
+      ENV["GTRON_ENV"] = "test"
+
+      initialize_gigantron
+      configure_gigantron_db
+      get_db_conn(GTRON_ENV)
+    end
+
+    context "with migration and model" do
+      setup do
+        run_generator('model', ['Foo'], sources)
+        
+        create_test_migration
+        get_db_conn(GTRON_ENV)
+
+        ENV["VERBOSE"] = "false"
+        Gigantron.migrate_dbs
+      end
+
+      should "create test db" do
+        assert File.exists?("#{APP_ROOT}/db/test.sqlite3")
+      end
+
+      should "create table foos and populate" do
+        assert_equal Foo.find(:all).size, 2
+      end
+    end
+
+    context "with Rake tasks" do
+      setup do
+        @rake = Rake::Application.new
+        Rake.application = @rake
+        load "#{APP_ROOT}/tasks/import.rake"
+      end
+
+      teardown { Rake.application = nil }
+
+      should "run empty import task" do
+        assert @rake['import'].invoke
+      end
+    end
+  end
+  
+  private
+  def sources
+    [RubiGen::PathSource.new(:test, File.join(File.dirname(__FILE__),"..", generator_path)),
+    RubiGen::PathSource.new(:test_components, File.join(File.dirname(__FILE__),"..", "gigantron_generators"))]
+  end
+  
+  def generator_path
+    "app_generators"
+  end
+
+  def initialize_gigantron
+    require File.join(APP_ROOT, "initialize.rb")
+    require 'gigantron/migrator'
+  end
+  
+  def configure_gigantron_db
+    if !File.exists? "test/template_database.yml"
+      flunk "No Database Configuration. Configure test DB in 'test/template_database.yml'!"
+    end
+    File.open("#{APP_ROOT}/database.yml", "w") do |f|
+      f.write ERB.new(
+        File.read("test/template_database.yml")).result(binding)
+    end
+  end
+
+  def create_test_migration
+    if !File.exists? "#{APP_ROOT}/db/migrate/001_create_foos.rb"
+      run_generator('migration', ['CreateFoos'], sources)
+    end
+ 
+    assert File.exists?("#{APP_ROOT}/db/migrate/001_create_foos.rb")
+    FileUtils.cp("test/template_migration.rb", 
+      "#{APP_ROOT}/db/migrate/001_create_foos.rb")
+  end
+end

data/test/test_helper.rb

@@ -1,4 +1,4 @@
-require 'test/unit'
-require 'rubygems'
-require 'shoulda'
-require File.dirname(__FILE__) + '/../lib/gigantron'
+require 'test/unit'
+require 'rubygems'
+require 'shoulda'
+require File.dirname(__FILE__) + '/../lib/gigantron'

data/test/test_mapreduce_task_generator.rb

@@ -1,50 +1,50 @@
-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
+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

data/test/test_migration_generator.rb

@@ -1,49 +1,49 @@
-require File.join(File.dirname(__FILE__), "test_generator_helper.rb")
-
-
-class TestMigrationGenerator < 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 = "CreateFoo"
-    run_generator('migration', [name], sources)
-    assert_directory_exists "db/"
-    assert_directory_exists "db/migrate/"
-    assert_generated_file   "db/migrate/001_create_foo.rb"
-    name2 = "AlterFoo"
-    run_generator('migration', [name2], sources)
-    assert_generated_file  "db/migrate/002_alter_foo.rb"
-  end
-
-  private
-  def sources
-    [RubiGen::PathSource.new(:test, File.join(File.dirname(__FILE__),"..", generator_path))
-    ]
-  end
-
-  def generator_path
-    "gigantron_generators"
-  end
-end
+require File.join(File.dirname(__FILE__), "test_generator_helper.rb")
+
+
+class TestMigrationGenerator < 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 = "CreateFoo"
+    run_generator('migration', [name], sources)
+    assert_directory_exists "db/"
+    assert_directory_exists "db/migrate/"
+    assert_generated_file   "db/migrate/001_create_foo.rb"
+    name2 = "AlterFoo"
+    run_generator('migration', [name2], sources)
+    assert_generated_file  "db/migrate/002_alter_foo.rb"
+  end
+
+  private
+  def sources
+    [RubiGen::PathSource.new(:test, File.join(File.dirname(__FILE__),"..", generator_path))
+    ]
+  end
+
+  def generator_path
+    "gigantron_generators"
+  end
+end

data/test/test_model_generator.rb

@@ -1,53 +1,53 @@
-require File.join(File.dirname(__FILE__), "test_generator_helper.rb")
-
-
-class TestModelGenerator < 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 = "FooBaz"
-    run_generator('model', [name], sources)
-    assert_directory_exists "models/"
-    assert_generated_file   "models/foo_baz.rb"
-    assert_directory_exists "test/"
-    assert_directory_exists "test/models/"
-    assert_generated_file   "test/models/test_foo_baz.rb"
-
-    #check migration
-    assert_directory_exists "db/"
-    assert_directory_exists "db/migrate/"
-    assert_generated_file   "db/migrate/001_create_foo_bazs.rb"
-  end
-
-  private
-  def sources
-    [RubiGen::PathSource.new(:test, File.join(File.dirname(__FILE__),"..", generator_path))
-    ]
-  end
-
-  def generator_path
-    "gigantron_generators"
-  end
-end
+require File.join(File.dirname(__FILE__), "test_generator_helper.rb")
+
+
+class TestModelGenerator < 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 = "FooBaz"
+    run_generator('model', [name], sources)
+    assert_directory_exists "models/"
+    assert_generated_file   "models/foo_baz.rb"
+    assert_directory_exists "test/"
+    assert_directory_exists "test/models/"
+    assert_generated_file   "test/models/test_foo_baz.rb"
+
+    #check migration
+    assert_directory_exists "db/"
+    assert_directory_exists "db/migrate/"
+    assert_generated_file   "db/migrate/001_create_foo_bazs.rb"
+  end
+
+  private
+  def sources
+    [RubiGen::PathSource.new(:test, File.join(File.dirname(__FILE__),"..", generator_path))
+    ]
+  end
+
+  def generator_path
+    "gigantron_generators"
+  end
+end