test/writer_test.rb in symmetric-encryption-2.0.0 vs test/writer_test.rb in symmetric-encryption-2.0.1

@@ -32,10 +32,11 @@
 
     should "encrypt to string stream" do
       stream = StringIO.new
       file = SymmetricEncryption::Writer.new(stream, :header => false, :random_key => false, :random_iv => false)
       written_len = @data.inject(0) {|sum,str| sum + file.write(str)}
+      assert_equal @data_len, file.size
       file.close
 
       assert_equal @data_len, written_len
       result = stream.string
       result.force_encoding('binary') if defined?(Encoding)
@@ -45,17 +46,19 @@
     should "encrypt to string stream using .open" do
       written_len = 0
       stream = StringIO.new
       SymmetricEncryption::Writer.open(stream) do |file|
         written_len = @data.inject(0) {|sum,str| sum + file.write(str)}
+        assert_equal @data_len, file.size
       end
       assert_equal @data_len, written_len
     end
 
     should "encrypt to file using .open" do
       written_len = nil
       SymmetricEncryption::Writer.open(@filename, :header => false, :random_key => false, :random_iv => false) do |file|
         written_len = @data.inject(0) {|sum,str| sum + file.write(str)}
+        assert_equal @data_len, file.size
       end
       assert_equal @data_len, written_len
       assert_equal @data_encrypted, File.open(@filename, 'rb') {|f| f.read }
     end
   end