package iso import ( "fmt" "reflect" "testing" "github.com/hashicorp/packer/builder/virtualbox/common" "github.com/hashicorp/packer/packer" ) func testConfig() map[string]interface{} { return map[string]interface{}{ "iso_checksum": "foo", "iso_checksum_type": "md5", "iso_url": "http://www.google.com/", "shutdown_command": "yes", "ssh_username": "foo", packer.BuildNameConfigKey: "foo", } } func TestBuilder_ImplementsBuilder(t *testing.T) { var raw interface{} raw = &Builder{} if _, ok := raw.(packer.Builder); !ok { t.Error("Builder must implement builder.") } } func TestBuilderPrepare_Defaults(t *testing.T) { var b Builder config := testConfig() warns, err := b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("should not have error: %s", err) } if b.config.GuestAdditionsMode != common.GuestAdditionsModeUpload { t.Errorf("bad guest additions mode: %s", b.config.GuestAdditionsMode) } if b.config.GuestOSType != "Other" { t.Errorf("bad guest OS type: %s", b.config.GuestOSType) } if b.config.VMName == "" { t.Errorf("bad vm name: %s", b.config.VMName) } if b.config.Format != "ovf" { t.Errorf("bad format: %s", b.config.Format) } } func TestBuilderPrepare_DiskSize(t *testing.T) { var b Builder config := testConfig() delete(config, "disk_size") warns, err := b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("bad err: %s", err) } if b.config.DiskSize != 40000 { t.Fatalf("bad size: %d", b.config.DiskSize) } config["disk_size"] = 60000 b = Builder{} warns, err = b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("should not have error: %s", err) } if b.config.DiskSize != 60000 { t.Fatalf("bad size: %d", b.config.DiskSize) } } func TestBuilderPrepare_FloppyFiles(t *testing.T) { var b Builder config := testConfig() delete(config, "floppy_files") warns, err := b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("bad err: %s", err) } if len(b.config.FloppyFiles) != 0 { t.Fatalf("bad: %#v", b.config.FloppyFiles) } floppies_path := "../../../common/test-fixtures/floppies" config["floppy_files"] = []string{fmt.Sprintf("%s/bar.bat", floppies_path), fmt.Sprintf("%s/foo.ps1", floppies_path)} b = Builder{} warns, err = b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("should not have error: %s", err) } expected := []string{fmt.Sprintf("%s/bar.bat", floppies_path), fmt.Sprintf("%s/foo.ps1", floppies_path)} if !reflect.DeepEqual(b.config.FloppyFiles, expected) { t.Fatalf("bad: %#v", b.config.FloppyFiles) } } func TestBuilderPrepare_InvalidFloppies(t *testing.T) { var b Builder config := testConfig() config["floppy_files"] = []string{"nonexistent.bat", "nonexistent.ps1"} b = Builder{} _, errs := b.Prepare(config) if errs == nil { t.Fatalf("Nonexistent floppies should trigger multierror") } if len(errs.(*packer.MultiError).Errors) != 2 { t.Fatalf("Multierror should work and report 2 errors") } } func TestBuilderPrepare_GuestAdditionsMode(t *testing.T) { var b Builder config := testConfig() // test default mode delete(config, "guest_additions_mode") warns, err := b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("bad err: %s", err) } // Test another mode config["guest_additions_mode"] = "attach" b = Builder{} warns, err = b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("should not have error: %s", err) } if b.config.GuestAdditionsMode != common.GuestAdditionsModeAttach { t.Fatalf("bad: %s", b.config.GuestAdditionsMode) } // Test bad mode config["guest_additions_mode"] = "teleport" b = Builder{} warns, err = b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err == nil { t.Fatal("should error") } } func TestBuilderPrepare_GuestAdditionsPath(t *testing.T) { var b Builder config := testConfig() delete(config, "guest_additions_path") warns, err := b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("bad err: %s", err) } if b.config.GuestAdditionsPath != "VBoxGuestAdditions.iso" { t.Fatalf("bad: %s", b.config.GuestAdditionsPath) } config["guest_additions_path"] = "foo" b = Builder{} warns, err = b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("should not have error: %s", err) } if b.config.GuestAdditionsPath != "foo" { t.Fatalf("bad size: %s", b.config.GuestAdditionsPath) } } func TestBuilderPrepare_GuestAdditionsSHA256(t *testing.T) { var b Builder config := testConfig() delete(config, "guest_additions_sha256") warns, err := b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("bad err: %s", err) } if b.config.GuestAdditionsSHA256 != "" { t.Fatalf("bad: %s", b.config.GuestAdditionsSHA256) } config["guest_additions_sha256"] = "FOO" b = Builder{} warns, err = b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("should not have error: %s", err) } if b.config.GuestAdditionsSHA256 != "foo" { t.Fatalf("bad size: %s", b.config.GuestAdditionsSHA256) } } func TestBuilderPrepare_GuestAdditionsURL(t *testing.T) { var b Builder config := testConfig() config["guest_additions_url"] = "" warns, err := b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("err: %s", err) } if b.config.GuestAdditionsURL != "" { t.Fatalf("should be empty: %s", b.config.GuestAdditionsURL) } config["guest_additions_url"] = "http://www.packer.io" b = Builder{} warns, err = b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Errorf("should not have error: %s", err) } } func TestBuilderPrepare_HardDriveInterface(t *testing.T) { var b Builder config := testConfig() // Test a default boot_wait delete(config, "hard_drive_interface") warns, err := b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("err: %s", err) } if b.config.HardDriveInterface != "ide" { t.Fatalf("bad: %s", b.config.HardDriveInterface) } // Test with a bad config["hard_drive_interface"] = "fake" b = Builder{} warns, err = b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err == nil { t.Fatal("should have error") } // Test with a good config["hard_drive_interface"] = "sata" b = Builder{} warns, err = b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("should not have error: %s", err) } } func TestBuilderPrepare_InvalidKey(t *testing.T) { var b Builder config := testConfig() // Add a random key config["i_should_not_be_valid"] = true warns, err := b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err == nil { t.Fatal("should have error") } } func TestBuilderPrepare_ISOInterface(t *testing.T) { var b Builder config := testConfig() // Test a default boot_wait delete(config, "iso_interface") warns, err := b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("err: %s", err) } if b.config.ISOInterface != "ide" { t.Fatalf("bad: %s", b.config.ISOInterface) } // Test with a bad config["iso_interface"] = "fake" b = Builder{} warns, err = b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err == nil { t.Fatal("should have error") } // Test with a good config["iso_interface"] = "sata" b = Builder{} warns, err = b.Prepare(config) if len(warns) > 0 { t.Fatalf("bad: %#v", warns) } if err != nil { t.Fatalf("should not have error: %s", err) } }