require File.join(File.dirname(__FILE__), 'test_helper')

class TestJail < Test::Unit::TestCase
  def setup
    @article = Article.new.to_jail
    @comment = @article.comments.first
    @comment_class = Comment.to_jail
  end

  def test_explicitly_allowed_instance_methods_should_be_accessible
    assert_nothing_raised { @article.title }
  end

  def test_explicitly_allowed_class_methods_should_be_accessible
    assert_nothing_raised { @comment_class.all(1) }
  end

  def test_jail_instance_methods_should_be_accessible
    assert_nothing_raised { @article.author_name }
  end

  def test_sending_to_jail_to_an_object_should_return_a_jail
    assert_equal "Article::Jail", @article.class.name
  end

  def test_jail_instances_should_have_limited_methods
    expected = ["class", "method_missing", "methods", "respond_to?", "respond_to_missing?", "to_jail", "to_s", "instance_variable_get"]
    expected.delete('respond_to_missing?') if RUBY_VERSION > '1.9.3' # respond_to_missing? is private in rubies above 1.9.3
    objects.each do |object|
      assert_equal expected.sort, reject_pretty_methods(object.to_jail.methods.map(&:to_s).sort)
    end
  end

  def test_jail_classes_should_have_limited_methods
    expected = ["new", "methods", "name", "inherited", "method_added",
                "allow", "allowed?", "allowed_methods", "init_allowed_methods",
                "allow_instance_method", "allow_class_method", "allowed_instance_method?",
                "allowed_class_method?", "allowed_instance_methods", "allowed_class_methods",
                "<", # < needed in Rails Object#subclasses_of
                "ancestors", "=="] # ancestors and == needed in Rails::Generator::Spec#lookup_class

    if defined?(JRUBY_VERSION)
      (expected << ['method_missing', 'singleton_method_undefined', 'singleton_method_added']).flatten!  # needed for running under jruby
    end

    objects.each do |object|
      assert_equal expected.sort, reject_pretty_methods(object.to_jail.class.methods.map(&:to_s).sort)
    end
  end

  def test_allowed_methods_should_be_propagated_to_subclasses
    assert_equal Article::Jail.allowed_methods, Article::ExtendedJail.allowed_methods
  end

  def test_respond_to_works_correctly
    assert @article.respond_to?(:title)
    assert !@article.respond_to?(:bogus)
  end

  private

  def objects
    [[], {}, 1..2, "a", :a, Time.now, 1, 1.0, nil, false, true, Comment]
  end

  def reject_pretty_methods(methods)
    methods.reject{ |method| method =~ /^pretty_/ }
  end

end