2023-09-14 16:15:03 -07:00

367 lines
11 KiB
Ruby

# Copyright (c) HashiCorp, Inc.
# SPDX-License-Identifier: BUSL-1.1
require File.expand_path("../../../../../base", __FILE__)
require Vagrant.source_root.join("plugins/commands/cloud/provider/update")
describe VagrantPlugins::CloudCommand::ProviderCommand::Command::Update do
include_context "unit"
let(:access_token) { double("token") }
let(:org_name) { "my-org" }
let(:box_name) { "my-box" }
let(:box_version) { "1.0.0" }
let(:box_version_provider) { "my-provider" }
let(:box_architecture) { "box-architecture" }
let(:account) { double("account") }
let(:organization) { double("organization") }
let(:box) { double("box", versions: [version]) }
let(:version) { double("version", version: box_version, provdiers: [provider]) }
let(:provider) { double("provider", name: box_version_provider) }
let(:provider_url) { nil }
describe "#update_provider" do
let(:argv) { [] }
let(:options) { {} }
let(:env) { double("env", ui: ui) }
let(:ui) { Vagrant::UI::Silent.new }
before do
allow(env).to receive(:ui).and_return(ui)
allow(VagrantCloud::Account).to receive(:new).
with(custom_server: anything, access_token: access_token).
and_return(account)
allow(subject).to receive(:with_provider).
with(
account: account,
org: org_name,
box: box_name,
version: box_version,
provider: box_version_provider,
architecture: box_architecture
).
and_yield(provider)
allow(provider).to receive(:save)
allow(subject).to receive(:format_box_results)
end
subject { described_class.new(argv, env) }
it "should update the provider" do
expect(provider).to receive(:save)
subject.update_provider(
org_name,
box_name,
box_version,
box_version_provider,
box_architecture,
provider_url,
access_token,
options
)
end
it "should return 0 on success" do
result = subject.update_provider(
org_name,
box_name,
box_version,
box_version_provider,
box_architecture,
provider_url,
access_token,
options
)
expect(result).to eq(0)
end
it "should return non-zero result on error" do
expect(provider).to receive(:save).and_raise(VagrantCloud::Error)
result = subject.update_provider(
org_name,
box_name,
box_version,
box_version_provider,
box_architecture,
provider_url,
access_token,
options
)
expect(result).not_to eq(0)
expect(result).to be_a(Integer)
end
it "should not update the URL when unset" do
expect(provider).not_to receive(:url=)
subject.update_provider(
org_name,
box_name,
box_version,
box_version_provider,
box_architecture,
provider_url,
access_token,
options
)
end
context "when URL is set" do
let(:provider_url) { double("provider-url") }
it "should update the URL" do
expect(provider).to receive(:url=).with(provider_url)
subject.update_provider(
org_name,
box_name,
box_version,
box_version_provider,
box_architecture,
provider_url,
access_token,
options
)
end
end
context "with options set" do
let(:checksum) { double("checksum") }
let(:checksum_type) { double("checksum_type") }
let(:options) { {} }
after do
subject.update_provider(
org_name,
box_name,
box_version,
box_version_provider,
box_architecture,
provider_url,
access_token,
options
)
end
it "should not modify option controlled values when unset" do
expect(provider).not_to receive(:checksum=)
expect(provider).not_to receive(:checksum_type=)
expect(provider).not_to receive(:architecture=)
expect(provider).not_to receive(:default_architecture=)
end
context "with checksum options set" do
let(:options) { {checksum: checksum, checksum_type: checksum_type} }
it "should set checksum options before saving" do
expect(provider).to receive(:checksum=).with(checksum)
expect(provider).to receive(:checksum_type=).with(checksum_type)
end
end
context "with architecture option set" do
let(:architecture) { double("architecture") }
let(:options) { {architecture: architecture} }
it "should set architecture before saving" do
expect(provider).to receive(:architecture=).with(architecture)
end
end
context "with default architecture option set" do
context "with true value" do
let(:options) { {default_architecture: true} }
it "should set default architecture to true" do
expect(provider).to receive(:default_architecture=).with(true)
end
end
context "with false value" do
let(:options) { {default_architecture: false} }
it "should set default architecture to false" do
expect(provider).to receive(:default_architecture=).with(false)
end
end
end
end
end
describe "#execute" do
let(:argv) { [] }
let(:iso_env) do
# We have to create a Vagrantfile so there is a root path
env = isolated_environment
env.vagrantfile("")
env.create_vagrant_env
end
subject { described_class.new(argv, iso_env) }
let(:action_runner) { double("action_runner") }
let(:client) { double("client", token: access_token) }
before do
allow(iso_env).to receive(:action_runner).and_return(action_runner)
allow(subject).to receive(:client_login).
and_return(client)
allow(subject).to receive(:update_provider)
end
context "with no arguments" do
it "shows help" do
expect { subject.execute }.
to raise_error(Vagrant::Errors::CLIInvalidUsage)
end
end
context "with box name argument" do
let(:argv) { ["#{org_name}/#{box_name}"] }
it "shows help" do
expect { subject.execute }.
to raise_error(Vagrant::Errors::CLIInvalidUsage)
end
context "with provider argument" do
let(:provider_arg) { "my-provider" }
before { argv << provider_arg }
it "shows help" do
expect { subject.execute }.
to raise_error(Vagrant::Errors::CLIInvalidUsage)
end
context "with version argument" do
let(:version_arg) { "1.0.0" }
before { argv << version_arg }
it "shows help" do
expect { subject.execute }.
to raise_error(Vagrant::Errors::CLIInvalidUsage)
end
context "with architecture argument" do
let(:architecture_arg) { "box-architecture" }
before { argv << architecture_arg }
it "should create the provider" do
expect(subject).
to receive(:update_provider).
with(
org_name,
box_name,
version_arg,
provider_arg,
architecture_arg,
any_args
)
subject.execute
end
it "should not provide URL value" do
expect(subject).
to receive(:update_provider).
with(
org_name,
box_name,
version_arg,
provider_arg,
architecture_arg,
nil,
any_args
)
subject.execute
end
context "with URL argument" do
let(:url_arg) { "provider-url" }
before { argv << url_arg }
it "should provide the URL value" do
expect(subject).
to receive(:update_provider).
with(
org_name,
box_name,
version_arg,
provider_arg,
architecture_arg,
url_arg,
any_args
)
subject.execute
end
end
context "with checksum and checksum type flags" do
let(:checksum_arg) { "checksum" }
let(:checksum_type_arg) { "checksum_type" }
before { argv.push("--checksum").push(checksum_arg).push("--checksum-type").push(checksum_type_arg) }
it "should include the checksum options" do
expect(subject).
to receive(:update_provider).
with(
org_name,
box_name,
version_arg,
provider_arg,
architecture_arg,
any_args,
hash_including(
checksum: checksum_arg,
checksum_type: checksum_type_arg
)
)
subject.execute
end
end
context "with architecture flag" do
let(:architecture_flag) { "test-arch" }
before { argv.push("--architecture").push(architecture_flag) }
it "should include the architecture flag" do
expect(subject).to receive(:update_provider) do |*_, opts|
expect(opts[:architecture]).to eq(architecture_flag)
end
subject.execute
end
end
context "with default architecture flag" do
context "enabled" do
before { argv.push("--default-architecture") }
it "should include default architecture set to true" do
expect(subject).to receive(:update_provider) do |*_, opts|
expect(opts[:default_architecture]).to be(true)
end
subject.execute
end
end
context "disabled" do
before { argv.push("--no-default-architecture") }
it "should include default architecture set to false" do
expect(subject).to receive(:update_provider) do |*_, opts|
expect(opts[:default_architecture]).to be(false)
end
subject.execute
end
end
end
end
end
end
end
end
end