test for fitting media types

leandrosilva · Dec 13, 2009 · d34c1fd · d34c1fd
1 parent cf8086a
commit d34c1fd
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/lib/restfulie/server/restfulie_controller.rb b/lib/restfulie/server/restfulie_controller.rb
@@ -5,10 +5,7 @@ module Controller
     def create
 
       type = model_for_this_controller
-      unless Restfulie::MediaType.supports?(request.headers['CONTENT_TYPE']) ||
-              Restfulie::MediaType.media_type(request.headers['CONTENT_TYPE']) == type
-        return head 415
-      end
+      return head 415 unless fits_content(type, request.headers['CONTENT_TYPE'])
 
       @model = Restfulie.from request
       if @model.save
@@ -23,6 +20,11 @@ def model_for_this_controller
       self.class.name[/(.*)Controller/,1].singularize.constantize
     end
 
+    def fits_content(type, content_type)
+      Restfulie::MediaType.supports?(content_type) &&
+              Restfulie::MediaType.media_type(content_type) == type
+    end
+
   end
 
 end