lib/heroku_config/aws_key.rb in heroku-config-0.3.0 vs lib/heroku_config/aws_key.rb in heroku-config-0.4.0

@@ -1,8 +1,9 @@
 module HerokuConfig
   class AwsKey < Base
     include AwsServices
+    include AwsKeyNameMap
     class MaxKeysError < StandardError; end
     class AccessKeyNotFound < StandardError; end
 
     def initialize(options, access_key_id)
       @options, @access_key_id = options, access_key_id
@@ -81,11 +82,11 @@
       puts "Old access key deleted: #{old_key.access_key_id}"
     end
 
     def update_heroku_config(new_key, new_secret)
       out = config.set(
-        "AWS_ACCESS_KEY_ID" => new_key,
-        "AWS_SECRET_ACCESS_KEY" => new_secret,
+        id_key_name => new_key,
+        secret_key_name => new_secret,
       )
       puts "Setting heroku config variables"
       puts out
     end