#85: Feature/use environment without preserve

lib/octocatalog-diff/catalog-util/builddir.rb

@@ -84,11 +84,14 @@ def create_symlinks(logger = nil)
             install_directory_symlink(logger, File.join(@options[:basedir], x), x)
           end
         else
-          if @options[:environment]
-            logger.warn '--environment is ignored unless --preserve-environments is used' unless logger.nil?
-          end
-          if @options[:create_symlinks]
+          if @options[:create_symlinks] && @options[:environment]
+            unless logger.nil?
+              logger.warn '--create-symlinks with --environment ignored unless --preserve-environments is used'
+            end
+          elsif @options[:create_symlinks]
             logger.warn '--create-symlinks is ignored unless --preserve-environments is used' unless logger.nil?
+          elsif @options[:environment]
+            return install_directory_symlink(logger, @options[:basedir], "environments/#{@options[:environment]}")
           end
           install_directory_symlink(logger, @options[:basedir])
         end
@@ -178,6 +181,7 @@ def install_fact_file(logger, options)
       def install_directory_symlink(logger, dir, target = 'environments/production')
         raise ArgumentError, "Called install_directory_symlink with #{dir.class} argument" unless dir.is_a?(String)
         raise Errno::ENOENT, "Specified directory #{dir} doesn't exist" unless File.directory?(dir)
+
         symlink_target = File.join(@tempdir, target)
 
         if target =~ %r{/}
@@ -330,7 +334,7 @@ def install_ssl_client_password(logger, password)
       end
 
       def environment
-        @options[:preserve_environments] ? @options.fetch(:environment, 'production') : 'production'
+        @options.fetch(:environment, 'production')
       end
     end
   end

lib/octocatalog-diff/catalog/computed.rb

@@ -80,7 +80,7 @@ def compilation_dir
 
       # Environment used to compile catalog
       def environment
-        @opts[:preserve_environments] ? @opts.fetch(:environment, 'production') : 'production'
+        @options.fetch(:environment, 'production')
       end
 
       private

spec/octocatalog-diff/integration/preserve_environments_spec.rb

@@ -40,20 +40,16 @@
             '-n', 'rspec-node.github.net',
             '--environment', 'asdfgh',
             '--to-catalog', OctocatalogDiff::Spec.fixture_path('catalogs/default-catalog-v4.json'),
-            '--hiera-config', 'environments/production/config/hiera.yaml',
+            '--hiera-config', 'config/hiera.yaml',
             '--hiera-path-strip', '/var/lib/puppet', '--no-parallel'
           ]
         )
       end
 
       it 'should exit without error' do
-        expect(@result.exitcode).to eq(0), OctocatalogDiff::Integration.format_exception(@result)
+        expect(@result.exitcode).to eq(2), OctocatalogDiff::Integration.format_exception(@result)
         expect(@result.exception).to be_nil, OctocatalogDiff::Integration.format_exception(@result)
       end
-
-      it 'should log warning about --environment being useless in this context' do
-        expect(@result.logs).to match(/WARN -- : --environment is ignored unless --preserve-environments is used/)
-      end
     end
 
     context 'with --create-symlinks set' do

spec/octocatalog-diff/tests/catalog-util/builddir_spec.rb

@@ -121,15 +121,18 @@
         end
       end
 
-      context 'with --environment' do
+      context 'with --create-symlinks and --environment' do
         context 'with logger' do
           it 'should log a warning message and install default symlink' do
             logger = double('Logger')
-            expect(logger).to receive(:warn).with('--environment is ignored unless --preserve-environments is used')
+            expect(logger).to receive(:warn).with(
+              '--create-symlinks with --environment ignored unless --preserve-environments is used'
+            )
 
             @described_object.instance_variable_set(
               '@options',
               basedir: '/tmp/basedir',
+              create_symlinks: %w(foo bar),
               environment: 'baz'
             )
 
@@ -146,6 +149,7 @@
             @described_object.instance_variable_set(
               '@options',
               basedir: '/tmp/basedir',
+              create_symlinks: %w(foo bar),
               environment: 'baz'
             )
 
@@ -157,6 +161,41 @@
           end
         end
       end
+      context 'with --environment' do
+        context 'with logger' do
+          it 'should install a symlink to the given environment' do
+            logger = double('Logger')
+
+            @described_object.instance_variable_set(
+              '@options',
+              basedir: '/tmp/basedir',
+              environment: 'baz'
+            )
+
+            expect(@described_object)
+              .to receive(:install_directory_symlink)
+              .with(logger, '/tmp/basedir', 'environments/baz')
+
+            expect { @described_object.send(:create_symlinks, logger) }.not_to raise_error
+          end
+        end
+
+        context 'without logger' do
+          it 'should install default symlink' do
+            @described_object.instance_variable_set(
+              '@options',
+              basedir: '/tmp/basedir',
+              environment: 'baz'
+            )
+
+            expect(@described_object)
+              .to receive(:install_directory_symlink)
+              .with(nil, '/tmp/basedir', 'environments/baz')
+
+            expect { @described_object.send(:create_symlinks) }.not_to raise_error
+          end
+        end
+      end
 
       context 'without --create-symlinks or --environment' do
         it 'should install directory symlink' do

spec/octocatalog-diff/tests/catalog/computed_spec.rb

@@ -422,17 +422,19 @@
     context 'when preserve_environments is not set' do
       context 'and environment is specified' do
         it 'should return without raising error when directory exists' do
-          allow(File).to receive(:"directory?").with('/tmp/assert/environments/production').and_return(true)
+          allow(File).to receive(:"directory?").with('/tmp/assert/environments/production').and_return(false)
+          allow(File).to receive(:"directory?").with('/tmp/assert/environments/custom').and_return(true)
           described_object = described_class.allocate
-          described_object.instance_variable_set('@opts', environment: 'nope')
+          described_object.instance_variable_set('@opts', environment: 'custom')
           described_object.instance_variable_set('@builddir', OpenStruct.new(tempdir: '/tmp/assert'))
           expect { described_object.send(:assert_that_puppet_environment_directory_exists) }.not_to raise_error
         end
 
         it 'should raise Errno::ENOENT when directory does not exist' do
-          allow(File).to receive(:"directory?").with('/tmp/assert/environments/production').and_return(false)
+          allow(File).to receive(:"directory?").with('/tmp/assert/environments/production').and_return(true)
+          allow(File).to receive(:"directory?").with('/tmp/assert/environments/custom').and_return(false)
           described_object = described_class.allocate
-          described_object.instance_variable_set('@opts', environment: 'yup')
+          described_object.instance_variable_set('@opts', environment: 'custom')
           described_object.instance_variable_set('@builddir', OpenStruct.new(tempdir: '/tmp/assert'))
           expect { described_object.send(:assert_that_puppet_environment_directory_exists) }.to raise_error(Errno::ENOENT)
         end