require 'spec_helper'

describe Fluentd::Agent do
  shared_examples_for "Agent has common behavior" do |klass|
    describe "#extra_options" do
      context "blank" do
        let(:options) { {} }
        it { instance.pid_file.should    == described_class.default_options[:pid_file] }
        it { instance.log_file.should    == described_class.default_options[:log_file] }
        it { instance.config_file.should == described_class.default_options[:config_file] }
      end

      context "given" do
        let(:options) do
          {
            :pid_file => pid_file,
            :log_file => log_file,
            :config_file => config_file,
          }
        end
        let(:pid_file) { "pid" }
        let(:log_file) { "log" }
        let(:config_file) { "config" }

        it { instance.pid_file.should == pid_file }
        it { instance.log_file.should == log_file }
        it { instance.config_file.should == config_file }
      end
    end
  end

  let(:instance) { described_class.new(options) }
  let(:options) { {} }

  describe "Fluentd" do
    let(:described_class) { Fluentd::Agent::Fluentd } # override nested described_class behavior as https://github.com/rspec/rspec-core/issues/1114

    it_should_behave_like "Agent has common behavior"

    describe "#options_to_argv" do
      subject { instance.options_to_argv }
      it { should include("-c #{instance.config_file}") }
      it { should include("-d #{instance.pid_file}") }
      it { should include("-o #{instance.log_file}") }
      it { should include("--use-v1-config") }
    end

    describe "#start" do
      before { instance.stub(:running?).and_return { running } }

      context "running" do
        let(:running) { true }
        after { instance.start }

        it { instance.should_not_receive(:validate_fluentd_options) }
      end

      context "not running" do
        let(:running) { false }
        after { instance.start }

        it { instance.should_receive(:validate_fluentd_options) }

        context "validate_fluentd_options success" do
          before { instance.stub(:validate_fluentd_options).and_return { true } }
          it { instance.should_receive(:actual_start) }
        end

        context "validate_fluentd_options fail" do
          before { instance.stub(:validate_fluentd_options).and_return { false } }
          it { instance.should_not_receive(:actual_start) }
        end
      end
    end

    describe "#stop" do
      before { instance.stub(:running?).and_return { running } }

      context "running" do
        let(:running) { true }
        after { instance.stop }

        it { instance.should_receive(:actual_stop) }
      end

      context "not running" do
        let(:running) { false }
        after { instance.stop }

        it { instance.should_not_receive(:actual_stop) }
      end
    end

    describe "#restart" do
      before { instance.stub(:running?).and_return { running } }

      context "running" do
        let(:running) { true }
        after { instance.restart }

        it { instance.should_receive(:actual_restart) }
      end

      context "not running" do
        let(:running) { false }
        after { instance.restart }

        it { instance.should_not_receive(:actual_restart) }
      end
    end
  end

  describe "TdAgent" do
    let(:described_class) { Fluentd::Agent::TdAgent } # override nested described_class behavior as https://github.com/rspec/rspec-core/issues/1114

    it_should_behave_like "Agent has common behavior"
  end
end