Add reload! to lotus console

lib/lotus/commands/console.rb

@@ -3,6 +3,13 @@
 module Lotus
   module Commands
     class Console
+      module Methods
+        def reload!
+          puts 'Reloading...'
+          Kernel.exec $0
+        end
+      end
+
       ENGINES = {
         'pry'  => 'Pry',
         'ripl' => 'Ripl',
@@ -20,6 +27,9 @@ def start
         ARGV.shift until ARGV.empty?
         require File.expand_path(options[:applications], Dir.pwd)
 
+        # Add convenience methods to the main:Object binding
+        TOPLEVEL_BINDING.eval('self').send(:include, Methods)
+
         engine.start
       end
 
@@ -28,6 +38,7 @@ def engine
       end
 
       private
+
       def engine_lookup
         (ENGINES.find {|_, klass| Object.const_defined?(klass) } || default_engine).first
       end

test/commands/console_test.rb

@@ -203,4 +203,58 @@ def remove_engine(engine)
       end
     end
   end
+
+  describe 'convenience methods' do
+    before do
+      @old_main = TOPLEVEL_BINDING
+      @main     = Minitest::Mock.new
+      Lotus::Utils::IO.silence_warnings { TOPLEVEL_BINDING = @main }
+      @main.expect(:eval, TOPLEVEL_BINDING, ['self'])
+
+      @engine = Minitest::Mock.new
+      @engine.expect(:start, nil)
+    end
+
+    after do
+      Lotus::Utils::IO.silence_warnings { TOPLEVEL_BINDING = @old_main }
+    end
+
+    it 'mixes convenience methods into the TOPLEVEL_BINDING' do
+      @main.expect(:include, true, [Lotus::Commands::Console::Methods])
+
+      opts[:applications] = 'test/fixtures/microservices/config/applications'
+      console.stub(:engine, @engine) { console.start }
+
+      @engine.verify
+      @main.verify
+    end
+  end
+end
+
+describe Lotus::Commands::Console::Methods do
+  describe '#reload!' do
+    before do
+      @binding = Class.new
+      @binding.send(:include, Lotus::Commands::Console::Methods)
+
+      @old_kernel = Kernel
+      Lotus::Utils::IO.silence_warnings { Kernel = Minitest::Mock.new }
+      Kernel.expect(:exec, true, [$0])
+    end
+
+    after do
+      Lotus::Utils::IO.silence_warnings { Kernel = @old_kernel }
+    end
+
+    it 're-executes the running process' do
+      begin
+        $stdout = StringIO.new
+        @binding.new.reload!
+      ensure
+        $stdout = STDOUT
+      end
+
+      Kernel.verify
+    end
+  end
 end