test/test_stdout.rb in beaker-hostgenerator-0.6.0 vs test/test_stdout.rb in beaker-hostgenerator-0.7.0

@@ -1,6 +1,7 @@
 require 'minitest/autorun'
+require 'beaker-hostgenerator/cli'
 require 'beaker-hostgenerator/data'
 
 class TestStdout < Minitest::Test
 
   def setup
@@ -10,14 +11,10 @@
     $stderr = @stderr
     $stdout = @stdout
   end
 
   def test_default_list_output
-    begin
-      BeakerHostGenerator::CLI.new(['--list'])
-    rescue BeakerHostGenerator::Exceptions::SafeEarlyExit
-    end
-
+    BeakerHostGenerator::CLI.new(['--list']).execute!
     BeakerHostGenerator::Data.get_platforms(0).each do |spec_key|
       assert_match(/.*#{spec_key}.*/, @stdout.string)
     end
   end
 end