Merge pull request #3 from monkbroc/namespace-block-form

Add block form to Nenv() helper method

README.md

@@ -85,6 +85,16 @@ puts git.pager
 puts git.editor
 ```
 
+Or in block form
+
+```ruby
+Nenv :git do |git|
+  puts git.browser
+  puts git.pager
+  puts git.editor
+end
+```
+
 ### Custom type handling
 
 ```ruby

lib/nenv.rb

@@ -4,7 +4,9 @@
 require 'nenv/builder'
 
 def Nenv(namespace = nil)
-  Nenv::AutoEnvironment.new(namespace)
+  Nenv::AutoEnvironment.new(namespace).tap do |env|
+    yield env if block_given?
+  end
 end
 
 module Nenv

spec/lib/nenv_spec.rb

@@ -16,6 +16,24 @@
     end
   end
 
+  describe 'Nenv() helper method with block' do
+    it 'reads from env' do
+      expect(ENV).to receive(:[]).with('GIT_BROWSER').and_return('chrome')
+      Nenv('git') do |git|
+        git.browser
+      end
+    end
+
+    it 'return the value from env' do
+      allow(ENV).to receive(:[]).with('GIT_BROWSER').and_return('firefox')
+      result = nil
+      Nenv('git') do |git|
+        result = git.browser
+      end
+      expect(result).to eq('firefox')
+    end
+  end
+
   describe 'Nenv module' do
     it 'reads from env' do
       expect(ENV).to receive(:[]).with('CI').and_return('true')