bbrowning-virtualbox 0.7.6.dev

data/test/virtualbox/storage_controller_test.rb

@@ -0,0 +1,197 @@
+require File.expand_path("../../test_helper", __FILE__)
+
+class StorageControllerTest < Test::Unit::TestCase
+  setup do
+    @interface = mock("interface")
+    @parent = mock("parent")
+
+    @klass = VirtualBox::StorageController
+  end
+
+  context "class methods" do
+    context "populating relationship" do
+      setup do
+        # Defaulting for non-specified parameters
+        @interface.stubs(:is_a?).returns(false)
+      end
+
+      should "populate array relationship if IMachine is given" do
+        @interface.expects(:is_a?).with(VirtualBox::COM::Util.versioned_interface(:Machine)).returns(true)
+        @klass.expects(:populate_array_relationship).once.with(@parent, @interface)
+        @klass.populate_relationship(@parent, @interface)
+      end
+
+      should "populate attachment relationship if MediumAttachment is given" do
+        @interface.expects(:is_a?).with(VirtualBox::MediumAttachment).returns(true)
+        @klass.expects(:populate_attachment_relationship).once.with(@parent, @interface)
+        @klass.populate_relationship(@parent, @interface)
+      end
+    end
+
+    context "populating array (has many) relationship" do
+      setup do
+        @instance = mock("instance")
+
+        @interface.stubs(:storage_controllers).returns([])
+
+        @klass.stubs(:device_type).returns(:all)
+        @klass.stubs(:new).returns(@instance)
+      end
+
+      def mock_controller(name)
+        controller = mock(name)
+        controller
+      end
+
+      should "return a proxied collection" do
+        result = @klass.populate_array_relationship(nil, @interface)
+        assert result.is_a?(VirtualBox::Proxies::Collection)
+      end
+
+      should "call new for every medium if device type is all" do
+        controllers = []
+        @interface.stubs(:storage_controllers).returns(controllers)
+        5.times { |i| controllers << mock_controller("c#{i}") }
+
+        expected_result = []
+        new_seq = sequence("new_seq")
+        controllers.each do |controller|
+          expected_value = "instance-#{controller.inspect}"
+          @klass.expects(:new).with(@parent, controller).in_sequence(new_seq).returns(expected_value)
+          expected_result << expected_value
+        end
+
+        assert_equal expected_result, @klass.populate_array_relationship(@parent, @interface)
+      end
+    end
+
+    context "populating relationship for a MediumAttachment" do
+      setup do
+        @controllers = []
+
+        @machine = mock("machine")
+        @machine.stubs(:storage_controllers).returns(@controllers)
+
+        @interface.stubs(:parent).returns(@machine)
+      end
+
+      should "return nil if no controllers match" do
+        assert_nil @klass.populate_attachment_relationship(@parent, @interface)
+      end
+
+      should "return the controller with matching name" do
+        name = :foo
+        @interface.stubs(:controller_name).returns(name)
+
+        controller = mock("controller")
+        controller.stubs(:name).returns(:foo)
+        @controllers << controller
+
+        assert_equal controller, @klass.populate_attachment_relationship(@parent, @interface)
+      end
+    end
+
+    context "saving relationship" do
+      should "call save on each item" do
+        items = (1..5).to_a.collect do |i|
+          item = mock("item-#{i}")
+          item.expects(:save).once
+          item
+        end
+
+        @klass.save_relationship(nil, items)
+      end
+    end
+  end
+
+  context "initializing" do
+    setup do
+      @klass.any_instance.stubs(:load_interface_attributes)
+    end
+
+    should "load interface attribtues" do
+      @klass.any_instance.expects(:load_interface_attributes).with(@interface).once
+      @klass.new(@parent, @interface)
+    end
+
+    should "not be dirty" do
+      @instance = @klass.new(@parent, @interface)
+      assert !@instance.changed?
+    end
+
+    should "be existing record" do
+      @instance = @klass.new(@parent, @interface)
+      assert !@instance.new_record?
+    end
+
+    should "setup parent" do
+      @instance = @klass.new(@parent, @interface)
+      assert_equal @parent, @instance.parent
+    end
+
+    should "setup interface" do
+      @instance = @klass.new(@parent, @interface)
+      assert_equal @interface, @instance.interface
+    end
+  end
+
+  context "instance methods" do
+    setup do
+      @klass.any_instance.stubs(:load_interface_attributes)
+
+      @parent = mock("parent")
+      @interface = mock("interface")
+      @instance = @klass.new(@parent, @interface)
+    end
+
+    context "medium attachments" do
+      setup do
+        @medium_attachments = []
+        @parent.stubs(:medium_attachments).returns(@medium_attachments)
+      end
+
+      def mock_medium_attachment(sc)
+        ma = mock("ma-#{sc.inspect}")
+        ma.stubs(:storage_controller).returns(sc)
+        ma
+      end
+
+      should "return all medium attachments which match the storage controller" do
+        foo_ma = mock_medium_attachment(@instance)
+        @medium_attachments << foo_ma
+        @medium_attachments << mock_medium_attachment(:bar)
+
+        assert_equal [foo_ma], @instance.medium_attachments
+      end
+    end
+
+    context "destroying" do
+      setup do
+        @ma = []
+        @instance.stubs(:medium_attachments).returns(@ma)
+
+        @machine = mock("machine")
+        @session = mock("session")
+        @session.stubs(:machine).returns(@machine)
+        @parent.stubs(:with_open_session).yields(@session)
+      end
+
+      should "remove all the attachments" do
+        ma = mock("medium_attachment")
+        @ma << ma
+
+        destroy_seq = sequence("destroy_seq")
+        ma.expects(:destroy).with(1,2,3).once.in_sequence(destroy_seq)
+        @machine.expects(:remove_storage_controller).with(@instance.name).in_sequence(destroy_seq)
+
+        @instance.destroy(1,2,3)
+      end
+
+      should "remove from the parent and save" do
+        destroy_seq = sequence("destroy_seq")
+        @machine.expects(:remove_storage_controller).with(@instance.name).in_sequence(destroy_seq)
+        @instance.destroy
+      end
+    end
+  end
+end

data/test/virtualbox/system_properties_test.rb

@@ -0,0 +1,87 @@
+require File.expand_path("../../test_helper", __FILE__)
+
+class SystemPropertiesTest < Test::Unit::TestCase
+  setup do
+    @klass = VirtualBox::SystemProperties
+    @interface = mock("interface")
+  end
+
+  context "initializing" do
+    should "load attributes from the machine" do
+      @klass.any_instance.expects(:initialize_attributes).with(@interface).once
+      @klass.new(@interface)
+    end
+  end
+
+  context "initializing attributes" do
+    setup do
+      @klass.any_instance.stubs(:load_interface_attributes)
+    end
+
+    should "load interface attribtues" do
+      @klass.any_instance.expects(:load_interface_attributes).with(@interface).once
+      @klass.new(@interface)
+    end
+
+    should "setup the interface" do
+      instance = @klass.new(@interface)
+      assert_equal @interface, instance.interface
+    end
+
+    should "not be dirty" do
+      @instance = @klass.new(@interface)
+      assert !@instance.changed?
+    end
+
+    should "be existing record" do
+      @instance = @klass.new(@interface)
+      assert !@instance.new_record?
+    end
+  end
+
+  context "class methods" do
+    context "populating relationship" do
+      setup do
+        @instance = mock("instance")
+
+        @klass.stubs(:new).returns(@instance)
+      end
+
+      should "return a SystemProperties instance" do
+        result = @klass.populate_relationship(nil, @interface)
+        assert_equal @instance, result
+      end
+
+      should "call new for every shared folder" do
+        @klass.expects(:new).with(@interface).returns(@instance)
+        result = @klass.populate_relationship(nil, @interface)
+        assert_equal @instance, result
+      end
+    end
+
+    context "saving relationship" do
+      setup do
+        @item = mock("item")
+      end
+
+      should "just call save on the item" do
+        @item.expects(:save)
+        @klass.save_relationship(nil, @item)
+      end
+    end
+  end
+
+  context "instance methods" do
+    setup do
+      @klass.any_instance.stubs(:load_interface_attributes)
+      @instance = @klass.new(@interface)
+    end
+
+    context "saving" do
+      should "save changed interface attributes" do
+        @instance.expects(:save_changed_interface_attributes).with(@interface)
+        @instance.save
+      end
+    end
+  end
+end

data/test/virtualbox/usb_controller_test.rb

@@ -0,0 +1,112 @@
+require File.expand_path("../../test_helper", __FILE__)
+
+class USBControllerTest < Test::Unit::TestCase
+  setup do
+    @klass = VirtualBox::USBController
+    @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)
+    end
+
+    should "load interface attribtues" do
+      @klass.any_instance.expects(:load_interface_attributes).with(@interface).once
+      @klass.new(@parent, @interface)
+    end
+
+    should "populate relationships" do
+      @klass.any_instance.expects(:populate_relationships).with(@interface).once
+      @klass.new(@parent, @interface)
+    end
+
+    should "setup the parent" do
+      instance = @klass.new(@parent, @interface)
+      assert_equal @parent, instance.parent
+    end
+
+    should "setup the interface" do
+      instance = @klass.new(@parent, @interface)
+      assert_equal @interface, instance.interface
+    end
+
+    should "not be dirty" do
+      @instance = @klass.new(@parent, @interface)
+      assert !@instance.changed?
+    end
+
+    should "be existing record" do
+      @instance = @klass.new(@parent, @interface)
+      assert !@instance.new_record?
+    end
+  end
+
+  context "class methods" do
+    context "populating relationship" do
+      setup do
+        @instance = mock("instance")
+
+        @klass.stubs(:new).returns(@instance)
+
+        @controller_interface = mock("controller_interface")
+        @interface.stubs(:usb_controller).returns(@controller_interface)
+      end
+
+      should "return a USBController instance" do
+        result = @klass.populate_relationship(nil, @interface)
+        assert_equal @instance, result
+      end
+
+      should "call new with the interface" do
+        @klass.expects(:new).with(@parent, @controller_interface).returns(@instance)
+        result = @klass.populate_relationship(@parent, @interface)
+        assert_equal @instance, result
+      end
+    end
+
+    context "saving relationship" do
+      setup do
+        @item = mock("item")
+      end
+
+      should "just call save on the item" do
+        @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_relationship)
+      @instance = @klass.new(@parent, @interface)
+    end
+
+    context "saving" do
+      setup do
+        @session = mock("session")
+        @machine = mock("machine")
+        @usb_controller = mock("usb_controller")
+        @session.stubs(:machine).returns(@machine)
+        @machine.stubs(:usb_controller).returns(@usb_controller)
+        @parent.stubs(:with_open_session).yields(@session)
+      end
+
+      should "save on the locked interface" do
+        @instance.expects(:save_changed_interface_attributes).with(@usb_controller).once
+        @instance.save
+      end
+    end
+  end
+end

data/test/virtualbox/usb_device_filter_test.rb

@@ -0,0 +1,93 @@
+require File.expand_path("../../test_helper", __FILE__)
+
+class USBDeviceFilterTestTest < Test::Unit::TestCase
+  setup do
+    @klass = VirtualBox::USBDeviceFilter
+    @interface = mock("interface")
+  end
+
+  context "class methods" do
+    context "populating relationships" do
+      setup do
+        @instance = mock("instance")
+
+        @interface.stubs(:device_filters).returns([])
+        @klass.stubs(:new).returns(@instance)
+      end
+
+      should "return a proxied collection" do
+        result = @klass.populate_relationship(nil, @interface)
+        assert result.is_a?(VirtualBox::Proxies::Collection)
+      end
+
+      should "call new for every interface" do
+        collection = []
+        @interface.stubs(:device_filters).returns(collection)
+        5.times { |i| collection << mock("a#{i}") }
+
+        expected_result = []
+        new_seq = sequence("new_seq")
+        collection.each do |item|
+          expected_value = "instance-#{item.inspect}"
+          @klass.expects(:new).with(item).in_sequence(new_seq).returns(expected_value)
+          expected_result << expected_value
+        end
+
+        assert_equal expected_result, @klass.populate_relationship(@parent, @interface)
+      end
+    end
+
+    context "saving relationship" do
+      setup do
+        @item = mock("item")
+      end
+
+      should "just call save on the item" do
+        @item.expects(:save)
+        @klass.save_relationship(nil, @item)
+      end
+    end
+  end
+
+  context "initializing" do
+    should "load attributes from the machine" do
+      @klass.any_instance.expects(:initialize_attributes).with(@interface).once
+      @klass.new(@interface)
+    end
+  end
+
+  context "initializing attributes" do
+    setup do
+      @klass.any_instance.stubs(:load_interface_attributes)
+    end
+
+    should "load interface attribtues" do
+      @klass.any_instance.expects(:load_interface_attributes).with(@interface).once
+      @klass.new(@interface)
+    end
+
+    should "not be dirty" do
+      @instance = @klass.new(@interface)
+      assert !@instance.changed?
+    end
+
+    should "be existing record" do
+      @instance = @klass.new(@interface)
+      assert !@instance.new_record?
+    end
+  end
+
+  context "instance methods" do
+    setup do
+      @klass.any_instance.stubs(:load_interface_attributes)
+
+      @parent = mock("parent")
+      @interface = mock("interface")
+      @instance = @klass.new(@interface)
+      @instance.stubs(:interface).returns(@interface)
+      @instance.stubs(:parent).returns(@parent)
+      @collection = VirtualBox::Proxies::Collection.new(@parent)
+      @collection << @instance
+    end
+  end
+end