BuildEnvironment: use separate sets for procs and symbols

This commit is contained in:
Jack Nagel 2013-02-25 14:01:02 -06:00
parent 6b96102fad
commit aca48deda0
2 changed files with 11 additions and 6 deletions

View File

@ -3,10 +3,14 @@ require 'set'
class BuildEnvironment
def initialize(*settings)
@settings = Set.new(settings)
@procs = Set.new
end
def <<(o)
@settings << o
case o
when Proc then @procs << o
else @settings << o
end
self
end
@ -19,12 +23,11 @@ class BuildEnvironment
end
def modify_build_environment(context=nil)
p = @settings.find { |s| Proc === s }
ENV.instance_exec(context, &p) unless p.nil?
@procs.each { |p| ENV.instance_exec(context, &p) }
end
def _dump(*)
@settings.dup.reject { |s| Proc === s }.join(":")
@settings.to_a.join(":")
end
def self._load(s)

View File

@ -21,8 +21,10 @@ class BuildEnvironmentTests < Test::Unit::TestCase
end
def test_modify_build_environment
@env << Proc.new { 1 }
assert_equal 1, @env.modify_build_environment
@env << Proc.new { raise StandardError }
assert_raises(StandardError) do
@env.modify_build_environment
end
end
def test_marshal