lib/leap_cli/config/secrets.rb in leap_cli-1.7.3 vs lib/leap_cli/config/secrets.rb in leap_cli-1.7.4

@@ -13,10 +13,11 @@
       @discovered_keys = {}
     end
 
     # we can't use fetch() or get(), since those already have special meanings
     def retrieve(key, environment)
+      environment ||= 'default'
       self.fetch(environment, {})[key.to_s]
     end
 
     def set(*args, &block)
       if block_given?
@@ -29,9 +30,10 @@
     def set_without_block(key, value, environment)
       set_with_block(key, environment) {value}
     end
 
     def set_with_block(key, environment, &block)
+      environment ||= 'default'
       key = key.to_s
       @discovered_keys[environment] ||= {}
       @discovered_keys[environment][key] = true
       self[environment] ||= {}
       self[environment][key] ||= yield