doo_dah 0.2.0 → 0.2.1

data/VERSION

@@ -1 +1 @@
-0.2.0
+0.2.1

data/doo_dah.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{doo_dah}
-  s.version = "0.2.0"
+  s.version = "0.2.1"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Michael Collas"]

data/lib/doo_dah/zip_output_stream.rb

@@ -12,6 +12,12 @@ module DooDah
       @output_stream = output_stream
       @total_bytes_written = 0
       @entries = []
+      return unless block_given?
+      begin
+        yield self
+      ensure
+        close
+      end
     end
 
     def create_entry(name, size=0, crc=0)

data/spec/zip_output_stream_spec.rb

@@ -10,6 +10,32 @@ module DooDah
       @entry = stub(:close => nil, :closed? => false)
       ZipEntry.stub(:new => @entry)
     end
+    
+    describe 'when constructing with a block' do
+      
+      it 'should yield itself to the block' do
+        block_parameter = nil
+        zip_output_stream = ZipOutputStream.new(@output_stream) {|output_stream| block_parameter = output_stream }
+        block_parameter.should == zip_output_stream
+      end
+      
+      it 'should close itself upon return from the block' do
+        zip_output_stream = ZipOutputStream.allocate
+        zip_output_stream.should_receive(:close)
+        zip_output_stream.send(:initialize, @output_stream) {|output_stream| }
+      end
+      
+      it 'should close itself even if an exception is thrown in the block' do
+        zip_output_stream = ZipOutputStream.allocate
+        zip_output_stream.should_receive(:close)
+        BadStuff = Class.new(Exception)
+        begin
+          zip_output_stream.send(:initialize, @output_stream) { |output_stream| raise BadStuff }
+        rescue BadStuff
+        end        
+      end
+      
+    end
 
     describe '#start_entry' do
 
@@ -158,6 +184,12 @@ module DooDah
       @zip.write('7 bytes')
       @zip.current_offset.should == 14
     end
+    
+    it 'should not close itself when constructed without a block' do
+      zip_output_stream = ZipOutputStream.allocate
+      zip_output_stream.should_not_receive(:close)
+      zip_output_stream.send(:initialize, @output_stream)      
+    end
 
   end
 

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification 
 name: doo_dah
 version: !ruby/object:Gem::Version 
-  hash: 23
+  hash: 21
   prerelease: 
   segments: 
   - 0
   - 2
-  - 0
-  version: 0.2.0
+  - 1
+  version: 0.2.1
 platform: ruby
 authors: 
 - Michael Collas