################################################################################
#
#      Author: Zachary Patten <zachary@jovelabs.net>
#   Copyright: Copyright (c) Jove Labs
#     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"

describe ZTK::Command do

  subject { ZTK::Command.new }

  before(:all) do
    $stdout = File.open("/dev/null", "w")
    $stderr = File.open("/dev/null", "w")
    $stdin = File.open("/dev/null", "r")
  end

  describe "class" do

    it "should be an instance of ZTK::Command" do
      subject.should be_an_instance_of ZTK::Command
    end

    describe "default config" do

      it "should use $stdout as the default" do
        subject.config.stdout.should be_a_kind_of $stdout.class
        subject.config.stdout.should == $stdout
      end

      it "should use $stderr as the default" do
        subject.config.stderr.should be_a_kind_of $stderr.class
        subject.config.stderr.should == $stderr
      end

      it "should use $stdin as the default" do
        subject.config.stdin.should be_a_kind_of $stdin.class
        subject.config.stdin.should == $stdin
      end

      it "should use $logger as the default" do
        subject.config.logger.should be_a_kind_of ZTK::Logger
        subject.config.logger.should == $logger
      end

    end

  end

  describe "behaviour" do

    describe "execute" do

      it "should be able to execute the command \"hostname -f\"" do
        stdout = StringIO.new
        subject.config do |config|
          config.stdout = stdout
        end
        hostname = %x(hostname -f).chomp
        status = subject.exec("hostname -f")
        status.exit_code.should == 0
        stdout.rewind
        stdout.read.chomp.should == hostname
      end

      it "should timeout after the period specified" do
        stdout = StringIO.new
        subject.config do |config|
          config.stdout = stdout
          config.timeout = WAIT_SMALL
        end
        hostname = %x(hostname -f).chomp
        lambda { subject.exec("hostname -f ; sleep 10") }.should raise_error ZTK::CommandError
      end

      it "should throw an exception if the exit status is not as expected" do
        stdout = StringIO.new
        subject.config do |config|
          config.stdout = stdout
        end
        lambda { subject.exec("/bin/bash -c 'exit 64'") }.should raise_error ZTK::CommandError
      end

      it "should return a instance of an OpenStruct object" do
        stdout = StringIO.new
        subject.config do |config|
          config.stdout = stdout
        end
        result = subject.exec(%q{echo "Hello World"})
        result.should be_an_instance_of OpenStruct
      end

      it "should return the exit code" do
        stdout = StringIO.new
        subject.config do |config|
          config.stdout = stdout
        end
        data = 64

        result = subject.exec(%Q{/bin/bash -c 'exit #{data}'}, :exit_code => data)
        result.exit_code.should == data
      end

      it "should return the output" do
        stdout = StringIO.new
        subject.config do |config|
          config.stdout = stdout
        end
        data = "Hello World @ #{Time.now.utc}"

        result = subject.exec(%Q{echo "#{data}"})
        result.output.match(data).should_not be nil
      end

      it "should allow us to change the expected exit code" do
        stdout = StringIO.new
        subject.config do |config|
          config.stdout = stdout
        end
        data = 32
        result = subject.exec(%Q{/bin/bash -c 'exit #{data}'}, :exit_code => data)
      end

      describe "stdout" do

        it "should capture STDOUT and send it to the appropriate pipe" do
          stdout, stderr, stdin = StringIO.new, StringIO.new, StringIO.new

          subject.config do |config|
            config.stdout = stdout
            config.stderr = stderr
            config.stdin = stdin
          end
          data = "Hello World @ #{Time.now.utc}"

          subject.exec(%Q{echo "#{data}" -f >&1})

          stdout.rewind
          stdout.read.match(data).should_not be nil

          stderr.rewind
          stderr.read.match(data).should be nil

          stdin.rewind
          stdin.read.match(data).should be nil
        end

      end

      describe "stderr" do

        it "should capture STDERR and send it to the appropriate pipe" do
          stdout, stderr, stdin = StringIO.new, StringIO.new, StringIO.new

          subject.config do |config|
            config.stdout = stdout
            config.stderr = stderr
            config.stdin = stdin
          end
          data = "Hello World @ #{Time.now.utc}"

          subject.exec(%Q{echo "#{data}" -f >&2})

          stdout.rewind
          stdout.read.match(data).should be nil

          stderr.rewind
          stderr.read.match(data).should_not be nil

          stdin.rewind
          stdin.read.match(data).should be nil
        end
      end

    end

    describe "upload" do

      it "should raise a 'Not Supported' exception when attempting to upload" do
        lambda { subject.upload("abc", "123") }.should raise_error
      end

    end

    describe "download" do

      it "should raise a 'Not Supported' exception when attempting to download" do
        lambda { subject.download("abc", "123") }.should raise_error
      end

    end

  end

end