climate_control 1.0.0 → 1.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 65eb94d5a3f27af965178320b8b470c81b36cc646c55eda3b6471db78fca49a4
-  data.tar.gz: ba3abd5ba34a63c23f77195a7f9d4c8e5853f313812dc95b5d9ed37032f964c8
+  metadata.gz: 1052215413e209a2ce0531a9b293d56f51df36db70688af69ae15e70571a043d
+  data.tar.gz: ed855bfdddc57fc46532edc296d9a8de6a2a3653f5b148ab10849433aeb61026
 SHA512:
-  metadata.gz: 0a456cc8d8f3fb15cce8bf3401fbbf954b6ad2e7427a20a02fb35abf1da7852833ee8da7862ca4dfa2f98ef9ed62a29841c18d6ab5f40a1d3192073cd73bc621
-  data.tar.gz: 21879ffe4cc583c24fcb7e61fe1617f5126c5f606b4327cc6384cc81dfbff061fff6262c343054dee166f75c109769981354918c2e0481da4412dbf2cf211207
+  metadata.gz: e292d7055db48aaa3cf664f447c6826d3ccf59cc1142dc71a89e8b8e97dbac26d553ebd2912a1f8b2a15b3f75b71de5f3345244cc0cb2f4dd8970f47b73b12c1
+  data.tar.gz: 96f9e0f2e8193c29d708f94c1cdced3d7a1b305ae0cb781729895b838be70233a0082f767f840a5d943e5b206cc55539606de156e4b301bab787e29ab45507b4

data/.github/workflows/ci.yml

@@ -6,7 +6,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        ruby-version: [3.0, 2.7, 2.6, 2.5]
+        ruby-version: [3.1, '3.0', 2.7, 2.6, 2.5]
     steps:
       - uses: actions/checkout@v2
       - name: Set up Ruby ${{ matrix.ruby-version }}

data/CHANGELOG.md

@@ -0,0 +1,45 @@
+# Changelog
+
+All notable changes to this project will be documented in this file.
+
+The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
+and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
+
+## Unreleased
+
+## 1.1.0 / 2022-05-26
+
+- Refactor to use `Monitor` instead of `Mutex`
+- Add documentation about thread-safety
+- Allow ClimateControl.modify to be called without environment variables
+- Add test for concurrent access needed to be inside block
+- Relax development dependencies
+
+## 1.0.1 / 2021-05-26
+
+- Require minimum Ruby version of 2.5.0
+
+# 1.0.0 / 2021-03-06
+
+- Commit to supporting latest patch versions of Ruby 2.5+
+- Improve documentation
+- Format code with StandardRB
+- Bump gem dependencies
+
+# 0.2.0 / 2017-05-12
+
+- Allow nested environment changes in the same thread
+
+# 0.1.0 / 2017-01-07
+
+- Remove ActiveSupport dependency
+
+# 0.0.4 / 2017-01-06
+
+- Improved thread safety
+- Handle TypeErrors during assignment
+- Improve documentation
+
+# 0.0.1 / 2012-11-28
+
+- Initial release

data/CODEOWNERS

@@ -0,0 +1 @@
+* @dorianmariefr

data/README.md

@@ -1,5 +1,7 @@
 # Climate Control
 
+![GitHub Actions CI](https://github.com/thoughtbot/climate_control/actions/workflows/ci.yml/badge.svg)
+
 Easily manage your environment.
 
 ## Installation
@@ -48,7 +50,7 @@ end
 To use with RSpec, you could define this in your spec:
 
 ```ruby
-def with_modified_env(options, &block)
+def with_modified_env(options = {}, &block)
   ClimateControl.modify(options, &block)
 end
 ```
@@ -105,6 +107,39 @@ manner becomes more difficult:
 Climate Control modifies environment variables only within the context of the
 block, ensuring values are managed properly and consistently.
 
+## Thread-safety
+
+When using threads, for instance when running tests concurrently in the same
+process, you may need to wrap your code inside `ClimateControl.modify` blocks,
+e.g.:
+
+```ruby
+first_thread = Thread.new do
+  ClimateControl.modify(SECRET: "1") do
+    p ENV["SECRET"] # => "1"
+    sleep 2
+    p ENV["SECRET"] # => "1"
+  end
+end
+
+second_thread = Thread.new do
+  ClimateControl.modify({}) do
+    sleep 1
+    p ENV["SECRET"] # => nil
+    sleep 1
+    p ENV["SECRET"] # => nil
+  end
+end
+
+first_thread.join
+second_thread.join
+```
+
+> The modification wraps ENV in a mutex. If there's contention (the env being used - including potentially mutating values), it blocks until the value is freed (we shift out of the Ruby block).
+>
+> <cite><a href="https://github.com/thoughtbot/climate_control/issues/32#issuecomment-800713686">Josh Clayton</a></cite>
+
+
 ## Contributing
 
 1. Fork it
@@ -118,3 +153,18 @@ This project uses [StandardRB](https://github.com/testdouble/standard) to ensure
 ## License
 
 climate_control is copyright 2012-2021 Joshua Clayton and thoughtbot, inc. It is free software and may be redistributed under the terms specified in the [LICENSE](https://github.com/thoughtbot/climate_control/blob/main/LICENSE) file.
+
+About thoughtbot
+----------------
+
+![thoughtbot](https://thoughtbot.com/brand_assets/93:44.svg)
+
+climate_control is maintained and funded by thoughtbot, inc.
+The names and logos for thoughtbot are trademarks of thoughtbot, inc.
+
+We love open source software!
+See [our other projects][community] or
+[hire us][hire] to design, develop, and grow your product.
+
+[community]: https://thoughtbot.com/community?utm_source=github
+[hire]: https://thoughtbot.com/hire-us?utm_source=github

data/climate_control.gemspec

@@ -5,8 +5,8 @@ require "climate_control/version"
 Gem::Specification.new do |gem|
   gem.name = "climate_control"
   gem.version = ClimateControl::VERSION
-  gem.authors = ["Joshua Clayton"]
-  gem.email = ["joshua.clayton@gmail.com"]
+  gem.authors = ["Joshua Clayton", "Dorian Marié"]
+  gem.email = ["joshua.clayton@gmail.com", "dorian@dorianmarie.fr"]
   gem.description = "Modify your ENV"
   gem.summary = "Modify your ENV easily with ClimateControl"
   gem.homepage = "https://github.com/thoughtbot/climate_control"
@@ -16,8 +16,10 @@ Gem::Specification.new do |gem|
   gem.test_files = gem.files.grep(%r{^(test|spec|features)/})
   gem.require_paths = ["lib"]
 
-  gem.add_development_dependency "rspec", "~> 3.10.0"
-  gem.add_development_dependency "rake", "~> 12.3.3"
-  gem.add_development_dependency "simplecov", "~> 0.9.1"
-  gem.add_development_dependency "standard", "~> 1.0.0"
+  gem.required_ruby_version = ">= 2.5.0"
+
+  gem.add_development_dependency "rspec"
+  gem.add_development_dependency "rake"
+  gem.add_development_dependency "simplecov"
+  gem.add_development_dependency "standard"
 end

data/lib/climate_control/version.rb

@@ -1,3 +1,3 @@
 module ClimateControl
-  VERSION = "1.0.0".freeze
+  VERSION = "1.1.1".freeze
 end

data/lib/climate_control.rb

@@ -1,16 +1,48 @@
-require "climate_control/environment"
 require "climate_control/errors"
-require "climate_control/modifier"
 require "climate_control/version"
+require "monitor"
 
 module ClimateControl
-  @@env = ClimateControl::Environment.new
+  extend self
 
-  def self.modify(environment_overrides, &block)
-    Modifier.new(env, environment_overrides, &block).process
+  SEMAPHORE = Monitor.new
+  private_constant :SEMAPHORE
+
+  def modify(environment_overrides = {}, &block)
+    environment_overrides = environment_overrides.transform_keys(&:to_s)
+
+    SEMAPHORE.synchronize do
+      previous = ENV.to_hash
+
+      begin
+        copy environment_overrides
+      ensure
+        middle = ENV.to_hash
+      end
+
+      block.call
+    ensure
+      after = ENV
+      (previous.keys | middle.keys | after.keys).each do |key|
+        if previous[key] != after[key] && middle[key] == after[key]
+          ENV[key] = previous[key]
+        end
+      end
+    end
   end
 
-  def self.env
-    @@env
+  def env
+    ENV
+  end
+
+  private
+
+  def copy(overrides)
+    overrides.each do |key, value|
+      ENV[key] = value
+    rescue TypeError => e
+      raise UnassignableValueError,
+        "attempted to assign #{value} to #{key} but failed (#{e.message})"
+    end
   end
 end

data/spec/acceptance/climate_control_spec.rb

@@ -113,6 +113,39 @@ describe "Climate control" do
     expect(ENV["BAZ"]).to be_nil
   end
 
+  it "handles threads accessing the same key wrapped in a block" do
+    first_thread = Thread.new {
+      with_modified_env do
+        with_modified_env CONFLICTING_KEY: "1" do
+          sleep 0.5
+          expect(ENV["CONFLICTING_KEY"]).to eq("1")
+        end
+
+        expect(ENV["CONFLICTING_KEY"]).to be_nil
+      end
+    }
+
+    second_thread = Thread.new {
+      with_modified_env do
+        sleep 0.25
+        expect(ENV["CONFLICTING_KEY"]).to be_nil
+
+        with_modified_env CONFLICTING_KEY: "2" do
+          expect(ENV["CONFLICTING_KEY"]).to eq("2")
+          sleep 0.5
+          expect(ENV["CONFLICTING_KEY"]).to eq("2")
+        end
+
+        expect(ENV["CONFLICTING_KEY"]).to be_nil
+      end
+    }
+
+    first_thread.join
+    second_thread.join
+
+    expect(ENV["CONFLICTING_KEY"]).to be_nil
+  end
+
   it "is re-entrant" do
     ret = with_modified_env(FOO: "foo") {
       with_modified_env(BAR: "bar") do
@@ -134,7 +167,22 @@ describe "Climate control" do
     }.to raise_error ClimateControl::UnassignableValueError, /attempted to assign .*Thing.* to FOO but failed \(#{message}\)$/
   end
 
-  def with_modified_env(options, &block)
+  it "restores the ENV even when an error was raised when assigning values" do
+    ENV["KEY_TO_OVERRIDE"] = "initial_value_1"
+    ENV["KEY_THAT_WILL_ERROR_OUT"] = "initial_value_2"
+
+    expect {
+      with_modified_env(
+        KEY_TO_OVERRIDE: "overwriten_value_1",
+        KEY_THAT_WILL_ERROR_OUT: :value_that_will_error_out
+      ) {}
+    }.to raise_error ClimateControl::UnassignableValueError
+
+    expect(ENV["KEY_TO_OVERRIDE"]).to eq("initial_value_1")
+    expect(ENV["KEY_THAT_WILL_ERROR_OUT"]).to eq("initial_value_2")
+  end
+
+  def with_modified_env(options = {}, &block)
     ClimateControl.modify(options, &block)
   end
 

metadata

@@ -1,91 +1,92 @@
 --- !ruby/object:Gem::Specification
 name: climate_control
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.1
 platform: ruby
 authors:
 - Joshua Clayton
-autorequire: 
+- Dorian Marié
+autorequire:
 bindir: bin
 cert_chain: []
-date: 2021-03-07 00:00:00.000000000 Z
+date: 2022-05-28 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rspec
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 3.10.0
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 3.10.0
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 12.3.3
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 12.3.3
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: simplecov
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 0.9.1
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 0.9.1
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: standard
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 1.0.0
+        version: '0'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - "~>"
+    - - ">="
       - !ruby/object:Gem::Version
-        version: 1.0.0
+        version: '0'
 description: Modify your ENV
 email:
 - joshua.clayton@gmail.com
+- dorian@dorianmarie.fr
 executables: []
 extensions: []
 extra_rdoc_files: []
 files:
 - ".github/workflows/ci.yml"
 - ".gitignore"
+- CHANGELOG.md
+- CODEOWNERS
 - CODE_OF_CONDUCT.md
 - Gemfile
 - LICENSE
-- NEWS
 - README.md
 - Rakefile
 - climate_control.gemspec
 - lib/climate_control.rb
-- lib/climate_control/environment.rb
 - lib/climate_control/errors.rb
-- lib/climate_control/modifier.rb
 - lib/climate_control/version.rb
 - spec/acceptance/climate_control_spec.rb
 - spec/spec_helper.rb
@@ -93,7 +94,7 @@ homepage: https://github.com/thoughtbot/climate_control
 licenses:
 - MIT
 metadata: {}
-post_install_message: 
+post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -101,15 +102,15 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: 2.5.0
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.1.4
-signing_key: 
+rubygems_version: 3.3.13
+signing_key:
 specification_version: 4
 summary: Modify your ENV easily with ClimateControl
 test_files:

data/NEWS

@@ -1,19 +0,0 @@
-1.0.0 (March 6, 2021)
-  Commit to supporting latest patch versions of Ruby 2.5+
-  Improve documentation
-  Format code with StandardRB
-  Bump gem dependencies
-
-0.2.0 (May 12, 2017)
-  Allow nested environment changes in the same thread
-
-0.1.0 (January 7, 2017)
-  Remove ActiveSupport dependency
-
-0.0.4 (January 6, 2017)
-  Improved thread safety
-  Handle TypeErrors during assignment
-  Improve documentation
-
-0.0.1 (November 28, 2012)
-  Initial release

data/lib/climate_control/environment.rb

@@ -1,33 +0,0 @@
-require "forwardable"
-
-module ClimateControl
-  class Environment
-    extend Forwardable
-
-    def initialize
-      @semaphore = Mutex.new
-      @owner = nil
-    end
-
-    def_delegators :env, :[]=, :to_hash, :[], :delete
-
-    def synchronize
-      if @owner == Thread.current
-        return yield if block_given?
-      end
-
-      @semaphore.synchronize do
-        @owner = Thread.current
-        yield if block_given?
-      ensure
-        @owner = nil
-      end
-    end
-
-    private
-
-    def env
-      ENV
-    end
-  end
-end

data/lib/climate_control/modifier.rb

@@ -1,92 +0,0 @@
-module ClimateControl
-  class Modifier
-    def initialize(env, environment_overrides = {}, &block)
-      @environment_overrides = stringify_keys(environment_overrides)
-      @block = block
-      @env = env
-    end
-
-    def process
-      @env.synchronize do
-        prepare_environment_for_block
-        run_block
-      ensure
-        cache_environment_after_block
-        delete_keys_that_do_not_belong
-        revert_changed_keys
-      end
-    end
-
-    private
-
-    def prepare_environment_for_block
-      @original_env = clone_environment
-      copy_overrides_to_environment
-      @env_with_overrides_before_block = clone_environment
-    end
-
-    def run_block
-      @block.call
-    end
-
-    def copy_overrides_to_environment
-      @environment_overrides.each do |key, value|
-        @env[key] = value
-      rescue TypeError => e
-        raise UnassignableValueError,
-          "attempted to assign #{value} to #{key} but failed (#{e.message})"
-      end
-    end
-
-    def keys_to_remove
-      @environment_overrides.keys
-    end
-
-    def keys_changed_by_block
-      @keys_changed_by_block ||= OverlappingKeysWithChangedValues.new(@env_with_overrides_before_block, @env_after_block).keys
-    end
-
-    def cache_environment_after_block
-      @env_after_block = clone_environment
-    end
-
-    def delete_keys_that_do_not_belong
-      (keys_to_remove - keys_changed_by_block).each { |key| @env.delete(key) }
-    end
-
-    def revert_changed_keys
-      (@original_env.keys - keys_changed_by_block).each do |key|
-        @env[key] = @original_env[key]
-      end
-    end
-
-    def clone_environment
-      @env.to_hash
-    end
-
-    def stringify_keys(env)
-      env.each_with_object({}) do |(key, value), hash|
-        hash[key.to_s] = value
-      end
-    end
-
-    class OverlappingKeysWithChangedValues
-      def initialize(hash_1, hash_2)
-        @hash_1 = hash_1 || {}
-        @hash_2 = hash_2
-      end
-
-      def keys
-        overlapping_keys.select do |overlapping_key|
-          @hash_1[overlapping_key] != @hash_2[overlapping_key]
-        end
-      end
-
-      private
-
-      def overlapping_keys
-        @hash_2.keys & @hash_1.keys
-      end
-    end
-  end
-end