Test and fix `bundle viz` command deprecation

3 files changed, 23 insertions, 2 deletions

diff --git a/lib/bundler/feature_flag.rb b/lib/bundler/feature_flag.rb
index e01676f97d..5e5d97ec42 100644
--- a/lib/bundler/feature_flag.rb
+++ b/lib/bundler/feature_flag.rb
@@ -56,7 +56,7 @@ module Bundler
     settings_flag(:unlock_source_unlocks_spec) { !bundler_2_mode? }
     settings_flag(:update_requires_all_flag) { bundler_3_mode? }
     settings_flag(:use_gem_version_promoter_for_major_updates) { bundler_2_mode? }
-    settings_flag(:viz_command) { !bundler_2_mode? }
+    settings_flag(:viz_command) { !bundler_3_mode? }
 
     settings_option(:default_cli_command) { bundler_2_mode? ? :cli_help : :install }
 
diff --git a/spec/commands/viz_spec.rb b/spec/commands/viz_spec.rb
index c4e77b9179..96cc21b5f7 100644
--- a/spec/commands/viz_spec.rb
+++ b/spec/commands/viz_spec.rb
@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
-RSpec.describe "bundle viz", :bundler => "< 2", :if => Bundler.which("dot") do
+RSpec.describe "bundle viz", :bundler => "< 3", :if => Bundler.which("dot") do
   let(:ruby_graphviz) do
     graphviz_glob = base_system_gems.join("cache/ruby-graphviz*")
     Pathname.glob(graphviz_glob).first
diff --git a/spec/other/major_deprecation_spec.rb b/spec/other/major_deprecation_spec.rb
index ab294ad444..3eeeb32b89 100644
--- a/spec/other/major_deprecation_spec.rb
+++ b/spec/other/major_deprecation_spec.rb
@@ -328,4 +328,25 @@ The :gist git source is deprecated, and will be removed in the future. Add this
         "bundle console will be replaced by `bin/console` generated by `bundle gem <name>`"
     end
   end
+
+  context "bundle viz" do
+    let(:ruby_graphviz) do
+      graphviz_glob = base_system_gems.join("cache/ruby-graphviz*")
+      Pathname.glob(graphviz_glob).first
+    end
+
+    before do
+      system_gems ruby_graphviz
+      create_file "gems.rb"
+      bundle "viz"
+    end
+
+    it "does not print a deprecation warning", :bundler => "< 2" do
+      expect(deprecations).to be_empty
+    end
+
+    it "prints a deprecation warning", :bundler => "2" do
+      expect(deprecations).to include "The `viz` command has been moved to the `bundle-viz` gem, see https://github.com/bundler/bundler-viz"
+    end
+  end
 end