cfddc4ac38
SPDK binaries built on Ubuntu 22.04 doesn't work on Ubuntu 24.04, so we need to use OS specific SPDK binaries.
105 lines
4.4 KiB
Ruby
105 lines
4.4 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
require_relative "../lib/spdk_setup"
|
|
|
|
RSpec.describe SpdkSetup do
|
|
subject(:spdk_setup) { described_class.new(spdk_version) }
|
|
|
|
let(:spdk_version) { DEFAULT_SPDK_VERSION }
|
|
|
|
describe "#prep" do
|
|
it "can prep host for spdk" do
|
|
expect(described_class).to receive(:r).with(/apt-get -y .*/)
|
|
expect(described_class).to receive(:r).with(/adduser .*/)
|
|
expect(FileUtils).to receive(:mkdir_p).with(SpdkPath.vhost_dir)
|
|
expect(FileUtils).to receive(:chown).with("spdk", "spdk", SpdkPath.vhost_dir)
|
|
expect { described_class.prep }.not_to raise_error
|
|
end
|
|
|
|
it "continues if user already exists" do
|
|
expect(described_class).to receive(:r).with(/apt-get -y .*/)
|
|
expect(described_class).to receive(:r).with(/adduser .*/).and_raise CommandFail.new("adduser: The user `spdk' already exists.", "", "")
|
|
expect(FileUtils).to receive(:mkdir_p).with(SpdkPath.vhost_dir)
|
|
expect(FileUtils).to receive(:chown).with("spdk", "spdk", SpdkPath.vhost_dir)
|
|
expect { described_class.prep }.not_to raise_error
|
|
end
|
|
|
|
it "fails if adduser fails with an unexpected error" do
|
|
expect(described_class).to receive(:r).with(/apt-get -y .*/)
|
|
expect(described_class).to receive(:r).with(/adduser .*/).and_raise CommandFail.new("adduser: some other error.", "", "")
|
|
expect { described_class.prep }.to raise_error CommandFail
|
|
end
|
|
end
|
|
|
|
describe "#package_url" do
|
|
it "fails if arch is unexpected" do
|
|
allow(Arch).to receive_messages(arm64?: false, x64?: false)
|
|
expect { spdk_setup.package_url(os_version: "ubuntu-22.04") }.to raise_error RuntimeError, "BUG: unexpected architecture"
|
|
end
|
|
|
|
it "returns a valid url for x64" do
|
|
allow(Arch).to receive_messages(arm64?: false, x64?: true)
|
|
expect(spdk_setup.package_url(os_version: "ubuntu-22.04")).to match(/https.*x64.*tar.gz/)
|
|
end
|
|
|
|
it "returns a valid url for arm64" do
|
|
allow(Arch).to receive_messages(arm64?: true, x64?: false)
|
|
expect(spdk_setup.package_url(os_version: "ubuntu-22.04")).to match(/https.*arm64.*tar.gz/)
|
|
end
|
|
end
|
|
|
|
describe "#install_package" do
|
|
it "can install the package" do
|
|
expect(spdk_setup).to receive(:package_url).and_return("package_url")
|
|
allow(spdk_setup).to receive(:install_path).and_return("install_path")
|
|
expect(spdk_setup).to receive(:r).with(/curl -L3 -o .* package_url/)
|
|
expect(FileUtils).to receive(:mkdir_p).with("install_path")
|
|
expect(FileUtils).to receive(:cd).with("install_path")
|
|
expect { spdk_setup.install_package(os_version: "ubuntu-22.04") }.not_to raise_error
|
|
end
|
|
end
|
|
|
|
describe "#create_service" do
|
|
it "creates the service file" do
|
|
expect(File).to receive(:write).with("/lib/systemd/system/spdk-#{spdk_version}.service", /.*/)
|
|
expect { spdk_setup.create_service(cpu_count: 4) }.not_to raise_error
|
|
end
|
|
end
|
|
|
|
describe "#create_hugepages_mount" do
|
|
it "creates the hugepages mount" do
|
|
expect(spdk_setup).to receive(:r).with("sudo --user=spdk mkdir -p /home/spdk/hugepages.#{spdk_version.tr("-", ".")}")
|
|
expect(File).to receive(:write).with("/lib/systemd/system/home-spdk-hugepages.#{spdk_version.tr("-", ".")}.mount", /.*/)
|
|
expect { spdk_setup.create_hugepages_mount }.not_to raise_error
|
|
end
|
|
end
|
|
|
|
describe "#enable_services" do
|
|
it "enables services" do
|
|
expect(spdk_setup).to receive(:r).with("systemctl enable spdk-#{spdk_version}.service")
|
|
expect(spdk_setup).to receive(:r).with("systemctl enable home-spdk-hugepages.#{spdk_version.tr("-", ".")}.mount")
|
|
expect { spdk_setup.enable_services }.not_to raise_error
|
|
end
|
|
end
|
|
|
|
describe "#start_services" do
|
|
it "starts services" do
|
|
expect(spdk_setup).to receive(:r).with("systemctl start spdk-#{spdk_version}.service")
|
|
expect(spdk_setup).to receive(:r).with("systemctl start home-spdk-hugepages.#{spdk_version.tr("-", ".")}.mount")
|
|
expect { spdk_setup.start_services }.not_to raise_error
|
|
end
|
|
end
|
|
|
|
describe "#verify_spdk" do
|
|
it "succeeds if is active" do
|
|
expect(spdk_setup).to receive(:r).with("systemctl is-active spdk-#{spdk_version}.service").and_return("active\n")
|
|
expect { spdk_setup.verify_spdk }.not_to raise_error
|
|
end
|
|
|
|
it "fails if not active" do
|
|
expect(spdk_setup).to receive(:r).with("systemctl is-active spdk-#{spdk_version}.service").and_return("inactive\n")
|
|
expect { spdk_setup.verify_spdk }.to raise_error RuntimeError, "SPDK failed to start"
|
|
end
|
|
end
|
|
end
|