env.rb 0.0.2 → 0.0.3

data/lib/env.rb

@@ -6,6 +6,8 @@ class EnvironmentError < StandardError
 end
 
 module Env
+  RACK   = %w{GEM_HOME TMPDIR HTTPS}
+  HEROKU = %w{TMP TEMP} + RACK
   @@dependencies = []
   @@env          = {}
   @@enforced     = false
@@ -27,7 +29,11 @@ module Env
     end
 
     def import(key)
-      export(key, ENV.get(key))  
+      if key.is_a? Symbol
+        const_get(key.to_s.upcase).each { |key| import(key) }
+      else
+        export(key, ENV.get(key))  
+      end
     end
 
     def load!
@@ -36,6 +42,20 @@ module Env
       File.exist?("Envfile")
     end
 
+    def unload
+      @@enforced and Env.unenforce
+      @@dependencies = []
+      @@env = {}
+    end
+
+    def unenforce
+      class << ENV
+        alias_method :[], :get  
+        alias_method :[]=, :set
+      end
+      @@enforced = false
+    end
+
     def enforce
       class << ENV
         alias_method :get, :[]  
@@ -54,7 +74,7 @@ module Env
 
     private 
     def _raise(key)
-      raise EnvironmentError, "#{key} is not a declared depency, add it to your Envfile"
+      raise EnvironmentError, "#{key} is not a declared dependency, add it to your Envfile"
     end
 
     def uri?(value)

data/lib/env/version.rb

@@ -1,3 +1,3 @@
 module Env
-  VERSION = "0.0.2"
+  VERSION = "0.0.3"
 end

data/spec/enforce_spec.rb

@@ -1,4 +1,4 @@
-require_relative '../lib/env'
+require 'spec_helper'
 
 describe Env, "::enforce" do
   before { Env.enforce }

data/spec/import_spec.rb

@@ -1,15 +1,17 @@
-require_relative '../lib/env'
+require 'spec_helper'
 
-describe Env, 'import' do
-  def envfile(string)
-    File.open("Envfile", 'w') do |f|
-      f << string
-    end
+describe Env, '::import' do
+  after { File.unlink('Envfile') }
+
+  let(:rack) do
+    %w{GEM_HOME TMPDIR HTTPS}
   end
 
-  after { File.unlink('Envfile') }
+  let(:heroku) do
+    %w{TMP TEMP} + rack
+  end
 
-  context "with an import statement" do
+  context "import 'FOO'" do
     before do
       ENV['FOO'] = 'bar'
       envfile(%{
@@ -18,8 +20,40 @@ describe Env, 'import' do
       Env.load!
     end
 
-    it "should use the value in the ENV" do
+    it "should use the value in ENV['FOO']" do
       ENV['FOO'].should == 'bar'
     end
   end
+
+  context "import :rack" do
+    before do
+      rack.each { |var| ENV[var] = 'foo' }
+      envfile(%{
+        import :rack
+      })
+      Env.load!
+    end
+
+    it "should import GEM_HOME, TMPDIR, and HTTPS" do
+      rack.each do |var|
+        lambda { ENV[var] }.should_not raise_error
+      end
+    end
+  end
+
+  context "import :heroku" do
+    before do
+      heroku.each { |var| ENV[var] = 'foo' }
+      envfile(%{
+        import :heroku
+      })
+      Env.load!
+    end
+
+    it "should import TMP, TEMP, and RACK" do
+      heroku.each do |var|
+        lambda { ENV[var] }.should_not raise_error
+      end
+    end
+  end
 end

data/spec/load_spec.rb

@@ -1,11 +1,7 @@
-require_relative '../lib/env'
+require 'spec_helper'
 
 describe Env, '::load!' do
-  def envfile(string)
-    File.open("Envfile", 'w') do |f|
-      f << string
-    end
-  end
+  after { Env.unload }
 
   context "with no Envfile" do
     it "should return false" do

data/spec/spec_helper.rb

@@ -0,0 +1,11 @@
+require_relative '../lib/env'
+
+def envfile(string)
+  File.open("Envfile", 'w') do |f|
+    f << string
+  end
+end
+
+RSpec.configure do |config|
+  config.after(:each) { Env.unload }
+end

data/spec/uri_spec.rb

@@ -1,4 +1,4 @@
-require_relative '../lib/env'
+require 'spec_helper'
 
 describe Env, 'uri support' do
   context "with a value FOO that is not a URI" do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: env.rb
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
   prerelease: 
 platform: ruby
 authors:
@@ -32,6 +32,7 @@ files:
 - spec/enforce_spec.rb
 - spec/import_spec.rb
 - spec/load_spec.rb
+- spec/spec_helper.rb
 - spec/uri_spec.rb
 has_rdoc: true
 homepage: ''
@@ -62,4 +63,5 @@ test_files:
 - spec/enforce_spec.rb
 - spec/import_spec.rb
 - spec/load_spec.rb
+- spec/spec_helper.rb
 - spec/uri_spec.rb