virtualbox 0.7.1 → 0.7.2

data/test/virtualbox/nat_engine_test.rb

@@ -0,0 +1,106 @@
+require File.join(File.dirname(__FILE__), '..', 'test_helper')
+
+class NATEngineTest < Test::Unit::TestCase
+  setup do
+    @klass = VirtualBox::NATEngine
+    @interface = mock("interface")
+    @parent = mock("parent")
+  end
+
+  context "initializing" do
+    should "load attributes from the machine" do
+      @klass.any_instance.expects(:initialize_attributes).with(@parent, @interface).once
+      @klass.new(@parent, @interface)
+    end
+  end
+
+  context "initializing attributes" do
+    setup do
+      @klass.any_instance.stubs(:load_interface_attributes)
+      @klass.any_instance.stubs(:populate_relationships)
+
+      @instance = @klass.new(@parent, @interface)
+    end
+
+    should "load interface attribtues" do
+      @klass.any_instance.expects(:load_interface_attributes).with(@interface).once
+      @klass.new(@parent, @interface)
+    end
+
+    should "setup the parent" do
+      assert_equal @parent, @instance.parent
+    end
+
+    should "setup the interface" do
+      assert_equal @interface, @instance.interface
+    end
+
+    should "not be dirty" do
+      assert !@instance.changed?
+    end
+
+    should "be existing record" do
+      assert !@instance.new_record?
+    end
+  end
+
+  context "class methods" do
+    context "populating relationship" do
+      should "return nil if the interface is nil" do
+        assert_nil @klass.populate_relationship(@parent, nil)
+      end
+
+      should "return an initialized NATEngine" do
+        @klass.any_instance.stubs(:load_interface_attributes)
+        @klass.any_instance.stubs(:populate_relationships)
+        result = @klass.populate_relationship(@parent, @interface)
+        assert result.is_a?(@klass)
+        assert_equal @parent, result.parent
+        assert_equal @interface, result.interface
+      end
+    end
+
+    context "saving relationship" do
+      should "call save on each item" do
+        item = mock("item")
+        item.expects(:save)
+        @klass.save_relationship(nil, item)
+      end
+    end
+  end
+
+  context "instance methods" do
+    setup do
+      @klass.any_instance.stubs(:load_interface_attributes)
+      @klass.any_instance.stubs(:populate_relationships)
+      @instance = @klass.new(@parent, @interface)
+    end
+
+    context "saving" do
+      setup do
+        @engine = mock("engine")
+        @instance.stubs(:modify_engine).yields(@engine)
+      end
+
+      should "save the interface attributes and relationships" do
+        @instance.expects(:save_changed_interface_attributes).with(@engine).once
+        @instance.expects(:save_relationships).once
+        @instance.save
+      end
+    end
+
+    context "modify engine" do
+      setup do
+        @adapter = mock("adapter")
+        @adapter.stubs(:nat_driver).returns(mock("nat_driver"))
+        @parent.stubs(:modify_adapter).yields(@adapter)
+      end
+
+      should "yield the NAT engine" do
+        @instance.modify_engine do |engine|
+          assert_equal @adapter.nat_driver, engine
+        end
+      end
+    end
+  end
+end

data/test/virtualbox/nat_forwarded_port_test.rb

@@ -0,0 +1,216 @@
+require File.join(File.dirname(__FILE__), '..', 'test_helper')
+
+class NATForwardedPortTest < Test::Unit::TestCase
+  setup do
+    @klass = VirtualBox::NATForwardedPort
+
+    @interface = mock("interface")
+    @caller = mock("caller")
+    @caller.stubs(:interface).returns(@interface)
+  end
+
+  context "validations" do
+    setup do
+      @collection = VirtualBox::Proxies::Collection.new(@caller)
+
+      @port = @klass.new
+      @port.name = "foo"
+      @port.guestport = "22"
+      @port.hostport = "2222"
+      @port.added_to_relationship(@collection)
+    end
+
+    should "be valid with all fields" do
+      assert @port.valid?
+    end
+
+    should "be invalid with no name" do
+      @port.name = nil
+      assert !@port.valid?
+    end
+
+    should "be invalid with no guest port" do
+      @port.guestport = nil
+      assert !@port.valid?
+    end
+
+    should "be invalid with no host port" do
+      @port.hostport = nil
+      assert !@port.valid?
+    end
+
+    should "be invalid if not in a relationship" do
+      @port.write_attribute(:parent, nil)
+      assert !@port.valid?
+    end
+  end
+
+  context "with an instance" do
+    setup do
+      @port = @klass.new({
+        :name => "foo",
+        :guestport => "22",
+        :hostport => "2222"
+      })
+
+      @collection = VirtualBox::Proxies::Collection.new(@caller)
+      @collection << @port
+
+      @port.clear_dirty!
+    end
+
+    context "initializing a new record" do
+      setup do
+        @port = @klass.new
+      end
+
+      should "be a new record" do
+        assert @port.new_record?
+      end
+
+      should "not be dirty" do
+        assert !@port.changed?
+      end
+    end
+
+    context "saving" do
+      setup do
+        @interface.stubs(:add_redirect)
+        @port.stubs(:destroy)
+      end
+
+      context "an existing record" do
+        setup do
+          @port.existing_record!
+          @caller.stubs(:modify_engine)
+        end
+
+        should "not do anything and return true if its unchanged" do
+          @caller.expects(:modify_engine).never
+          assert @port.save
+        end
+
+        should "clear the dirty state after saving" do
+          @port.name = "diff"
+          @port.save
+          assert !@port.changed?
+        end
+
+        should "call destroy if not a new record" do
+          @port.name = "diff"
+          @port.expects(:destroy).once
+          @port.save
+        end
+      end
+
+      context "a new record" do
+        setup do
+          @port.stubs(:valid?).returns(true)
+          @port.new_record!
+          assert @port.new_record?
+
+          @caller.stubs(:modify_engine)
+        end
+
+        should "no longer be a new record after saving" do
+          @port.save
+          assert !@port.new_record?
+        end
+
+        should "not call destroy" do
+          @port.expects(:destroy).never
+          @port.save
+        end
+
+        should "raise a ValidationFailedException if invalid and raise_errors is true" do
+          @port.expects(:valid?).returns(false)
+          assert_raises(VirtualBox::Exceptions::ValidationFailedException) {
+            @port.save
+          }
+        end
+
+        should "add the redirect to the nat engine" do
+          nat = mock("nat")
+          @caller.stubs(:modify_engine).yields(nat)
+          nat.expects(:add_redirect).with(@port.name,
+                                                 @port.protocol,
+                                                 "", @port.hostport,
+                                                 "", @port.guestport)
+          @port.save
+        end
+      end
+    end
+
+    context "destroying" do
+      setup do
+        @interface.stubs(:remove_redirect)
+        @nat = mock("nat")
+        @nat.stubs(:remove_redirect)
+        @caller.stubs(:modify_engine).yields(@nat)
+      end
+
+      should "remove itself from it's collection" do
+        assert @collection.include?(@port)
+        @port.destroy
+        assert !@collection.include?(@port)
+      end
+
+      should "remove the redirect from the nat engine interface" do
+        @nat.expects(:remove_redirect).with(@port.name).once
+        @port.destroy
+      end
+
+      should "do nothing if the record is new" do
+        @port.new_record!
+        @nat.expects(:remove_redirect).never
+        @port.destroy
+      end
+
+      should "be a new record after destroying" do
+        @port.destroy
+        assert @port.new_record?
+      end
+    end
+  end
+
+  context "relationships" do
+    context "saving" do
+      should "call #save on every object" do
+        objects = []
+        5.times do |i|
+          object = mock("object#{i}")
+          object.expects(:save).once
+          objects.push(object)
+        end
+
+        @klass.save_relationship(@caller, objects)
+      end
+    end
+
+    context "populating" do
+      setup do
+        @interface.stubs(:redirects).returns([
+                                              "foo,1,,2222,,22"
+                                              ])
+        @objects = @klass.populate_relationship(@caller, @interface)
+      end
+
+      should "return an array of ForwardedPorts" do
+        assert @objects.is_a?(VirtualBox::Proxies::Collection)
+        assert @objects.all? { |o| o.is_a?(@klass) }
+      end
+
+      should "have the proper data" do
+        object = @objects.first
+        assert_equal "22", object.guestport
+        assert_equal "2222", object.hostport
+        assert_equal :tcp, object.protocol
+        assert_equal @objects, object.parent_collection
+      end
+
+      should "be existing records" do
+        assert !@objects.first.new_record?
+      end
+    end
+  end
+end

data/test/virtualbox/network_adapter_test.rb

@@ -136,6 +136,7 @@ class NetworkAdapterTest < Test::Unit::TestCase
       should "save the attachment type and interface attributes on the open adapter" do
         @instance.expects(:save_attachment_type).with(@adapter).once
         @instance.expects(:save_changed_interface_attributes).with(@adapter).once
+        @instance.expects(:save_relationships).once
         @instance.save
       end
     end
@@ -171,6 +172,7 @@ class NetworkAdapterTest < Test::Unit::TestCase
         @machine = mock("machine")
         @session = mock("session")
         @session.stubs(:machine).returns(@machine)
+        @instance.expects(:parent_machine).returns(@parent)
 
         @adapter = mock("adapter")
       end

data/test/virtualbox/shared_folder_test.rb

@@ -89,6 +89,7 @@ class SharedFolderTest < Test::Unit::TestCase
     setup do
       @klass.any_instance.stubs(:load_interface_attributes)
       @instance = @klass.new(@interface)
+      @instance.stubs(:parent_machine).returns(@parent)
       @collection = VirtualBox::Proxies::Collection.new(@parent)
       @collection << @instance
 

data/test/virtualbox/vm_test.rb

@@ -318,8 +318,7 @@ class VMTest < Test::Unit::TestCase
           :stop => :power_down,
           :pause => :pause,
           :resume => :resume,
-          :save_state => :save_state,
-          :discard_state => [:forget_saved_state, true]
+          :save_state => :save_state
         }
 
         methods.each do |method, control|
@@ -328,6 +327,21 @@ class VMTest < Test::Unit::TestCase
           @instance.send(method)
         end
       end
+
+      context "discard state" do
+        setup do
+          @session = mock("session")
+          @console = mock("console")
+          @session.stubs(:console).returns(@console)
+          @instance.stubs(:with_open_session)
+        end
+
+        should "discard the session in an open state" do
+          @instance.expects(:with_open_session).yields(@session)
+          @console.expects(:forget_saved_state).with(true).once
+          @instance.discard_state
+        end
+      end
     end
 
     context "saving" do

data/virtualbox.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{virtualbox}
-  s.version = "0.7.1"
+  s.version = "0.7.2"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Mitchell Hashimoto"]
-  s.date = %q{2010-06-10}
+  s.date = %q{2010-06-17}
   s.description = %q{Create and modify virtual machines in VirtualBox using pure ruby.}
   s.email = %q{mitchell.hashimoto@gmail.com}
   s.extra_rdoc_files = [
@@ -32,6 +32,7 @@ Gem::Specification.new do |s|
      "lib/virtualbox/abstract_model/interface_attributes.rb",
      "lib/virtualbox/abstract_model/relatable.rb",
      "lib/virtualbox/abstract_model/validatable.rb",
+     "lib/virtualbox/abstract_model/version_matcher.rb",
      "lib/virtualbox/appliance.rb",
      "lib/virtualbox/audio_adapter.rb",
      "lib/virtualbox/bios.rb",
@@ -196,6 +197,8 @@ Gem::Specification.new do |s|
      "lib/virtualbox/media.rb",
      "lib/virtualbox/medium.rb",
      "lib/virtualbox/medium_attachment.rb",
+     "lib/virtualbox/nat_engine.rb",
+     "lib/virtualbox/nat_forwarded_port.rb",
      "lib/virtualbox/network_adapter.rb",
      "lib/virtualbox/proxies/collection.rb",
      "lib/virtualbox/shared_folder.rb",
@@ -214,6 +217,7 @@ Gem::Specification.new do |s|
      "test/virtualbox/abstract_model/interface_attributes_test.rb",
      "test/virtualbox/abstract_model/relatable_test.rb",
      "test/virtualbox/abstract_model/validatable_test.rb",
+     "test/virtualbox/abstract_model/version_matcher_test.rb",
      "test/virtualbox/abstract_model_test.rb",
      "test/virtualbox/appliance_test.rb",
      "test/virtualbox/audio_adapter_test.rb",
@@ -245,6 +249,8 @@ Gem::Specification.new do |s|
      "test/virtualbox/lib_test.rb",
      "test/virtualbox/medium_attachment_test.rb",
      "test/virtualbox/medium_test.rb",
+     "test/virtualbox/nat_engine_test.rb",
+     "test/virtualbox/nat_forwarded_port_test.rb",
      "test/virtualbox/network_adapter_test.rb",
      "test/virtualbox/proxies/collection_test.rb",
      "test/virtualbox/shared_folder_test.rb",
@@ -274,10 +280,12 @@ Gem::Specification.new do |s|
      "test/virtualbox/vm_test.rb",
      "test/virtualbox/virtual_system_description_test.rb",
      "test/virtualbox/vrdp_server_test.rb",
+     "test/virtualbox/nat_forwarded_port_test.rb",
      "test/virtualbox/appliance_test.rb",
      "test/virtualbox/medium_attachment_test.rb",
      "test/virtualbox/audio_adapter_test.rb",
      "test/virtualbox/cpu_test.rb",
+     "test/virtualbox/abstract_model/version_matcher_test.rb",
      "test/virtualbox/abstract_model/validatable_test.rb",
      "test/virtualbox/abstract_model/attributable_test.rb",
      "test/virtualbox/abstract_model/relatable_test.rb",
@@ -315,7 +323,8 @@ Gem::Specification.new do |s|
      "test/virtualbox/ext/byte_normalizer_test.rb",
      "test/virtualbox/ext/subclass_listing_test.rb",
      "test/virtualbox/ext/platform_test.rb",
-     "test/virtualbox/version_test.rb"
+     "test/virtualbox/version_test.rb",
+     "test/virtualbox/nat_engine_test.rb"
   ]
 
   if s.respond_to? :specification_version then