spec/rake/funnel/tasks/bin_path_spec.rb in rake-funnel-0.16.1 vs spec/rake/funnel/tasks/bin_path_spec.rb in rake-funnel-0.17.0

@@ -11,20 +11,22 @@
   end
 
   describe 'execution' do
     let(:default_path) { 'default PATH contents' }
     let(:search_pattern) { %w(foo bar) }
+    let(:path_modifier) { nil }
 
     before {
       allow(ENV).to receive(:[]).with('PATH').and_return(default_path)
       allow(ENV).to receive(:[]=)
       allow(Rake).to receive(:rake_output_message)
     }
 
     subject {
       described_class.new do |t|
         t.search_pattern = search_pattern
+        t.path_modifier = path_modifier unless path_modifier.nil?
       end
     }
 
     context 'paths to add' do
       before {
@@ -47,10 +49,37 @@
 
       it 'should report added paths' do
         expect(Rake).to have_received(:rake_output_message).with(%r|/foo$|)
         expect(Rake).to have_received(:rake_output_message).with(%r|/bar$|)
       end
+
+      describe 'path modifier' do
+        context 'adding paths' do
+          let(:path_modifier) { proc { |paths| paths.push('added path') } }
+
+          it 'adds extra paths' do
+            expect(ENV).to have_received(:[]=).with('PATH',/added path/)
+          end
+
+          it 'retains other paths' do
+            expect(ENV).to have_received(:[]=).with('PATH', /foo/)
+            expect(ENV).to have_received(:[]=).with('PATH', /bar/)
+          end
+        end
+
+        context 'rejecting paths' do
+          let(:path_modifier) { proc { |paths| paths.reject { |p| p =~ /foo/ } } }
+
+          it 'removes matching paths' do
+            expect(ENV).not_to have_received(:[]=).with('PATH', /foo/)
+          end
+
+          it 'retains non-matching paths' do
+            expect(ENV).to have_received(:[]=).with('PATH', /bar/)
+          end
+        end
+      end
     end
 
     context 'no paths to add' do
       before {
         allow(Dir).to receive(:[]).with(*search_pattern).and_return([])
@@ -60,9 +89,13 @@
         Task[subject.name].invoke
       }
 
       it 'should not print message' do
         expect(Rake).not_to have_received(:rake_output_message)
+      end
+
+      it 'should not set environment variable' do
+        expect(ENV).not_to have_received(:[]=).with('PATH')
       end
     end
   end
 end