# Copyright (c) HashiCorp, Inc.
# SPDX-License-Identifier: BUSL-1.1

require File.expand_path("../../../base", __FILE__)

describe Vagrant::Action::Runner do
  let(:instance) { described_class.new(action_name: "test") }

  it "should raise an error if an invalid callable is given" do
    expect { instance.run(7) }.to raise_error(ArgumentError, /must be a callable/)
  end

  it "should be able to use a Proc as a callable" do
    callable = Proc.new { raise Exception, "BOOM" }
    expect { instance.run(callable) }.to raise_error(Exception, "BOOM")
  end

  it "should be able to use a Method instance as a callable" do
    klass = Class.new do
      def action(env)
        raise Exception, "BANG"
      end
    end

    callable = klass.new.method(:action)
    expect { instance.run(callable) }.to raise_error(Exception, "BANG")
  end

  it "should be able to use a Class as a callable" do
    callable = Class.new do
      def initialize(app, env)
      end

      def self.name
        "TestAction"
      end

      def call(env)
        raise Exception, "BOOM"
      end
    end

    expect { instance.run(callable) }.to raise_error(Exception, "BOOM")
  end

  it "should be able to use a Class as a callable with no name attribute" do
    callable = Class.new do
      def initialize(app, env)
      end

      def call(env)
        raise Exception, "BOOM"
      end
    end

    expect { instance.run(callable) }.to raise_error(Exception, "BOOM")
  end

  it "should return the resulting environment" do
    callable = lambda do |env|
      env[:data] = "value"

      # Return nil so we can make sure it isn't using this return value
      nil
    end

    result = instance.run(callable)
    expect(result[:data]).to eq("value")
  end

  it "should pass options into hash given to callable" do
    result = nil
    callable = lambda do |env|
      result = env["data"]
    end

    instance.run(callable, "data" => "foo")
    expect(result).to eq("foo")
  end

  it "should pass global options into the hash" do
    result = nil
    callable = lambda do |env|
      result = env["data"]
    end

    instance = described_class.new("data" => "bar", action_name: "test")
    instance.run(callable)
    expect(result).to eq("bar")
  end

  it "should yield the block passed to the init method to get lazy loaded globals" do
    result = nil
    callable = lambda do |env|
      result = env["data"]
    end

    instance = described_class.new { { "data" => "bar", action_name: "test" } }
    instance.run(callable)
    expect(result).to eq("bar")
  end

  describe "triggers" do
    let(:environment) { double("environment", ui: nil) }
    let(:machine) { double("machine", triggers: machine_triggers, name: "") }
    let(:env_triggers) { double("env_triggers", find: []) }
    let(:machine_triggers) { double("machine_triggers", find: []) }

    before do
      allow(environment).to receive_message_chain(:vagrantfile, :config, :trigger)
      allow(Vagrant::Plugin::V2::Trigger).to receive(:new).
        and_return(env_triggers)
    end

    context "when only environment is provided" do
      let(:instance) { described_class.new(action_name: "test", env: environment) }

      it "should use environment to build new trigger" do
        expect(environment).to receive_message_chain(:vagrantfile, :config, :trigger)
        instance.run(lambda{|_|})
      end

      it "should pass environment based triggers to callable" do
        callable = lambda { |env| expect(env[:triggers]).to eq(env_triggers) }
        instance.run(callable)
      end
    end

    context "when only machine is provided" do
      let(:instance) { described_class.new(action_name: "test", machine: machine) }

      it "should pass machine based triggers to callable" do
        callable = lambda { |env| expect(env[:triggers]).to eq(machine_triggers) }
        instance.run(callable)
      end
    end

    context "when both environment and machine is provided" do
      let(:instance) { described_class.new(action_name: "test", machine: machine, env: environment) }

      it "should pass machine based triggers to callable" do
        callable = lambda { |env| expect(env[:triggers]).to eq(machine_triggers) }
        instance.run(callable)
      end
    end
  end
end
