Update template directory in spec

drwl · Sep 8, 2023 · cd066ae · cd066ae
1 parent 6dd32cc
commit cd066ae
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/spec/integration/cli_spec.rb b/spec/integration/cli_spec.rb
@@ -32,7 +32,7 @@
 
   context "when running in a directory with a Rakefile and a Gemfile" do
     let(:help_banner_fragment) { "Usage: annotaterb [command] [options]" }
-    let(:templates_dir) { File.join(aruba.config.root_directory, "spec/templates/#{ENV["DATABASE_ADAPTER"]}") }
+    let(:templates_dir) { File.join(aruba.config.root_directory, "spec/templates/#{IntegrationTestHelper.ruby_version}/#{ENV["DATABASE_ADAPTER"]}") }
 
     it "outputs the help message" do
       _cmd = run_command("bundle exec annotaterb", fail_on_error: true, exit_timeout: command_timeout_seconds)

diff --git a/spec/integration_spec_helper.rb b/spec/integration_spec_helper.rb
@@ -6,3 +6,9 @@
 Aruba.configure do |config|
   config.allow_absolute_paths = true
 end
+
+module IntegrationTestHelper
+  def self.ruby_version
+    RUBY_VERSION.split(".").first(2).join("_") # => "3_0"
+  end
+end