lib/berkshelf/commands/test_command.rb in berkshelf-2.0.3 vs lib/berkshelf/commands/test_command.rb in berkshelf-2.0.4

@@ -1,11 +1,13 @@
-require 'kitchen/cli'
+begin
+  require 'kitchen/cli'
+rescue LoadError; end
 
 module Berkshelf
   class TestCommand < Kitchen::CLI
     namespace "test"
   end
 
   class Cli < Thor
     register(Berkshelf::TestCommand, 'test', 'test [COMMAND]', 'Testing tasks for your cookbook', hide: true)
   end
-end
+end if defined?(Kitchen::CLI)