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)

    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,
      )

      object_name = args[0]

      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

      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)

        PDK::CLI::Util.analytics_screen_view('new_test', opts)

        generator.new(PDK::Util.module_root, obj['name'], opts.merge(spec_only: true)).run
      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 }
      end
    end
  end
end