diff --git a/app/models/document_category.rb b/app/models/document_category.rb index a7a2fd9..c22b81e 100644 --- a/app/models/document_category.rb +++ b/app/models/document_category.rb @@ -31,4 +31,12 @@ class DocumentCategory < Enumeration def transfer_relations(to) documents.update_all("category_id = #{to.id}") end + + def self.default + d = super + if d.nil? + d = find(:first) + end + d + end end diff --git a/test/unit/document_category_test.rb b/test/unit/document_category_test.rb index d3d3978..30ef168 100644 --- a/test/unit/document_category_test.rb +++ b/test/unit/document_category_test.rb @@ -38,4 +38,8 @@ class DocumentCategoryTest < ActiveSupport::TestCase e.update_attributes(:is_default => true) assert_equal 3, DocumentCategory.default.id end + + def test_force_default + assert_equal 1, DocumentCategory.default.id + end end