pleschev-vagrant-hostmaster 0.8.1

data/lib/vagrant_init.rb

@@ -0,0 +1 @@
+require 'vagrant/hostmaster'

data/test/vagrant/hostmaster/command_tests/list_test.rb

@@ -0,0 +1,43 @@
+require 'test/unit'
+require 'mocha'
+require 'vagrant/hostmaster'
+require 'vagrant/hostmaster/command/list'
+
+module Vagrant
+  module Hostmaster
+    module CommandTests
+      class ListTest < Test::Unit::TestCase
+        include Vagrant::TestHelpers
+
+        def setup
+          @env = vagrant_env
+          @vm = @env.vms.values.first
+          @vm.stubs(:state).returns(:running)
+          @command = Vagrant::Hostmaster::Command::List.new([], @env)
+        end
+
+        def test_not_created
+          @vm.stubs(:state).returns(:not_created)
+          Vagrant::Hostmaster::Command::List.expects(:new).with([], @env).returns(@command)
+          Vagrant::Hostmaster::VM.expects(:new).never
+          @env.cli('hosts', 'list')
+        end
+
+        def test_not_running
+          @vm.stubs(:state).returns(:poweroff)
+          Vagrant::Hostmaster::Command::List.expects(:new).with([], @env).returns(@command)
+          Vagrant::Hostmaster::VM.expects(:new).never
+          @env.cli('hosts', 'list')
+        end
+
+        def test_list
+          Vagrant::Hostmaster::Command::List.stubs(:new).with([], @env).returns(@command)
+          vm = Vagrant::Hostmaster::VM.new(@vm)
+          Vagrant::Hostmaster::VM.expects(:new).with(@vm).returns(vm)
+          vm.expects(:list).with()
+          @env.cli('hosts', 'list')
+        end
+      end
+    end
+  end
+end

data/test/vagrant/hostmaster/command_tests/remove_test.rb

@@ -0,0 +1,43 @@
+require 'test/unit'
+require 'mocha'
+require 'vagrant/hostmaster'
+require 'vagrant/hostmaster/command/remove'
+
+module Vagrant
+  module Hostmaster
+    module CommandTests
+      class RemoveTest < Test::Unit::TestCase
+        include Vagrant::TestHelpers
+
+        def setup
+          @env = vagrant_env
+          @vm = @env.vms.values.first
+          @vm.stubs(:state).returns(:running)
+          @command = Vagrant::Hostmaster::Command::Remove.new([], @env)
+        end
+
+        def test_not_created
+          @vm.stubs(:state).returns(:not_created)
+          Vagrant::Hostmaster::Command::Remove.expects(:new).with([], @env).returns(@command)
+          Vagrant::Hostmaster::VM.expects(:new).never
+          @env.cli('hosts', 'remove')
+        end
+
+        def test_not_running
+          @vm.stubs(:state).returns(:poweroff)
+          Vagrant::Hostmaster::Command::Remove.expects(:new).with([], @env).returns(@command)
+          Vagrant::Hostmaster::VM.expects(:new).never
+          @env.cli('hosts', 'remove')
+        end
+
+        def test_remove
+          Vagrant::Hostmaster::Command::Remove.stubs(:new).with([], @env).returns(@command)
+          vm = Vagrant::Hostmaster::VM.new(@vm)
+          Vagrant::Hostmaster::VM.expects(:new).with(@vm).returns(vm)
+          vm.expects(:remove).with()
+          @env.cli('hosts', 'remove')
+        end
+      end
+    end
+  end
+end

data/test/vagrant/hostmaster/command_tests/root_test.rb

@@ -0,0 +1,49 @@
+require 'test/unit'
+require 'mocha'
+require 'vagrant/hostmaster'
+require 'vagrant/hostmaster/command/list'
+require 'vagrant/hostmaster/command/remove'
+require 'vagrant/hostmaster/command/update'
+
+module Vagrant
+  module Hostmaster
+    module CommandTests
+      class RootTest < Test::Unit::TestCase
+        include Vagrant::TestHelpers
+
+        def setup
+          @env = vagrant_env
+          @vm = @env.vms.values.first
+        end
+
+        def test_no_command
+          command = Vagrant::Hostmaster::Command::Root.new([], @env)
+          Vagrant::Hostmaster::Command::Root.expects(:new).with([], @env).returns(command)
+          command.expects(:help)
+          @env.cli('hosts')
+        end
+
+        def test_list
+          command = Vagrant::Hostmaster::Command::List.new([], @env)
+          Vagrant::Hostmaster::Command::List.expects(:new).with([], @env).returns(command)
+          command.expects(:execute).with()
+          @env.cli('hosts', 'list')
+        end
+
+        def test_remove
+          command = Vagrant::Hostmaster::Command::Remove.new([], @env)
+          Vagrant::Hostmaster::Command::Remove.expects(:new).with([], @env).returns(command)
+          command.expects(:execute).with()
+          @env.cli('hosts', 'remove')
+        end
+
+        def test_update
+          command = Vagrant::Hostmaster::Command::Update.new([], @env)
+          Vagrant::Hostmaster::Command::Update.expects(:new).with([], @env).returns(command)
+          command.expects(:execute).with()
+          @env.cli('hosts', 'update')
+        end
+      end
+    end
+  end
+end

data/test/vagrant/hostmaster/command_tests/update_test.rb

@@ -0,0 +1,43 @@
+require 'test/unit'
+require 'mocha'
+require 'vagrant/hostmaster'
+require 'vagrant/hostmaster/command/update'
+
+module Vagrant
+  module Hostmaster
+    module CommandTests
+      class UpdateTest < Test::Unit::TestCase
+        include Vagrant::TestHelpers
+
+        def setup
+          @env = vagrant_env
+          @vm = @env.vms.values.first
+          @vm.stubs(:state).returns(:running)
+          @command = Vagrant::Hostmaster::Command::Update.new([], @env)
+        end
+
+        def test_not_created
+          @vm.stubs(:state).returns(:not_created)
+          Vagrant::Hostmaster::Command::Update.expects(:new).with([], @env).returns(@command)
+          Vagrant::Hostmaster::VM.expects(:new).never
+          @env.cli('hosts', 'update')
+        end
+
+        def test_not_running
+          @vm.stubs(:state).returns(:poweroff)
+          Vagrant::Hostmaster::Command::Update.expects(:new).with([], @env).returns(@command)
+          Vagrant::Hostmaster::VM.expects(:new).never
+          @env.cli('hosts', 'update')
+        end
+
+        def test_update
+          Vagrant::Hostmaster::Command::Update.stubs(:new).with([], @env).returns(@command)
+          vm = Vagrant::Hostmaster::VM.new(@vm)
+          Vagrant::Hostmaster::VM.expects(:new).with(@vm).returns(vm)
+          vm.expects(:update).with()
+          @env.cli('hosts', 'update')
+        end
+      end
+    end
+  end
+end

data/test/vagrant/hostmaster/config_test.rb

@@ -0,0 +1,29 @@
+require 'test/unit'
+require "mocha"
+require 'vagrant/hostmaster'
+
+module Vagrant
+  module Hostmaster
+    class ConfigTest < Test::Unit::TestCase
+      include Vagrant::TestHelpers
+
+      def setup
+        @config = Vagrant::Hostmaster::Config.new
+        @errors = Vagrant::Config::ErrorRecorder.new
+        @env = vagrant_env
+      end
+
+      def test_default_valid
+        @config.validate(@env, @errors)
+        assert @errors.errors.empty?, "Default configuration should not have any errors."
+      end
+
+      def test_valid
+        @config.name = 'hostmaster.dev'
+        @config.aliases = %w(www.hostmaster.dev)
+        @config.validate(@env, @errors)
+        assert @errors.errors.empty?, "Configuration should not have any errors."
+      end
+    end
+  end
+end

data/test/vagrant/hostmaster/middleware_tests/remove_test.rb

@@ -0,0 +1,40 @@
+require 'test/unit'
+require "mocha"
+require 'vagrant/hostmaster'
+
+module Vagrant
+  module Hostmaster
+    module MiddlewareTests
+      class RemoveTest < Test::Unit::TestCase
+        include Vagrant::TestHelpers
+
+        def setup
+          @app, @env = action_env(vagrant_env.vms.values.first.env)
+          @env['vm'].stubs(:state).returns(:running)
+          @middleware = Vagrant::Hostmaster::Middleware::Remove.new(@app, @env)
+        end
+
+        def test_not_created
+          @env['vm'].stubs(:state).returns(:not_created)
+          Vagrant::Hostmaster::VM.expects(:new).never
+          @middleware.call(@env)
+        end
+
+        def test_not_running
+          @env['vm'].stubs(:state).returns(:poweroff)
+          vm = Vagrant::Hostmaster::VM.new(@env['vm'])
+          Vagrant::Hostmaster::VM.expects(:new).with(@env['vm']).returns(vm)
+          vm.expects(:remove).with(:guests => false)
+          @middleware.call(@env)
+        end
+
+        def test_remove
+          vm = Vagrant::Hostmaster::VM.new(@env['vm'])
+          Vagrant::Hostmaster::VM.expects(:new).with(@env['vm']).returns(vm)
+          vm.expects(:remove).with(:guests => false)
+          @middleware.call(@env)
+        end
+      end
+    end
+  end
+end

data/test/vagrant/hostmaster/middleware_tests/update_test.rb

@@ -0,0 +1,38 @@
+require 'test/unit'
+require "mocha"
+require 'vagrant/hostmaster'
+
+module Vagrant
+  module Hostmaster
+    module MiddlewareTests
+      class UpdateTest < Test::Unit::TestCase
+        include Vagrant::TestHelpers
+
+        def setup
+          @app, @env = action_env(vagrant_env.vms.values.first.env)
+          @env['vm'].stubs(:state).returns(:running)
+          @middleware = Vagrant::Hostmaster::Middleware::Update.new(@app, @env)
+        end
+
+        def test_not_created
+          @env['vm'].stubs(:state).returns(:not_created)
+          Vagrant::Hostmaster::VM.expects(:new).never
+          @middleware.call(@env)
+        end
+
+        def test_not_running
+          @env['vm'].stubs(:state).returns(:poweroff)
+          Vagrant::Hostmaster::VM.expects(:new).never
+          @middleware.call(@env)
+        end
+
+        def test_update
+          vm = Vagrant::Hostmaster::VM.new(@env['vm'])
+          Vagrant::Hostmaster::VM.expects(:new).with(@env['vm']).returns(vm)
+          vm.expects(:update).with()
+          @middleware.call(@env)
+        end
+      end
+    end
+  end
+end

data/test/vagrant/hostmaster/multiple_vm_test.rb

@@ -0,0 +1,88 @@
+require 'test/unit'
+require "mocha"
+require 'vagrant/hostmaster'
+require 'vagrant/hostmaster/test_helpers'
+
+module Vagrant
+  module Hostmaster
+    class MultipleVMTest < Test::Unit::TestCase
+      include Vagrant::TestHelpers
+      include Vagrant::Hostmaster::TestHelpers
+
+      def setup
+        super
+
+        # TODO: test both windows and non-windows
+        Util::Platform.stubs(:windows?).returns(false)
+        @hosts_file = Tempfile.new('hosts')
+        hostmaster_box :one, 'one.hostmaster.dev', '10.0.0.100', '11111111-1111-1111-1111-111111111111'
+        hostmaster_box :two, 'two.hostmaster.dev', '10.0.0.200', '22222222-2222-2222-2222-222222222222'
+        @vms = hostmaster_vms(hostmaster_env)
+      end
+
+      def teardown
+        @hosts_file.unlink
+        clean_paths
+        super
+      end
+
+      def test_add
+        Vagrant::Hostmaster::VM.stubs(:hosts_path).returns @hosts_file.path
+        @vms.each do |vm|
+          hostmaster_boxes.each do |name,box|
+            next if name == vm.name
+            param = %Q(sh -c 'echo \"#{box[:address]}  #{box[:host_name]}  # VAGRANT: #{vm.uuid} \(#{name}\)\" >>#{vm.hosts_path}')
+            vm.channel.expects(:sudo).with(param)
+          end
+          vm.add
+        end
+        assert_local_host_entries_for @vms, @hosts_file
+      end
+
+      def test_list
+        Vagrant::Hostmaster::VM.stubs(:hosts_path).returns @hosts_file.path
+        write_local_host_entries @hosts_file
+        hostmaster_boxes.each do |target_name,target_box|
+          address, host_name, uuid, vm = target_box.values_at(:address, :host_name, :uuid, :vm)
+          hostmaster_boxes.each do |name,box|
+            next if name == target_name
+            param = "grep '# VAGRANT: #{uuid} (#{name})$' #{vm.hosts_path}"
+            vm.channel.expects(:execute).with(param, :error_check => false)
+          end
+          output = Vagrant::Hostmaster::UI::Capture.capture { vm.list }
+          assert_match /^\[local\] #{address} #{host_name} # VAGRANT: #{uuid} \(#{target_name}\)$/, output
+        end
+      end
+
+      def test_remove_local_hosts_entry
+        Vagrant::Hostmaster::VM.stubs(:hosts_path).returns @hosts_file.path
+        write_local_host_entries @hosts_file
+        @vms.each do |vm|
+          hostmaster_boxes.each do |name,box|
+            next if name == vm.name
+            param = %Q(sed -e '/# VAGRANT: #{vm.uuid} (#{name})$/ d' -ibak #{vm.hosts_path})
+            vm.channel.expects(:sudo).with(param)
+          end
+          vm.remove
+        end
+        assert_no_local_host_entries_for @vms, @hosts_file
+      end
+
+      def test_update_local_hosts_entry
+        Vagrant::Hostmaster::VM.stubs(:hosts_path).returns @hosts_file.path
+        write_local_host_entries @hosts_file, :one => {:address => '10.10.10.11', :host_name => 'a.hostmaster.dev'}, :two => {:address => '10.10.10.22', :host_name => 'b.hostmaster.dev'}
+        @vms.each do |vm|
+          hostmaster_boxes.each do |name,box|
+            next if name == vm.name
+            param = %Q(sed -e '/# VAGRANT: #{vm.uuid} (#{name})$/ d' -ibak #{vm.hosts_path})
+            vm.channel.expects(:sudo).with(param).returns(0)
+            param = %Q(sh -c 'echo \"#{box[:address]}  #{box[:host_name]}  # VAGRANT: #{vm.uuid} \(#{name}\)\" >>#{vm.hosts_path}')
+            vm.channel.expects(:sudo).with(param).returns(0)
+          end
+          vm.update
+        end
+        assert_local_host_entries_for @vms, @hosts_file
+      end
+    end
+  end
+end

data/test/vagrant/hostmaster/simple_vm_test.rb

@@ -0,0 +1,78 @@
+require 'test/unit'
+require "mocha"
+require 'vagrant/hostmaster'
+require 'vagrant/hostmaster/test_helpers'
+
+module Vagrant
+  module Hostmaster
+    class SimpleVMTest < Test::Unit::TestCase
+      include Vagrant::TestHelpers
+      include Vagrant::Hostmaster::TestHelpers
+
+      def setup
+        super
+
+        # TODO: test both windows and non-windows
+        Util::Platform.stubs(:windows?).returns(false)
+        @hosts_file = Tempfile.new('hosts')
+        hostmaster_box :default, 'hostmaster.dev', '123.45.67.89', '01234567-89ab-cdef-fedc-ba9876543210'
+        @vms = hostmaster_vms(hostmaster_env)
+      end
+
+      def teardown
+        @hosts_file.unlink
+        clean_paths
+        super
+      end
+
+      def test_hosts_path
+        assert_equal '/etc/hosts', Vagrant::Hostmaster::VM.hosts_path
+      end
+
+      def test_hosts_path_on_windows
+        Util::Platform.stubs(:windows?).returns(true)
+        ENV.stubs(:[]).with('SYSTEMROOT').returns('C:\Windows')
+        Vagrant::Hostmaster::VM.stubs(:expand_path).with('system32/drivers/etc/hosts', 'C:\Windows').returns('C:/Windows/system32/drivers/etc/hosts')
+        assert_equal 'C:/Windows/system32/drivers/etc/hosts', Vagrant::Hostmaster::VM.hosts_path
+      end
+
+      def test_vm_hosts_path
+        assert_equal '/etc/hosts', @vms.first.hosts_path
+      end
+
+      def test_vm_hosts_path_on_windows
+        Util::Platform.stubs(:windows?).returns(true)
+        assert_equal '/etc/hosts', @vms.first.hosts_path
+      end
+
+      def test_add_local_hosts_entry
+        Vagrant::Hostmaster::VM.stubs(:hosts_path).returns @hosts_file.path
+        @vms.each { |vm| vm.add }
+        assert_local_host_entries_for @vms, @hosts_file
+      end
+
+      def test_list_local_hosts_entry
+        Vagrant::Hostmaster::VM.stubs(:hosts_path).returns @hosts_file.path
+        write_local_host_entries @hosts_file
+        hostmaster_boxes.each do |name,box|
+          output = Vagrant::Hostmaster::UI::Capture.capture { box[:vm].list }
+          assert_match /^\[local\]\s+#{box[:address]}\s+#{box[:host_name]}\s*# VAGRANT: #{box[:uuid]} \(#{name}\)$/, output
+        end
+      end
+
+      def test_remove_local_hosts_entry
+        Vagrant::Hostmaster::VM.stubs(:hosts_path).returns @hosts_file.path
+        write_local_host_entries @hosts_file
+        @vms.each { |vm| vm.remove }
+        assert_no_local_host_entries_for @vms, @hosts_file
+      end
+
+      def test_update
+        Vagrant::Hostmaster::VM.stubs(:hosts_path).returns @hosts_file.path
+        write_local_host_entries @hosts_file, :default => {:address => '10.10.10.10', :host_name => 'www.hostmaster.dev'}
+        @vms.each { |vm| vm.update }
+        assert_local_host_entries_for @vms, @hosts_file
+      end
+    end
+  end
+end