wip

Author: tangrufus

lib/rails/app_env/credentials.rb

@@ -1,11 +1,15 @@
+require_relative "error"
+
 module Rails
   module AppEnv
     module Credentials
+      class AlreadyInitializedError < Rails::AppEnv::Error; end
+
       class << self
         attr_reader :original
 
         def initialize!
-          raise "Rails::AppEnv::Credentials has been already initialized." if @initialized
+          raise AlreadyInitializedError.new "Rails::AppEnv::Credentials has been already initialized." if @initialized
           @initialized = true
 
           @original = Rails.application.config.credentials

lib/rails/app_env/error.rb

@@ -0,0 +1,5 @@
+module Rails
+  module AppEnv
+    class Error < StandardError; end
+  end
+end

lib/rails/rails_ext/credentials_command.rb

@@ -2,7 +2,7 @@ module Rails
   module Command
     class CredentialsCommand
       def config
-        Rails.application.config.app_env.original_credentials
+        Rails::AppEnv::Credentials.original
       end
     end
   end

test/units/app_env/credentials_test.rb

@@ -2,23 +2,45 @@
 require_relative "../../test_helper"
 
 class Rails::AppEnv::CredentialsTest < ActiveSupport::TestCase
+  test "#initialize! can only be invoked once" do
+    reset_credentials
+
+    Rails::AppEnv::Credentials.initialize!
+
+    assert_raises Rails::AppEnv::Credentials::AlreadyInitializedError do
+      Rails::AppEnv::Credentials.initialize!
+    end
+  end
+
+  test "#initialize! backup original Rails.application.config.credentials" do
+    reset_credentials
+
+    expected = Object.new
+
+    Rails.application.config.stub :credentials, expected do
+      Rails::AppEnv::Credentials.initialize!
+    end
+
+    assert_same expected, Rails::AppEnv::Credentials.original
+  end
+
   test "#configuration is a kind of ActiveSupport::Credentials" do
     assert_kind_of ActiveSupport::InheritableOptions, Rails::AppEnv::Credentials.configuration
   end
 
   test "#configuration.content_path is config/credentials/{Rails.app_env}.yml.enc" do
     Dir.mktmpdir do |tmp_dir|
-        Rails.stub :root, Pathname(tmp_dir) do
-          Rails.stub :app_env, Rails::AppEnv::EnvironmentInquirer.new("foo") do
-            path = Rails.root.join("config/credentials/foo.yml.enc")
+      Rails.stub :root, Pathname(tmp_dir) do
+        Rails.stub :app_env, Rails::AppEnv::EnvironmentInquirer.new("foo") do
+          path = Rails.root.join("config/credentials/foo.yml.enc")
 
-            FileUtils.mkdir_p File.dirname(path)
-            FileUtils.touch path
+          FileUtils.mkdir_p File.dirname(path)
+          FileUtils.touch path
 
-            assert_equal path, Rails::AppEnv::Credentials.configuration.content_path
-          end
+          assert_equal path, Rails::AppEnv::Credentials.configuration.content_path
         end
       end
+    end
   end
 
   test "#configuration.content_path falls back to config/credentials.yml.enc when config/credentials/{Rails.app_env}.yml.enc not exist" do
@@ -59,4 +81,11 @@ class Rails::AppEnv::CredentialsTest < ActiveSupport::TestCase
       end
     end
   end
+
+  private
+
+  def reset_credentials
+    Rails::AppEnv::Credentials.instance_variable_set :@initialized, nil
+    Rails::AppEnv::Credentials.instance_variable_set :@original, nil
+  end
 end