virtualbox 0.1.0

data/test/virtualbox/command_test.rb

@@ -0,0 +1,30 @@
+require File.join(File.dirname(__FILE__), '..', 'test_helper')
+
+class CommandTest < Test::Unit::TestCase
+  context "shell escaping" do
+    should "convert value to string" do
+      assert_nothing_raised do
+        assert_equal "400", VirtualBox::Command.shell_escape(400)
+      end
+    end
+  end
+  
+  context "testing command results" do
+    setup do
+      @command = "foo"
+      VirtualBox::Command.stubs(:execute)
+    end
+    
+    should "return true if the exit code is 0" do
+      system("echo 'hello' 1>/dev/null")
+      assert_equal 0, $?.to_i
+      assert VirtualBox::Command.test(@command)
+    end
+    
+    should "return false if the exit code is 1" do
+      system("there_is_no_way_this_can_exist_1234567890")
+      assert_not_equal 0, $?.to_i
+      assert !VirtualBox::Command.test(@command)      
+    end
+  end
+end

data/test/virtualbox/dvd_test.rb

@@ -0,0 +1,58 @@
+require File.join(File.dirname(__FILE__), '..', 'test_helper')
+
+class DVDTest < Test::Unit::TestCase
+  context "retrieving all dvds" do
+    setup do
+      @expectations = {
+        "d3252617-8176-4f8c-9d73-1c9c82b23960" => {
+          :location   => "/Users/mitchellh/Downloads/jeos-8.04.3-jeos-i386.iso",
+          :accessible => "yes"
+        },
+        
+        "4a08f52c-bca3-4908-8da4-4f48aaa4ebba" => {
+          :location   => "/Applications/VirtualBox.app/Contents/MacOS/VBoxGuestAdditions.iso",
+          :accessible => "yes"
+        }
+      }
+      
+      @valid = <<-valid
+VirtualBox Command Line Management Interface Version 3.1.2
+(C) 2005-2009 Sun Microsystems, Inc.
+All rights reserved.
+
+UUID:       d3252617-8176-4f8c-9d73-1c9c82b23960
+Path:       /Users/mitchellh/Downloads/jeos-8.04.3-jeos-i386.iso
+Accessible: yes
+
+UUID:       4a08f52c-bca3-4908-8da4-4f48aaa4ebba
+Path:       /Applications/VirtualBox.app/Contents/MacOS/VBoxGuestAdditions.iso
+Accessible: yes
+Usage:      TestJeOS (UUID: 3d0f87b4-50f7-4fc5-ad89-93375b1b32a3)
+valid
+
+      VirtualBox::Command.expects(:vboxmanage).with("list dvds").returns(@valid)
+    end
+    
+    should "return an array of DVD objects" do
+      result = VirtualBox::DVD.all
+      assert result.is_a?(Array)
+      
+      result.each { |v| assert v.is_a?(VirtualBox::DVD) }
+    end
+    
+    should "return the proper results" do
+      result = VirtualBox::DVD.all
+      assert result.is_a?(Array)
+      assert_equal @expectations.length, result.length
+      
+      result.each do |image|
+        expected_image = @expectations[image.uuid]
+        assert expected_image
+        
+        expected_image.each do |k,v|
+          assert_equal v, image.read_attribute(k)
+        end
+      end
+    end
+  end
+end

data/test/virtualbox/ext/subclass_listing_test.rb

@@ -0,0 +1,25 @@
+require File.join(File.dirname(__FILE__), '..', '..', 'test_helper')
+require 'virtualbox/ext/subclass_listing'
+
+class SubclassListingTest < Test::Unit::TestCase
+  class A
+    include VirtualBox::SubclassListing
+  end
+  class B < A; end
+  class C < B; end
+  class D < A; end
+  class E
+    include VirtualBox::SubclassListing
+  end
+  class F < E; end
+  
+  should "list subclasses, including sub-subclasses, etc" do
+    assert_equal [F], E.subclasses
+    assert_equal [C], B.subclasses
+    assert_equal [B, C, D], A.subclasses.sort_by { |c| c.name }
+  end
+  
+  should "list direct subclasses if flag is set" do
+    assert_equal [B, D], A.subclasses(true).sort_by { |c| c.name }
+  end
+end

data/test/virtualbox/hard_drive_test.rb

@@ -0,0 +1,161 @@
+require File.join(File.dirname(__FILE__), '..', 'test_helper')
+
+class HardDriveTest < Test::Unit::TestCase
+  setup do
+    VirtualBox::Command.stubs(:execute)
+    
+    @find_raw = <<-raw
+VirtualBox Command Line Management Interface Version 3.1.2
+(C) 2005-2009 Sun Microsystems, Inc.
+All rights reserved.
+
+UUID:                 11dedd14-57a1-4bdb-adeb-dd1d67f066e1
+Accessible:           yes
+Description:          
+Logical size:         20480 MBytes
+Current size on disk: 1218 MBytes
+Type:                 normal (base)
+Storage format:       VDI
+In use by VMs:        FooVM (UUID: 696249ad-00b6-4087-b47f-9b82629efc31)
+Location:             /Users/mitchellh/Library/VirtualBox/HardDisks/foo.vdi
+raw
+    @name = "foo"
+    VirtualBox::Command.stubs(:vboxmanage).with("showhdinfo #{@name}").returns(@find_raw)
+  end
+  
+  context "cloning a hard drive" do
+    setup do
+      @hd = VirtualBox::HardDrive.find(@name)
+      VirtualBox::Command.expects(:vboxmanage).with("clonehd #{@hd.uuid} bar --format VDI --remember").returns(@find_raw)
+    end
+    
+    should "call vboxmanage with the clone command" do
+      VirtualBox::HardDrive.expects(:find).returns(nil)
+      @hd.clone("bar")
+    end
+    
+    should "return the newly cloned hard drive" do
+      @new_hd = mock("hd")
+      VirtualBox::HardDrive.expects(:find).returns(@new_hd)
+      assert_equal @new_hd, @hd.clone("bar")
+    end
+  end
+  
+  context "creating a hard drive" do
+    setup do
+      @location = "foo.foo"
+      @size = "758"
+      @format = "VDI"
+      
+      @hd = VirtualBox::HardDrive.new
+      @hd.location = @location
+      @hd.size = @size
+      
+      @fake_hd = mock("hd")
+      @fake_hd.stubs(:attributes).returns({
+        :uuid => "foo"
+      })
+      
+      VirtualBox::HardDrive.stubs(:find).returns(@fake_hd)
+      VirtualBox::Command.stubs(:vboxmanage).returns("UUID: FOO")
+    end
+    
+    should "call create on save" do
+      @hd.expects(:create).once
+      
+      assert @hd.new_record?
+      @hd.save
+    end
+    
+    should "call not call create on existing records" do
+      @hd.save
+      assert !@hd.new_record?
+      
+      @hd.expects(:create).never
+      @hd.save
+    end
+    
+    should "call createhd" do
+      VirtualBox::Command.expects(:vboxmanage).with("createhd --filename #{@location} --size #{@size} --format #{@format} --remember")
+      @hd.save
+    end
+    
+    should "replace attributes with those of the newly created hard drive" do
+      @hd.save
+      
+      assert_equal "foo", @hd.uuid
+    end
+  end
+  
+  context "finding a single hard drive" do
+    should "parse proper fields" do
+      VirtualBox::Command.expects(:vboxmanage).with("showhdinfo #{@name}").returns(@find_raw)
+
+      @expected = {
+        :uuid => "11dedd14-57a1-4bdb-adeb-dd1d67f066e1",
+        :accessible => "yes",
+        :size => "20480",
+        :location => "/Users/mitchellh/Library/VirtualBox/HardDisks/foo.vdi"
+      }
+      
+      hd = VirtualBox::HardDrive.find(@name)
+      assert hd.is_a?(VirtualBox::HardDrive)
+      
+      @expected.each do |k,v|
+        assert_equal v, hd.send(k)
+      end
+    end
+    
+    should "return nil if finding a non-existent hard drive" do
+      VirtualBox::Command.expects(:vboxmanage).with("showhdinfo 12").returns("UH OH (VERR_FILE_NOT_FOUND)")
+      
+      assert_nothing_raised do
+        assert_nil VirtualBox::HardDrive.find(12)
+      end
+    end
+  end
+  
+  context "retrieving all hard drives" do
+    setup do
+      @valid = <<-valid
+VirtualBox Command Line Management Interface Version 3.1.2
+(C) 2005-2009 Sun Microsystems, Inc.
+All rights reserved.
+      
+UUID:       9d2e4353-d1e9-466c-ac58-f2249264147b
+Format:     VDI
+Location:   /Users/mitchellh/Library/VirtualBox/HardDisks/foo.vdi
+Accessible: yes
+Type:       normal
+Usage:      TestJeOS (UUID: 3d0f87b4-50f7-4fc5-ad89-93375b1b32a3)
+
+UUID:       11dedd14-57a1-4bdb-adeb-dd1d67f066e1
+Format:     VDI
+Location:   /Users/mitchellh/Library/VirtualBox/HardDisks/bar.vdi
+Accessible: yes
+Type:       normal
+Usage:      HoboBase (UUID: 696249ad-00b6-4087-b47f-9b82629efc31)
+
+UUID:       5e090af6-7d71-4f40-8b03-33aa665f9ecf
+Format:     VMDK
+Location:   /Users/mitchellh/Library/VirtualBox/HardDisks/baz.vmdk
+Accessible: yes
+Type:       normal
+Usage:      foo (UUID: 8710d3db-d96a-46ed-9004-59fa891fda90)
+valid
+
+      VirtualBox::Command.expects(:vboxmanage).with("list hdds").returns(@valid)
+      
+      @hd = mock("hd")
+      @hd.stubs(:is_a?).with(VirtualBox::HardDrive).returns(true)
+      VirtualBox::HardDrive.expects(:find).at_least(0).returns(@hd)
+    end
+    
+    should "return an array of HardDrive objects" do
+      result = VirtualBox::HardDrive.all
+      assert result.is_a?(Array)
+      
+      result.each { |v| assert v.is_a?(VirtualBox::HardDrive) }
+    end
+  end
+end

data/test/virtualbox/image_test.rb

@@ -0,0 +1,113 @@
+require File.join(File.dirname(__FILE__), '..', 'test_helper')
+
+class ImageTest < Test::Unit::TestCase
+  context "recording subclasses" do
+    should "list all subclasses" do
+      assert_nothing_raised { VirtualBox::Image.subclasses }
+    end
+  end
+  
+  context "parsing raw" do
+    setup do
+      @raw = <<-raw
+VirtualBox Command Line Management Interface Version 3.1.2
+(C) 2005-2009 Sun Microsystems, Inc.
+All rights reserved.
+
+UUID:       9d2e4353-d1e9-466c-ac58-f2249264147b
+Format:     VDI
+Location:   /Users/mitchellh/Library/VirtualBox/HardDisks/TestJeOS.vdi
+Accessible: yes
+Type:       normal
+Usage:      TestJeOS (UUID: 3d0f87b4-50f7-4fc5-ad89-93375b1b32a3)
+
+UUID:       11dedd14-57a1-4bdb-adeb-dd1d67f066e1
+Format:     VDI
+Location:   /Users/mitchellh/Library/VirtualBox/HardDisks/HoboBase.vdi
+Accessible: yes
+Type:       normal
+Usage:      HoboBase (UUID: 696249ad-00b6-4087-b47f-9b82629efc31)
+
+UUID:       322f79fd-7da6-416f-a16f-e70066ccf165
+Format:     VMDK
+Location:   /Users/mitchellh/Library/VirtualBox/HardDisks/HoboBase.vmdk
+Accessible: yes
+Type:       normal
+Usage:      HoboBase_2 (UUID: 3cc36c5d-562a-4030-8acf-f061f44170c4)      
+raw
+    end
+    
+    should "call parse block the correct number of times" do
+      VirtualBox::Image.expects(:parse_block).times(4).returns({})
+      VirtualBox::Image.parse_raw(@raw)
+    end
+    
+    should "remove nil (invalid) blocks from result" do
+      result = VirtualBox::Image.parse_raw(@raw)
+      assert_equal 3, result.length
+    end
+  end
+  
+  context "parsing multiple blocks" do
+    setup do
+      @raw = <<-raw
+one
+
+two
+
+three
+raw
+    end
+    
+    should "call parse block for each multiline" do
+      parse_seq = sequence("parse")
+      VirtualBox::Image.expects(:parse_block).with("one").in_sequence(parse_seq)
+      VirtualBox::Image.expects(:parse_block).with("two").in_sequence(parse_seq)
+      VirtualBox::Image.expects(:parse_block).with("three").in_sequence(parse_seq)
+      VirtualBox::Image.parse_blocks(@raw)
+    end
+    
+    should "return an array of the parses with nil ignored" do
+      parse_seq = sequence("parse")
+      VirtualBox::Image.expects(:parse_block).with("one").returns({}).in_sequence(parse_seq)
+      VirtualBox::Image.expects(:parse_block).with("two").returns(nil).in_sequence(parse_seq)
+      VirtualBox::Image.expects(:parse_block).with("three").returns({}).in_sequence(parse_seq)
+      result = VirtualBox::Image.parse_blocks(@raw)
+      
+      assert result.is_a?(Array)
+      assert_equal 2, result.length
+    end
+  end
+  
+  context "parsing a single block" do
+    setup do
+      @expected = {
+        :uuid => "1234567890",
+        :path => "foo",
+        :accessible => "yes"
+      }
+        
+      @block = ""
+      @expected.each { |k,v| @block += "#{k}: #{v}\n" }
+    end
+    
+    should "return nil for an invalid block" do
+      assert VirtualBox::Image.parse_block("HI").nil?
+    end
+    
+    should "mirror location to path" do
+      result = VirtualBox::Image.parse_block(@block)
+      assert_equal "foo", result[:location]
+    end
+      
+    should "properly parse if all properties are available" do
+      result = VirtualBox::Image.parse_block(@block)
+      assert !result.nil?
+      
+      @expected.each do |k,v|
+        k = :location if k == :path
+        assert_equal v, result[k]
+      end
+    end
+  end
+end

data/test/virtualbox/nic_test.rb

@@ -0,0 +1,119 @@
+require File.join(File.dirname(__FILE__), '..', 'test_helper')
+
+class NicTest < Test::Unit::TestCase
+  setup do
+    @data = {
+      :nic1 => "bridged",
+      :nic2 => "foo",
+      :nic3 => "bar"
+    }
+    
+    @caller = mock("caller")
+    @caller.stubs(:name).returns("foo")
+    
+    VirtualBox::VM.stubs(:human_info).returns(<<-raw)
+NIC 1:           MAC: 08002745B49F, Attachment: Bridged Interface 'en0: Ethernet', Cable connected: on, Trace: off (file: none), Type: Am79C973, Reported speed: 0 Mbps
+NIC 2:           MAC: 08002745B49F, Attachment: Bridged Interface 'en0: Ethernet', Cable connected: on, Trace: off (file: none), Type: Am79C973, Reported speed: 0 Mbps
+NIC 3:           MAC: 08002745B49F, Attachment: Bridged Interface 'en0: Ethernet', Cable connected: on, Trace: off (file: none), Type: Am79C973, Reported speed: 0 Mbps
+raw
+  end
+  
+  context "saving" do
+    setup do
+      @nic = VirtualBox::Nic.populate_relationship(@caller, @data)
+      @vmname = "myvm"
+    end
+    
+    should "use the vmname strung through the save" do
+      VirtualBox::Command.expects(:vboxmanage).with("modifyvm #{@vmname} --nic1 foo")
+      
+      nic = @nic[0]
+      nic.nic = "foo"
+      nic.save(@vmname)
+    end
+    
+    should "use the proper index" do
+      VirtualBox::Command.expects(:vboxmanage).with("modifyvm #{@vmname} --nic2 far")
+      
+      nic = @nic[1]
+      nic.nic = "far"
+      nic.save(@vmname)
+    end
+    
+    should "save the nictype" do
+      VirtualBox::Command.expects(:vboxmanage).with("modifyvm #{@vmname} --nictype1 ZOO")
+      
+      nic = @nic[0]
+      nic.nictype = "ZOO"
+      assert nic.nictype_changed?
+      nic.save(@vmname)
+    end
+  end
+  
+  context "populating relationships" do
+    setup do
+      @value = VirtualBox::Nic.populate_relationship(@caller, @data)
+    end
+
+    should "create the correct amount of objects" do
+      assert_equal 3, @value.length
+    end
+    
+    should "parse the type" do
+      assert_equal "Am79C973", @value[0].nictype
+    end
+  end
+  
+  context "parsing nic data from human readable output" do
+    setup do
+      @raw = "NIC 1:           MAC: 08002745B49F, Attachment: Bridged Interface 'en0: Ethernet', Cable connected: on, Trace: off (file: none), Type: Am79C973, Reported speed: 0 Mbps"
+      
+      @multiline_raw = <<-raw
+Storage Controller Port Count (0):      2
+Storage Controller Name (1):            Floppy Controller
+Storage Controller Type (1):            I82078
+Storage Controller Instance Number (1): 0
+Storage Controller Max Port Count (1):  1
+Storage Controller Port Count (1):      1
+IDE Controller (0, 0): /Users/mitchellh/Library/VirtualBox/HardDisks/HoboBase.vmdk (UUID: fb0256a9-5685-4bc2-98ff-5b7503586bf3)
+IDE Controller (1, 0): Empty
+Floppy Controller (0, 0): Empty
+NIC 1:           MAC: 08002745B49F, Attachment: Bridged Interface 'en0: Ethernet', Cable connected: on, Trace: off (file: none), Type: Am79C973, Reported speed: 0 Mbps
+NIC 2:           disabled      
+raw
+    end
+    
+    should "only return valid objects in hash" do
+      VirtualBox::VM.expects(:human_info).returns(@multiline_raw)
+      result = VirtualBox::Nic.nic_data("foo")
+      assert result.is_a?(Hash)
+      assert_equal 1, result.length
+    end
+    
+    should "return nil if its an invalid string" do
+      assert_nil VirtualBox::Nic.parse_nic("FOO")
+    end
+    
+    should "return proper data for valid string" do
+      @name = :nic1
+      @expected = {
+        :mac => "08002745B49F",
+        :attachment => "Bridged Interface 'en0: Ethernet'",
+        :trace => "off (file: none)",
+        :type => "Am79C973"
+      }
+      
+      name, result = VirtualBox::Nic.parse_nic(@raw)
+      assert_equal @name, name
+      assert result.is_a?(Hash)
+      
+      @expected.each do |k,v|
+        assert_equal v, result[k]
+      end
+    end
+    
+    should "ignore nics that are disabled" do
+      assert_nil VirtualBox::Nic.parse_nic("NIC 1:    disabled")
+    end
+  end
+end