Enumerate MIME::Type objects instead of assoc arrays

mime-types · Apr 15, 2013 · 644259d · 644259d
1 parent fd53425
commit 644259d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/lib/mime/types.rb b/lib/mime/types.rb
@@ -597,12 +597,13 @@ def defined_types #:nodoc:
       @type_variants.values.flatten
     end
 
+    # Returns the number of known types. A shortcut of MIME::Types[//].size
     def count
-      @type_variants.size
+      defined_types.size
     end
 
     def each
-      @type_variants.each { |t| yield t }
+     defined_types.each { |t| yield t }
     end
 
     @__types__ = self.new(VERSION)

diff --git a/test/test_mime_types.rb b/test/test_mime_types.rb
@@ -72,7 +72,7 @@ def test_class_of
   end
 
   def test_class_enumerable
-    assert( MIME::Types.any? {|types| types[0] == 'text/plain'} )
+    assert( MIME::Types.any? {|type| type.content_type == 'text/plain'} )
   end
 
   def test_class_count