test.rb in pdk-3.0.0

- old
+ new
@@ -1,52 +1,54 @@
-module PDK::CLI
-  @new_define_cmd = @new_cmd.define_command do
-    name 'test'
-    usage 'test [options] <name>'
-    summary 'Create a new test for the object named <name>'
-    flag :u, :unit, 'Create a new unit test.'
-    PDK::CLI.puppet_version_options(self)
-    PDK::CLI.puppet_dev_option(self)
+module PDK
+  module CLI
+    @new_define_cmd = @new_cmd.define_command do
+      name 'test'
+      usage 'test [options] <name>'
+      summary 'Create a new test for the object named <name>'
+      flag :u, :unit, 'Create a new unit test.'
+      PDK::CLI.puppet_version_options(self)
+      PDK::CLI.puppet_dev_option(self)
 
-    run do |opts, args, _cmd|
-      require 'pdk/util/puppet_strings'
-      require 'pdk/util/bundler'
+      run do |opts, args, _cmd|
+        require 'pdk/util/puppet_strings'
+        require 'pdk/util/bundler'
 
-      PDK::CLI::Util.validate_puppet_version_opts(opts)
-      PDK::CLI::Util.ensure_in_module!(
-        message: 'Tests can only be created from inside a valid module directory.',
-        log_level: :info,
-      )
+        PDK::CLI::Util.validate_puppet_version_opts(opts)
+        PDK::CLI::Util.ensure_in_module!(
+          message: 'Tests can only be created from inside a valid module directory.',
+          log_level: :info
+        )
 
-      object_name = args[0]
+        object_name = args[0]
 
-      if object_name.nil? || object_name.empty?
-        puts command.help
-        exit 1
-      end
+        if object_name.nil? || object_name.empty?
+          puts command.help
+          exit 1
+        end
 
-      unless opts[:unit]
-        # At a future time, we'll replace this conditional with an interactive
-        # question to choose the test type.
-        PDK.logger.info 'Test type not specified, assuming unit.'
-        opts[:unit] = true
-      end
+        unless opts[:unit]
+          # At a future time, we'll replace this conditional with an interactive
+          # question to choose the test type.
+          PDK.logger.info 'Test type not specified, assuming unit.'
+          opts[:unit] = true
+        end
 
-      puppet_env = PDK::CLI::Util.puppet_from_opts_or_env(opts)
-      PDK::Util::RubyVersion.use(puppet_env[:ruby_version])
-      PDK::Util::Bundler.ensure_bundle!(puppet_env[:gemset])
+        puppet_env = PDK::CLI::Util.puppet_from_opts_or_env(opts)
+        PDK::Util::RubyVersion.use(puppet_env[:ruby_version])
+        PDK::Util::Bundler.ensure_bundle!(puppet_env[:gemset])
 
-      begin
-        generator, obj = PDK::Util::PuppetStrings.find_object(object_name)
+        begin
+          generator, obj = PDK::Util::PuppetStrings.find_object(object_name)
 
-        PDK::CLI::Util.analytics_screen_view('new_test', opts)
+          PDK::CLI::Util.analytics_screen_view('new_test', opts)
 
-        updates = generator.new(PDK.context, obj['name'], opts.merge(spec_only: true)).run
-        PDK::CLI::Util::UpdateManagerPrinter.print_summary(updates, tense: :past)
-      rescue PDK::Util::PuppetStrings::NoObjectError
-        raise PDK::CLI::ExitWithError, 'Unable to find anything called "%{object}" to generate unit tests for.' % { object: object_name }
-      rescue PDK::Util::PuppetStrings::NoGeneratorError => e
-        raise PDK::CLI::ExitWithError, 'PDK does not support generating unit tests for "%{object_type}" objects.' % { object_type: e.message }
+          updates = generator.new(PDK.context, obj['name'], opts.merge(spec_only: true)).run
+          PDK::CLI::Util::UpdateManagerPrinter.print_summary(updates, tense: :past)
+        rescue PDK::Util::PuppetStrings::NoObjectError
+          raise PDK::CLI::ExitWithError, format('Unable to find anything called "%{object}" to generate unit tests for.', object: object_name)
+        rescue PDK::Util::PuppetStrings::NoGeneratorError => e
+          raise PDK::CLI::ExitWithError, format('PDK does not support generating unit tests for "%{object_type}" objects.', object_type: e.message)
+        end
       end
     end
   end
 end