correct type check

crunchy · Jun 28, 2012 · e476c9b · e476c9b
1 parent 1e7dc36
commit e476c9b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/lib/arturo.rb b/lib/arturo.rb
@@ -22,7 +22,7 @@ def feature_enabled_for?(feature_name, recipient)
 
     def feature_enabled?(feature_name)
       f = self::Feature.to_feature(feature_name)
-      method_missing :feature_enabled?, feature_name unless Arturo::GlobalFeature === f.class
+      method_missing :feature_enabled?, feature_name unless f.class == Arturo::GlobalFeature
 
       f && f.enabled?
     end

diff --git a/test/dummy_app/test/unit/global_feature_test.rb b/test/dummy_app/test/unit/global_feature_test.rb
@@ -13,6 +13,11 @@ def test_correct_class_name_used
     assert_equal feature.class_name, "Arturo::GlobalFeature"
   end
 
+  def test_responds_to_feature_enabled_helper
+    feature.enable!
+    assert ::Arturo.feature_enabled?(feature.symbol)
+  end
+
   def test_to_feature_finds_global_feature
     assert_equal feature, ::Arturo::Feature.to_feature(feature)
     assert_equal feature, ::Arturo::Feature.to_feature(feature.symbol)