Fail when accessing registry with invalid key

2014-09-26 00:24:19 +00:00
parent ff30b7474c
commit 2d4e04b2e5
4 changed files with 20 additions and 1 deletions
--- a/lib/producer/core/env.rb
+++ b/lib/producer/core/env.rb
@@ -18,7 +18,9 @@ module Producer
      end

      def [](key)
-        @registry[key]
+        @registry.fetch key
+      rescue KeyError
+        fail RegistryKeyError, key.inspect
      end
      alias get []

--- a/lib/producer/core/errors.rb
+++ b/lib/producer/core/errors.rb
@@ -4,5 +4,6 @@ module Producer
    RuntimeError                = Class.new(RuntimeError)
    ConditionNotMetError        = Class.new(Error)
    RemoteCommandExecutionError = Class.new(RuntimeError)
+    RegistryKeyError            = Class.new(RuntimeError)
  end
 end