#
# Author:: AJ Christensen (<aj@junglist.gen.nz>)
# Copyright:: Copyright (c) 2008 Opscode, Inc.
# License:: Apache License, Version 2.0
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

require 'spec_helper'
require "#{CHEF_SPEC_DATA}/knife_subcommand/test_yourself"

describe Chef::Application::Knife do
  include SpecHelpers::Knife

  before(:all) do
    class NoopKnifeCommand < Chef::Knife
      option :opt_with_default,
        :short => "-D VALUE",
        :long => "-optwithdefault VALUE",
        :default => "default-value"

      def run
      end
    end
  end

  after(:each) do
    # reset some really nasty global state
    NoopKnifeCommand.reset_config_loader!
  end

  before(:each) do
    # Prevent code from getting loaded on every test invocation.
    Chef::Knife.stub(:load_commands)

    @knife = Chef::Application::Knife.new
    @knife.stub(:puts)
    @knife.stub(:trap)
    Chef::Knife.stub(:list_commands)
  end

  it "should exit 1 and print the options if no arguments are given at all" do
    with_argv([]) do
      lambda { @knife.run }.should raise_error(SystemExit) { |e| e.status.should == 1 }
    end
  end

  it "should exit 2 if run without a sub command" do
    with_argv("--user", "adam") do
      Chef::Log.should_receive(:error).with(/you need to pass a sub\-command/i)
      lambda { @knife.run }.should raise_error(SystemExit) { |e| e.status.should == 2 }
    end
  end

  it "should run a sub command with the applications command line option prototype" do
    with_argv(*%w{noop knife command with some args}) do
      knife = double(Chef::Knife)
      Chef::Knife.should_receive(:run).with(ARGV, @knife.options).and_return(knife)
      @knife.should_receive(:exit).with(0)
      @knife.run
    end
  end

  it "should set the colored output to false by default on windows and true otherwise" do
    with_argv(*%w{noop knife command}) do
      @knife.should_receive(:exit).with(0)
      @knife.run
    end
    if windows?
      Chef::Config[:color].should be_false
    else
      Chef::Config[:color].should be_true
    end
  end

  describe "when given a path to the client key" do
    it "expands a relative path relative to the CWD" do
      relative_path = '.chef/client.pem'
      Dir.stub(:pwd).and_return(CHEF_SPEC_DATA)
      with_argv(*%W{noop knife command -k #{relative_path}}) do
        @knife.should_receive(:exit).with(0)
        @knife.run
      end
      Chef::Config[:client_key].should == File.join(CHEF_SPEC_DATA, relative_path)
    end

    it "expands a ~/home/path to the correct full path" do
      home_path = '~/.chef/client.pem'
      with_argv(*%W{noop knife command -k #{home_path}}) do
        @knife.should_receive(:exit).with(0)
        @knife.run
      end
      Chef::Config[:client_key].should == File.join(ENV['HOME'], '.chef/client.pem').gsub((File::ALT_SEPARATOR || '\\'), File::SEPARATOR)
    end

    it "does not expand a full path" do
      full_path = if windows?
        'C:/chef/client.pem'
      else
        '/etc/chef/client.pem'
      end
      with_argv(*%W{noop knife command -k #{full_path}}) do
        @knife.should_receive(:exit).with(0)
        @knife.run
      end
      Chef::Config[:client_key].should == full_path
    end
  end

  describe "with environment configuration" do
    before do
      Chef::Config[:environment] = nil
    end

    it "should default to no environment" do
      with_argv(*%w{noop knife command}) do
        @knife.should_receive(:exit).with(0)
        @knife.run
      end
      Chef::Config[:environment].should == nil
    end

    it "should load the environment from the config file" do
      config_file = File.join(CHEF_SPEC_DATA,"environment-config.rb")
      with_argv(*%W{noop knife command -c #{config_file}}) do
        @knife.should_receive(:exit).with(0)
        @knife.run
      end
      Chef::Config[:environment].should == 'production'
    end

    it "should load the environment from the CLI options" do
      with_argv(*%W{noop knife command -E development}) do
        @knife.should_receive(:exit).with(0)
        @knife.run
      end
      Chef::Config[:environment].should == 'development'
    end

    it "should override the config file environment with the CLI environment" do
      config_file = File.join(CHEF_SPEC_DATA,"environment-config.rb")
      with_argv(*%W{noop knife command -c #{config_file} -E override}) do
        @knife.should_receive(:exit).with(0)
        @knife.run
      end
      Chef::Config[:environment].should == 'override'
    end

    it "should override the config file environment with the CLI environment regardless of order" do
      config_file = File.join(CHEF_SPEC_DATA,"environment-config.rb")
      with_argv(*%W{noop knife command -E override -c #{config_file}}) do
        @knife.should_receive(:exit).with(0)
        @knife.run
      end
      Chef::Config[:environment].should == 'override'
    end

    it "should run a sub command with the applications command line option prototype" do
      with_argv(*%w{noop knife command with some args}) do
        knife = double(Chef::Knife)
        Chef::Knife.should_receive(:run).with(ARGV, @knife.options).and_return(knife)
        @knife.should_receive(:exit).with(0)
        @knife.run
      end
    end
  end

end
