code cleanup, test added

lib/mimemagic.rb

@@ -10,8 +10,7 @@ class MimeMagic
   # Mime type by type string
   # Mime type by type string
   def initialize(type)
   def initialize(type)
     @type = type
     @type = type
-    @mediatype = @type.split('/')[0]
-    @subtype   = @type.split('/')[1]
+    @mediatype, @subtype = type.split('/', 2)
   end
   end
 
 
   # Add custom mime type. Arguments:
   # Add custom mime type. Arguments:

test/mimemagic_test.rb

@@ -2,6 +2,12 @@ require 'bacon'
 require 'mimemagic'
 require 'mimemagic'
 
 
 describe 'MimeMagic' do
 describe 'MimeMagic' do
+  it 'should have type, mediatype and subtype' do
+    MimeMagic.new('text/html').type.should.equal 'text/html'
+    MimeMagic.new('text/html').mediatype.should.equal 'text'
+    MimeMagic.new('text/html').subtype.should.equal 'html'
+  end
+
   it 'should have mediatype helpers' do
   it 'should have mediatype helpers' do
     MimeMagic.new('text/plain').should.be.text
     MimeMagic.new('text/plain').should.be.text
     MimeMagic.new('text/html').should.be.text
     MimeMagic.new('text/html').should.be.text