RubyGems - maestrano-rails - Versions diffs - 0.1.0 - Mend

maestrano-rails 0.1.0

Files changed (96) hide show

data/LICENSE +21 -0
data/README.md +164 -0
data/Rakefile +38 -0
data/app/controllers/maestrano/rails/saml_base_controller.rb +39 -0
data/lib/generators/active_record/maestrano_group_generator.rb +38 -0
data/lib/generators/active_record/maestrano_user_generator.rb +38 -0
data/lib/generators/active_record/templates/migration.rb +12 -0
data/lib/generators/maestrano/USAGE +2 -0
data/lib/generators/maestrano/group_generator.rb +11 -0
data/lib/generators/maestrano/install_generator.rb +30 -0
data/lib/generators/maestrano/orm_helpers.rb +75 -0
data/lib/generators/maestrano/templates/maestrano.rb +84 -0
data/lib/generators/maestrano/templates/saml_controller.rb +52 -0
data/lib/generators/maestrano/user_generator.rb +11 -0
data/lib/generators/mongoid/maestrano_group_generator.rb +26 -0
data/lib/generators/mongoid/maestrano_user_generator.rb +26 -0
data/lib/maestrano/rails/controllers/maestrano_security.rb +32 -0
data/lib/maestrano/rails/models/maestrano_auth_resource.rb +96 -0
data/lib/maestrano/rails/version.rb +5 -0
data/lib/maestrano/rails.rb +10 -0
data/lib/maestrano-rails.rb +1 -0
data/test/controllers/generic_controller_test.rb +54 -0
data/test/controllers/saml_controller_test.rb +117 -0
data/test/dummy/README.rdoc +261 -0
data/test/dummy/Rakefile +7 -0
data/test/dummy/app/assets/javascripts/application.js +15 -0
data/test/dummy/app/assets/javascripts/pages.js +2 -0
data/test/dummy/app/assets/stylesheets/application.css +13 -0
data/test/dummy/app/assets/stylesheets/pages.css +4 -0
data/test/dummy/app/controllers/application_controller.rb +3 -0
data/test/dummy/app/controllers/maestrano/auth/saml_controller.rb +14 -0
data/test/dummy/app/controllers/pages_controller.rb +4 -0
data/test/dummy/app/helpers/application_helper.rb +2 -0
data/test/dummy/app/helpers/pages_helper.rb +2 -0
data/test/dummy/app/models/admin/monster.rb +2 -0
data/test/dummy/app/models/admin.rb +5 -0
data/test/dummy/app/models/mno_crew.rb +7 -0
data/test/dummy/app/models/mno_monster.rb +9 -0
data/test/dummy/app/models/monster.rb +2 -0
data/test/dummy/app/views/layouts/application.html.erb +14 -0
data/test/dummy/app/views/pages/home.html.erb +2 -0
data/test/dummy/config/application.rb +56 -0
data/test/dummy/config/boot.rb +10 -0
data/test/dummy/config/database.yml +25 -0
data/test/dummy/config/environment.rb +5 -0
data/test/dummy/config/environments/development.rb +37 -0
data/test/dummy/config/environments/production.rb +67 -0
data/test/dummy/config/environments/test.rb +37 -0
data/test/dummy/config/initializers/backtrace_silencers.rb +7 -0
data/test/dummy/config/initializers/inflections.rb +15 -0
data/test/dummy/config/initializers/maestrano.rb +84 -0
data/test/dummy/config/initializers/mime_types.rb +5 -0
data/test/dummy/config/initializers/secret_token.rb +7 -0
data/test/dummy/config/initializers/session_store.rb +8 -0
data/test/dummy/config/initializers/wrap_parameters.rb +14 -0
data/test/dummy/config/locales/en.yml +5 -0
data/test/dummy/config/routes.rb +70 -0
data/test/dummy/config.ru +4 -0
data/test/dummy/db/development.sqlite3 +0 -0
data/test/dummy/db/migrate/20140526125222_create_monsters.rb +8 -0
data/test/dummy/db/migrate/20140526125242_create_admin_monsters.rb +8 -0
data/test/dummy/db/migrate/20140526144828_create_mno_monsters.rb +13 -0
data/test/dummy/db/migrate/20140526151139_create_mno_crews.rb +11 -0
data/test/dummy/db/schema.rb +44 -0
data/test/dummy/db/test.sqlite3 +0 -0
data/test/dummy/log/development.log +76 -0
data/test/dummy/log/test.log +3732 -0
data/test/dummy/public/404.html +26 -0
data/test/dummy/public/422.html +26 -0
data/test/dummy/public/500.html +25 -0
data/test/dummy/public/favicon.ico +0 -0
data/test/dummy/script/rails +6 -0
data/test/dummy/test/fixtures/admin/monsters.yml +11 -0
data/test/dummy/test/fixtures/mno_crews.yml +11 -0
data/test/dummy/test/fixtures/mno_monsters.yml +15 -0
data/test/dummy/test/fixtures/monsters.yml +11 -0
data/test/dummy/test/functional/pages_controller_test.rb +9 -0
data/test/dummy/test/unit/admin/monster_test.rb +7 -0
data/test/dummy/test/unit/helpers/pages_helper_test.rb +4 -0
data/test/dummy/test/unit/mno_crew_test.rb +7 -0
data/test/dummy/test/unit/mno_monster_test.rb +7 -0
data/test/dummy/test/unit/monster_test.rb +7 -0
data/test/generators/group/active_record_generator_test.rb +79 -0
data/test/generators/group/mongoid_generator_test.rb +76 -0
data/test/generators/group_generator_test.rb +39 -0
data/test/generators/install_generator_test.rb +36 -0
data/test/generators/user/active_record_generator_test.rb +79 -0
data/test/generators/user/mongoid_generator_test.rb +76 -0
data/test/generators/user_generator_test.rb +39 -0
data/test/maestrano-rails_test.rb +7 -0
data/test/models/maestrano_group_via_test.rb +66 -0
data/test/models/maestrano_user_via_test.rb +70 -0
data/test/test_files/config/routes.rb +58 -0
data/test/test_helper.rb +24 -0
data/test/tmp/app/models/monster.rb +20 -0
metadata +287 -0

data/test/dummy/public/404.html ADDED Viewed

@@ -0,0 +1,26 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>The page you were looking for doesn't exist (404)</title>
+  <style type="text/css">
+    body { background-color: #fff; color: #666; text-align: center; font-family: arial, sans-serif; }
+    div.dialog {
+      width: 25em;
+      padding: 0 4em;
+      margin: 4em auto 0 auto;
+      border: 1px solid #ccc;
+      border-right-color: #999;
+      border-bottom-color: #999;
+    }
+    h1 { font-size: 100%; color: #f00; line-height: 1.5em; }
+  </style>
+</head>
+<body>
+  <!-- This file lives in public/404.html -->
+  <div class="dialog">
+    <h1>The page you were looking for doesn't exist.</h1>
+    <p>You may have mistyped the address or the page may have moved.</p>
+  </div>
+</body>
+</html>

data/test/dummy/public/422.html ADDED Viewed

@@ -0,0 +1,26 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>The change you wanted was rejected (422)</title>
+  <style type="text/css">
+    body { background-color: #fff; color: #666; text-align: center; font-family: arial, sans-serif; }
+    div.dialog {
+      width: 25em;
+      padding: 0 4em;
+      margin: 4em auto 0 auto;
+      border: 1px solid #ccc;
+      border-right-color: #999;
+      border-bottom-color: #999;
+    }
+    h1 { font-size: 100%; color: #f00; line-height: 1.5em; }
+  </style>
+</head>
+<body>
+  <!-- This file lives in public/422.html -->
+  <div class="dialog">
+    <h1>The change you wanted was rejected.</h1>
+    <p>Maybe you tried to change something you didn't have access to.</p>
+  </div>
+</body>
+</html>

data/test/dummy/public/500.html ADDED Viewed

@@ -0,0 +1,25 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>We're sorry, but something went wrong (500)</title>
+  <style type="text/css">
+    body { background-color: #fff; color: #666; text-align: center; font-family: arial, sans-serif; }
+    div.dialog {
+      width: 25em;
+      padding: 0 4em;
+      margin: 4em auto 0 auto;
+      border: 1px solid #ccc;
+      border-right-color: #999;
+      border-bottom-color: #999;
+    }
+    h1 { font-size: 100%; color: #f00; line-height: 1.5em; }
+  </style>
+</head>
+<body>
+  <!-- This file lives in public/500.html -->
+  <div class="dialog">
+    <h1>We're sorry, but something went wrong.</h1>
+  </div>
+</body>
+</html>

data/test/dummy/public/favicon.ico ADDED Viewed

File without changes

data/test/dummy/script/rails ADDED Viewed

@@ -0,0 +1,6 @@
+#!/usr/bin/env ruby
+# This command will automatically be run when you run "rails" with Rails 3 gems installed from the root of your application.
+APP_PATH = File.expand_path('../../config/application',  __FILE__)
+require File.expand_path('../../config/boot',  __FILE__)
+require 'rails/commands'

data/test/dummy/test/fixtures/admin/monsters.yml ADDED Viewed

@@ -0,0 +1,11 @@
+# Read about fixtures at http://api.rubyonrails.org/classes/ActiveRecord/Fixtures.html
+# This model initially had no columns defined.  If you add columns to the
+# model remove the '{}' from the fixture names and add the columns immediately
+# below each fixture, per the syntax in the comments below
+#
+one: {}
+# column: value
+#
+two: {}
+#  column: value

data/test/dummy/test/fixtures/mno_crews.yml ADDED Viewed

@@ -0,0 +1,11 @@
+# Read about fixtures at http://api.rubyonrails.org/classes/ActiveRecord/Fixtures.html
+one:
+  name: MyString
+  provider: MyString
+  uid: MyString
+two:
+  name: MyString
+  provider: MyString
+  uid: MyString

data/test/dummy/test/fixtures/mno_monsters.yml ADDED Viewed

@@ -0,0 +1,15 @@
+# Read about fixtures at http://api.rubyonrails.org/classes/ActiveRecord/Fixtures.html
+one:
+  first_name: MyString
+  last_name: MyString
+  email: MyString
+  provider: MyString
+  uid: MyString
+two:
+  first_name: MyString
+  last_name: MyString
+  email: MyString
+  provider: MyString
+  uid: MyString

data/test/dummy/test/fixtures/monsters.yml ADDED Viewed

@@ -0,0 +1,11 @@
+# Read about fixtures at http://api.rubyonrails.org/classes/ActiveRecord/Fixtures.html
+# This model initially had no columns defined.  If you add columns to the
+# model remove the '{}' from the fixture names and add the columns immediately
+# below each fixture, per the syntax in the comments below
+#
+one: {}
+# column: value
+#
+two: {}
+#  column: value

data/test/dummy/test/functional/pages_controller_test.rb ADDED Viewed

@@ -0,0 +1,9 @@
+require 'test_helper'
+class PagesControllerTest < ActionController::TestCase
+  test "should get home" do
+    get :home
+    assert_response :success
+  end
+end

data/test/dummy/test/unit/admin/monster_test.rb ADDED Viewed

@@ -0,0 +1,7 @@
+require 'test_helper'
+class Admin::MonsterTest < ActiveSupport::TestCase
+  # test "the truth" do
+  #   assert true
+  # end
+end

data/test/dummy/test/unit/helpers/pages_helper_test.rb ADDED Viewed

@@ -0,0 +1,4 @@
+require 'test_helper'
+class PagesHelperTest < ActionView::TestCase
+end

data/test/dummy/test/unit/mno_crew_test.rb ADDED Viewed

@@ -0,0 +1,7 @@
+require 'test_helper'
+class MnoCrewTest < ActiveSupport::TestCase
+  # test "the truth" do
+  #   assert true
+  # end
+end

data/test/dummy/test/unit/mno_monster_test.rb ADDED Viewed

@@ -0,0 +1,7 @@
+require 'test_helper'
+class MnoMonsterTest < ActiveSupport::TestCase
+  # test "the truth" do
+  #   assert true
+  # end
+end

data/test/dummy/test/unit/monster_test.rb ADDED Viewed

@@ -0,0 +1,7 @@
+require 'test_helper'
+class MonsterTest < ActiveSupport::TestCase
+  # test "the truth" do
+  #   assert true
+  # end
+end

data/test/generators/group/active_record_generator_test.rb ADDED Viewed

@@ -0,0 +1,79 @@
+require "test_helper"
+if TEST_ORM == :active_record
+  require "generators/active_record/maestrano_group_generator"
+  class ActiveRecordGroupGeneratorTest < Rails::Generators::TestCase
+    tests ActiveRecord::Generators::MaestranoGroupGenerator
+    destination File.expand_path("../../../tmp", __FILE__)
+    setup :prepare_destination
+    context "normal model" do
+      setup { copy_model('monster') }
+      should "create the right migration" do
+        run_generator %w(monster)
+        assert_migration "db/migrate/add_maestrano_to_monsters.rb", /def self.up/
+      end
+      should "edit the model content" do
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb", /maestrano_group_via/
+      end
+      should "not include parameters in rails 4" do
+        ActiveRecord::Generators::MaestranoGroupGenerator.any_instance.stubs(:rails_3?).returns(false)
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb" do |content|
+          assert_no_match /attr_protected :provider/, content
+        end
+      end
+      should "not include parameters if rails 3 and strong_parameter present" do
+        ActiveRecord::Generators::MaestranoGroupGenerator.any_instance.stubs(:rails_3?).returns(true)
+        ActiveRecord::Generators::MaestranoGroupGenerator.any_instance.stubs(:strong_parameters_enabled?).returns(true)
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb" do |content|
+          assert_no_match /attr_protected :provider/, content
+        end
+      end
+      should "include parameters if rails 3 and strong_parameter not present" do
+        ActiveRecord::Generators::MaestranoGroupGenerator.any_instance.stubs(:rails_3?).returns(true)
+        ActiveRecord::Generators::MaestranoGroupGenerator.any_instance.stubs(:strong_parameters_enabled?).returns(false)
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb" do |content|
+          assert_match /attr_protected :provider/, content
+        end
+      end
+    end
+    context "namespaced model" do
+      setup { copy_model('admin/monster') }
+      should "create the right migration" do
+        run_generator %w(admin/monster)
+        assert_migration "db/migrate/add_maestrano_to_admin_monsters.rb", /def self.up/
+      end
+      should "edit the model content" do
+        run_generator %w(admin/monster)
+        assert_file "app/models/admin/monster.rb", /maestrano_group_via/
+      end
+    end
+    # Copy model like: 'monster' or 'admin/monster'
+    def copy_model(relative_model_path)
+      model_path = File.expand_path("../../../dummy/app/models/#{relative_model_path}.rb", __FILE__)
+      if (path_ary = relative_model_path.split('/')) && path_ary.size > 1
+        destination = File.join(destination_root, "app", "models",*path_ary[0..-2])
+      else
+        destination = File.join(destination_root, "app", "models")
+      end
+      FileUtils.mkdir_p(destination)
+      FileUtils.cp model_path, destination
+    end
+  end
+end

data/test/generators/group/mongoid_generator_test.rb ADDED Viewed

@@ -0,0 +1,76 @@
+require "test_helper"
+if TEST_ORM == :mongoid
+  require "generators/mongoid/maestrano_group_generator"
+  class MongoidGroupGeneratorTest < Rails::Generators::TestCase
+    tests Mongoid::Generators::MaestranoGroupGenerator
+    destination File.expand_path("../../../tmp", __FILE__)
+    setup :prepare_destination
+    context "normal model" do
+      setup { copy_model('monster') }
+      should "edit the model content" do
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb", /maestrano_group_via/
+      end
+      should "not include parameters in rails 4" do
+        Mongoid::Generators::MaestranoGroupGenerator.any_instance.stubs(:rails_3?).returns(false)
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb" do |content|
+          assert_no_match /attr_protected :provider/, content
+        end
+      end
+      should "not include parameters if rails 3 and strong_parameter present" do
+        Mongoid::Generators::MaestranoGroupGenerator.any_instance.stubs(:rails_3?).returns(true)
+        Mongoid::Generators::MaestranoGroupGenerator.any_instance.stubs(:strong_parameters_enabled?).returns(true)
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb" do |content|
+          assert_no_match /attr_protected :provider/, content
+        end
+      end
+      should "include parameters if rails 3 and strong_parameter not present" do
+        Mongoid::Generators::MaestranoGroupGenerator.any_instance.stubs(:rails_3?).returns(true)
+        Mongoid::Generators::MaestranoGroupGenerator.any_instance.stubs(:strong_parameters_enabled?).returns(false)
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb" do |content|
+          assert_match /attr_protected :provider/, content
+        end
+      end
+    end
+    context "namespaced model" do
+      setup { copy_model('admin/monster') }
+      should "edit the model content" do
+        run_generator %w(admin/monster)
+        assert_file "app/models/admin/monster.rb", /maestrano_group_via/
+      end
+    end
+    # Copy model like: 'monster' or 'admin/monster'
+    def copy_model(relative_model_path)
+      model_path = File.expand_path("../../../dummy/app/models/#{relative_model_path}.rb", __FILE__)
+      if (path_ary = relative_model_path.split('/')) && path_ary.size > 1
+        destination = File.join(destination_root, "app", "models",*path_ary[0..-2])
+      else
+        destination = File.join(destination_root, "app", "models")
+      end
+      FileUtils.mkdir_p(destination)
+      FileUtils.cp model_path, destination
+      # Replace class name
+      class_name = path_ary.map(&:capitalize).join("::")
+      full_path = File.join(destination,"#{path_ary[-1]}.rb")
+      text = File.read(full_path)
+      replace = text.gsub("class #{class_name} < ActiveRecord::Base", "class #{class_name}\ninclude Mongoid::Document\n")
+      File.open(full_path, "w") {|file| file.puts replace}
+    end
+  end
+end

data/test/generators/group_generator_test.rb ADDED Viewed

@@ -0,0 +1,39 @@
+require 'test_helper'
+class GroupGeneratorTest < Rails::Generators::TestCase
+  tests Maestrano::Generators::GroupGenerator
+  destination File.expand_path("../../tmp", __FILE__)
+  setup :prepare_destination
+  should "call the active_record migration generator successfully" do
+    run_generator %w(monster --orm=active_record)
+    assert_migration "db/migrate/add_maestrano_to_monsters.rb", /def self.up/
+  end
+  should "call the mongoid migration generator successfully" do
+    copy_model('monster')
+    run_generator %w(monster --orm=mongoid)
+    assert_file "app/models/monster.rb", /maestrano_group_via/
+  end
+  # Copy model like: 'monster' or 'admin/monster'
+  def copy_model(relative_model_path)
+    model_path = File.expand_path("../../dummy/app/models/#{relative_model_path}.rb", __FILE__)
+    if (path_ary = relative_model_path.split('/')) && path_ary.size > 1
+      destination = File.join(destination_root, "app", "models",*path_ary[0..-2])
+    else
+      destination = File.join(destination_root, "app", "models")
+    end
+    FileUtils.mkdir_p(destination)
+    FileUtils.cp model_path, destination
+    # Replace class name
+    class_name = path_ary.map(&:capitalize).join("::")
+    full_path = File.join(destination,"#{path_ary[-1]}.rb")
+    text = File.read(full_path)
+    replace = text.gsub("class #{class_name} < ActiveRecord::Base", "class #{class_name}\ninclude Mongoid::Document\n")
+    File.open(full_path, "w") {|file| file.puts replace}
+  end
+end

data/test/generators/install_generator_test.rb ADDED Viewed

@@ -0,0 +1,36 @@
+require 'test_helper'
+class InstallGeneratorTest < Rails::Generators::TestCase
+  tests Maestrano::Generators::InstallGenerator
+  destination File.expand_path("../../tmp", __FILE__)
+  setup do
+    prepare_destination
+    copy_routes
+  end
+  should "create the maestrano initializer" do
+    run_generator
+    assert_file "config/initializers/maestrano.rb"
+  end
+  should "create the maestrano/auth/saml controller" do
+    run_generator
+    assert_file "app/controllers/maestrano/auth/saml_controller.rb"
+  end
+  should "create the maestrano routes" do
+    run_generator
+    match = /namespace\s+:maestrano\s+do\n\s*namespace\s+:auth\s+do\n\s*resources\s+:saml,\s+only:\[\]\s+do\n\s*get 'init',\son:\s:collection\n\s*post\s'consume',\s+on:\s+:collection(\n\s*end){3}/
+    assert_file "config/routes.rb", match
+  end
+  def copy_routes
+    routes = File.expand_path("../../test_files/config/routes.rb", __FILE__)
+    destination = File.join(destination_root, "config")
+    FileUtils.mkdir_p(destination)
+    FileUtils.cp routes, destination
+  end
+end

data/test/generators/user/active_record_generator_test.rb ADDED Viewed

@@ -0,0 +1,79 @@
+require "test_helper"
+if TEST_ORM == :active_record
+  require "generators/active_record/maestrano_user_generator"
+  class ActiveRecordUserGeneratorTest < Rails::Generators::TestCase
+    tests ActiveRecord::Generators::MaestranoUserGenerator
+    destination File.expand_path("../../../tmp", __FILE__)
+    setup :prepare_destination
+    context "normal model" do
+      setup { copy_model('monster') }
+      should "create the right migration" do
+        run_generator %w(monster)
+        assert_migration "db/migrate/add_maestrano_to_monsters.rb", /def self.up/
+      end
+      should "edit the model content" do
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb", /maestrano_user_via/
+      end
+      should "not include parameters in rails 4" do
+        ActiveRecord::Generators::MaestranoUserGenerator.any_instance.stubs(:rails_3?).returns(false)
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb" do |content|
+          assert_no_match /attr_protected :provider/, content
+        end
+      end
+      should "not include parameters if rails 3 and strong_parameter present" do
+        ActiveRecord::Generators::MaestranoUserGenerator.any_instance.stubs(:rails_3?).returns(true)
+        ActiveRecord::Generators::MaestranoUserGenerator.any_instance.stubs(:strong_parameters_enabled?).returns(true)
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb" do |content|
+          assert_no_match /attr_protected :provider/, content
+        end
+      end
+      should "include parameters if rails 3 and strong_parameter not present" do
+        ActiveRecord::Generators::MaestranoUserGenerator.any_instance.stubs(:rails_3?).returns(true)
+        ActiveRecord::Generators::MaestranoUserGenerator.any_instance.stubs(:strong_parameters_enabled?).returns(false)
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb" do |content|
+          assert_match /attr_protected :provider/, content
+        end
+      end
+    end
+    context "namespaced model" do
+      setup { copy_model('admin/monster') }
+      should "create the right migration" do
+        run_generator %w(admin/monster)
+        assert_migration "db/migrate/add_maestrano_to_admin_monsters.rb", /def self.up/
+      end
+      should "edit the model content" do
+        run_generator %w(admin/monster)
+        assert_file "app/models/admin/monster.rb", /maestrano_user_via/
+      end
+    end
+    # Copy model like: 'monster' or 'admin/monster'
+    def copy_model(relative_model_path)
+      model_path = File.expand_path("../../../dummy/app/models/#{relative_model_path}.rb", __FILE__)
+      if (path_ary = relative_model_path.split('/')) && path_ary.size > 1
+        destination = File.join(destination_root, "app", "models",*path_ary[0..-2])
+      else
+        destination = File.join(destination_root, "app", "models")
+      end
+      FileUtils.mkdir_p(destination)
+      FileUtils.cp model_path, destination
+    end
+  end
+end

data/test/generators/user/mongoid_generator_test.rb ADDED Viewed

@@ -0,0 +1,76 @@
+require "test_helper"
+if TEST_ORM == :mongoid
+  require "generators/mongoid/maestrano_user_generator"
+  class MongoidUserGeneratorTest < Rails::Generators::TestCase
+    tests Mongoid::Generators::MaestranoUserGenerator
+    destination File.expand_path("../../../tmp", __FILE__)
+    setup :prepare_destination
+    context "normal model" do
+      setup { copy_model('monster') }
+      should "edit the model content" do
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb", /maestrano_user_via/
+      end
+      should "not include parameters in rails 4" do
+        Mongoid::Generators::MaestranoUserGenerator.any_instance.stubs(:rails_3?).returns(false)
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb" do |content|
+          assert_no_match /attr_protected :provider/, content
+        end
+      end
+      should "not include parameters if rails 3 and strong_parameter present" do
+        Mongoid::Generators::MaestranoUserGenerator.any_instance.stubs(:rails_3?).returns(true)
+        Mongoid::Generators::MaestranoUserGenerator.any_instance.stubs(:strong_parameters_enabled?).returns(true)
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb" do |content|
+          assert_no_match /attr_protected :provider/, content
+        end
+      end
+      should "include parameters if rails 3 and strong_parameter not present" do
+        Mongoid::Generators::MaestranoUserGenerator.any_instance.stubs(:rails_3?).returns(true)
+        Mongoid::Generators::MaestranoUserGenerator.any_instance.stubs(:strong_parameters_enabled?).returns(false)
+        run_generator %w(monster)
+        assert_file "app/models/monster.rb" do |content|
+          assert_match /attr_protected :provider/, content
+        end
+      end
+    end
+    context "namespaced model" do
+      setup { copy_model('admin/monster') }
+      should "edit the model content" do
+        run_generator %w(admin/monster)
+        assert_file "app/models/admin/monster.rb", /maestrano_user_via/
+      end
+    end
+    # Copy model like: 'monster' or 'admin/monster'
+    def copy_model(relative_model_path)
+      model_path = File.expand_path("../../../dummy/app/models/#{relative_model_path}.rb", __FILE__)
+      if (path_ary = relative_model_path.split('/')) && path_ary.size > 1
+        destination = File.join(destination_root, "app", "models",*path_ary[0..-2])
+      else
+        destination = File.join(destination_root, "app", "models")
+      end
+      FileUtils.mkdir_p(destination)
+      FileUtils.cp model_path, destination
+      # Replace class name
+      class_name = path_ary.map(&:capitalize).join("::")
+      full_path = File.join(destination,"#{path_ary[-1]}.rb")
+      text = File.read(full_path)
+      replace = text.gsub("class #{class_name} < ActiveRecord::Base", "class #{class_name}\ninclude Mongoid::Document\n")
+      File.open(full_path, "w") {|file| file.puts replace}
+    end
+  end
+end

data/test/generators/user_generator_test.rb ADDED Viewed

@@ -0,0 +1,39 @@
+require 'test_helper'
+class UserGeneratorTest < Rails::Generators::TestCase
+  tests Maestrano::Generators::UserGenerator
+  destination File.expand_path("../../tmp", __FILE__)
+  setup :prepare_destination
+  should "call the active_record migration generator successfully" do
+    run_generator %w(monster --orm=active_record)
+    assert_migration "db/migrate/add_maestrano_to_monsters.rb", /def self.up/
+  end
+  should "call the mongoid migration generator successfully" do
+    copy_model('monster')
+    run_generator %w(monster --orm=mongoid)
+    assert_file "app/models/monster.rb", /maestrano_user_via/
+  end
+  # Copy model like: 'monster' or 'admin/monster'
+  def copy_model(relative_model_path)
+    model_path = File.expand_path("../../dummy/app/models/#{relative_model_path}.rb", __FILE__)
+    if (path_ary = relative_model_path.split('/')) && path_ary.size > 1
+      destination = File.join(destination_root, "app", "models",*path_ary[0..-2])
+    else
+      destination = File.join(destination_root, "app", "models")
+    end
+    FileUtils.mkdir_p(destination)
+    FileUtils.cp model_path, destination
+    # Replace class name
+    class_name = path_ary.map(&:capitalize).join("::")
+    full_path = File.join(destination,"#{path_ary[-1]}.rb")
+    text = File.read(full_path)
+    replace = text.gsub("class #{class_name} < ActiveRecord::Base", "class #{class_name}\ninclude Mongoid::Document\n")
+    File.open(full_path, "w") {|file| file.puts replace}
+  end
+end

data/test/maestrano-rails_test.rb ADDED Viewed

@@ -0,0 +1,7 @@
+require 'test_helper'
+class MaestranoRailsTest < ActiveSupport::TestCase
+  test "truth" do
+    assert_kind_of Module, Maestrano::Rails
+  end
+end