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

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

require Vagrant.source_root.join("plugins/commands/destroy/command")

describe VagrantPlugins::CommandDestroy::Command do
  include_context "unit"

  let(:entry_klass) { Vagrant::MachineIndex::Entry }
  let(:argv)     { [] }
  let(:vagrantfile_content) { "" }
  let(:iso_env) do
    env = isolated_environment
    env.vagrantfile(vagrantfile_content)
    env.create_vagrant_env
  end

  subject { described_class.new(argv, iso_env) }

  let(:action_runner) { double("action_runner") }

  def new_entry(name)
    entry_klass.new.tap do |e|
      e.name = name
      e.vagrantfile_path = "/bar"
    end
  end

  before do
    allow(iso_env).to receive(:action_runner).and_return(action_runner)
  end

  context "with no argument" do
    before { @state_var = :running }
    let(:vagrantfile_content){ "Vagrant.configure(2){|config| config.vm.box = 'dummy'}" }
    let(:state) { double("state", id: :running) }
    let(:machine) do
      iso_env.machine(iso_env.machine_names[0], :dummy).tap do |m|
        allow(m).to receive(:state).and_return(state)
        allow(m).to receive(:name).and_return("default")
      end
    end

    before do
      allow(subject).to receive(:with_target_vms) { |&block| block.call machine }
    end

    it "should destroy the default box" do
      allow_any_instance_of(Vagrant::BatchAction).to receive(:action) .with(machine, :destroy, force_confirm_destroy: false, force_halt: true)

      expect(machine.state).to receive(:id).and_return(:running)
      expect(machine.state).to receive(:id).and_return(:dead)
      expect(subject.execute).to eq(0)
    end

    it "exits 0 if vms are not created" do
      allow_any_instance_of(Vagrant::BatchAction).to receive(:action) .with(machine, :destroy, anything)
      allow(machine.state).to receive(:id).and_return(:not_created)

      expect(machine.state).to receive(:id).and_return(:not_created)
      expect(subject.execute).to eq(0)
    end

    it "exits 1 if a vms destroy was declined" do
      allow_any_instance_of(Vagrant::BatchAction).to receive(:action) .with(machine, :destroy, anything)

      expect(machine.state).to receive(:id).and_return(:running)
      expect(machine.state).to receive(:id).and_return(:running)
      expect(subject.execute).to eq(1)
    end

    context "with multiple machines" do
      let(:vagrantfile_content) do 
        <<-VF
        Vagrant.configure("2") do |config|
          config.vm.box = "base"
          config.vm.define :machine1
          config.vm.define :machine2
        end
        VF
      end

      let(:state2) { "" }

      let(:machine2) do
        iso_env.machine(iso_env.machine_names[1], :dummy).tap do |m|
          allow(m).to receive(:state).and_return(state2)
          allow(m).to receive(:name).and_return("not_default")
        end
      end

      before do
        allow(subject).to receive(:with_target_vms).and_yield(machine).and_yield(machine2)
        allow_any_instance_of(Vagrant::BatchAction).to receive(:action) .with(machine2, :destroy, anything)
        allow_any_instance_of(Vagrant::BatchAction).to receive(:action) .with(machine, :destroy, anything)
      end
      
      context "second machine is not created" do
        let(:state2) { double("state", id: :not_created) }
        let(:state) { double("state", id: :not_created) }


        it "exits 0 if vms are successfully destroyed" do
          expect(machine.state).to receive(:id).and_return(:running)
          expect(machine.state).to receive(:id).and_return(:dead)
          expect(machine2.state).to receive(:id).and_return(:not_created)
          expect(subject.execute).to eq(0)
        end

        it "exits 0 if vms are not created" do
          expect(machine.state).to receive(:id).and_return(:not_created)
          expect(machine2.state).to receive(:id).and_return(:not_created)
          expect(subject.execute).to eq(0)
        end
      end
      
      context "second machine is running" do
        let(:state2) { double("state", id: :running) }
        
        it "exits 0 if vms are not successfully destroyed" do
          expect(machine.state).to receive(:id).and_return(:running)
          expect(machine.state).to receive(:id).and_return(:dead)
          expect(machine2.state).to receive(:id).and_return(:running)
          expect(machine2.state).to receive(:id).and_return(:dead)
          expect(subject.execute).to eq(0)
        end

        it "exits 1 if vms are not successfully destroyed" do
          expect(machine.state).to receive(:id).and_return(:running)
          expect(machine2.state).to receive(:id).and_return(:running)
          expect(subject.execute).to eq(1)
        end

        it "exits 2 if some vms are not successfully destroyed" do
          expect(machine.state).to receive(:id).and_return(:running)
          expect(machine.state).to receive(:id).and_return(:dead)
          expect(machine2.state).to receive(:id).and_return(:running)
          expect(subject.execute).to eq(2)
        end
      end
    end

    context "with VAGRANT_DEFAULT_PROVIDER set" do
      before do
        if ENV["VAGRANT_DEFAULT_PROVIDER"]
          @original_default = ENV["VAGRANT_DEFAULT_PROVIDER"]
        end
        ENV["VAGRANT_DEFAULT_PROVIDER"] = "unknown"
      end
      after do
        if @original_default
          ENV["VAGRANT_DEFAULT_PROVIDER"] = @original_default
        else
          ENV.delete("VAGRANT_DEFAULT_PROVIDER")
        end
      end

      it "should attempt to use dummy provider" do
        expect{ subject.execute }.to raise_error(Vagrant::Errors::UnimplementedProviderAction)
      end
    end
  end

  context "with --parallel set" do
    let(:argv){ ["--parallel", "--force"] }

    it "passes in true to batch" do
      batch = double("environment_batch")
      expect(iso_env).to receive(:batch).with(true).and_yield(batch)
      expect(batch).to receive(:action).with(anything, :destroy, anything) do |machine,action,args|
        expect(machine).to be_kind_of(Vagrant::Machine)
        expect(action).to eq(:destroy)
      end
      subject.execute
    end
  end

  context "with --graceful set" do
    let(:argv){ ["--graceful", "--force"] }

    it "passes in true to batch" do
      batch = double("environment_batch")
      expect(iso_env).to receive(:batch).and_yield(batch)
      expect(batch).to receive(:action).with(anything, :destroy, force_confirm_destroy: true, force_halt: false)
      subject.execute
    end
  end

  context "with a global machine" do
    let(:argv){ ["1234"] }

    it "destroys a vm with an id" do

      global_env = isolated_environment
      global_env.vagrantfile("Vagrant.configure(2){|config| config.vm.box = 'dummy'}")
      global_venv = global_env.create_vagrant_env
      global_machine = global_venv.machine(global_venv.machine_names[0], :dummy)
      global_machine.id = "1234"
      global = new_entry(global_machine.name)
      global.provider = "dummy"
      global.vagrantfile_path = global_env.workdir
      locked = iso_env.machine_index.set(global)
      iso_env.machine_index.release(locked)

      allow(subject).to receive(:with_target_vms) { |&block| block.call global_machine }


      batch = double("environment_batch")
      expect(iso_env).to receive(:batch).and_yield(batch)
      expect(batch).to receive(:action).with(global_machine, :destroy, anything) do |machine,action,args|
        expect(machine).to be_kind_of(Vagrant::Machine)
        expect(action).to eq(:destroy)
      end
      subject.execute
    end
  end

  context "with an argument" do
    let(:vagrantfile_content) do
        <<-VF
        Vagrant.configure("2") do |config|
          config.vm.define "app"
          config.vm.define "db"
        end
        VF
    end
    let(:argv){ ["app"] }
    let(:machine) { iso_env.machine(iso_env.machine_names[0], :dummy) }

    it "destroys a vm" do
      batch = double("environment_batch")
      expect(iso_env).to receive(:batch).and_yield(batch)
      expect(batch).to receive(:action).with(machine, :destroy, anything) do |machine,action,args|
        expect(machine).to be_kind_of(Vagrant::Machine)
        expect(action).to eq(:destroy)
      end
      subject.execute
    end

    context "with machine that does not exist" do
      let(:argv){ ["notweb"] }
      it "raises an exception" do
        expect { subject.execute }.to raise_error(Vagrant::Errors::MachineNotFound)
      end
    end

    context "with an invalid argument" do
      let(:argv){ [""] }
      it "raises an exception" do
        expect { subject.execute }.to raise_error(Vagrant::Errors::MachineNotFound)
      end
    end
  end
end
