FileYard checks the file size against max_upload_size attribute

test/unit/file_yard_test.rb

@@ -13,6 +13,11 @@ class FileYardTest < Test::Unit::TestCase
     @yard = FileYard.new(file_path, 100)
   end
 
+  def test_check_upload_size
+    assert_nothing_raised { @yard.check_upload_size(100.kilobytes) }
+    assert_raises(Instiki::ValidationError) { @yard.check_upload_size(100.kilobytes + 1) }
+  end
+
   def test_files
     assert_equal [], @yard.files
     
@@ -28,6 +33,32 @@ class FileYardTest < Test::Unit::TestCase
     assert_equal "#{file_path}/abcd", @yard.file_path('abcd')
   end
 
+  def test_size_limit
+    @yard = FileYard.new(file_path, 1)
+    one_kylobyte_string = "a" * 1024
+
+    # as StringIO
+    assert_nothing_raised { 
+      @yard.upload_file('acceptable_file', StringIO.new(one_kylobyte_string)) 
+    }
+    assert_raises(Instiki::ValidationError) { 
+      @yard.upload_file('one_byte_too_long', StringIO.new(one_kylobyte_string + 'a')) 
+    }
+
+    # as Tempfile
+    require 'tempfile'
+    Tempfile.open('acceptable_file') { |f| f.write(one_kylobyte_string) } 
+      assert_nothing_raised { 
+        @yard.upload_file('acceptable_file', f) 
+      }
+    }
+    Tempfile.open('one_byte_too_long') { |f| f.write(one_kylobyte_string + 'a')
+      assert_nothing_raised { 
+        @yard.upload_file('one_byte_too_long_2', f)
+      }
+    }
+  end
+
   def file_path
     "#{RAILS_ROOT}/storage/test/instiki"
   end