package digitalocean import ( "strconv" "testing" "time" "github.com/hashicorp/packer/packer" ) func testConfig() map[string]interface{} { return map[string]interface{}{ "api_token": "bar", "region": "nyc2", "size": "512mb", "ssh_username": "root", "image": "foo", } } func TestBuilder_ImplementsBuilder(t *testing.T) { var raw interface{} raw = &Builder{} if _, ok := raw.(packer.Builder); !ok { t.Fatalf("Builder should be a builder") } } func TestBuilder_Prepare_BadType(t *testing.T) { b := &Builder{} c := map[string]interface{}{ "api_key": []string{}, } warnings, err := b.Prepare(c) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err == nil { t.Fatalf("prepare should fail") } } func TestBuilderPrepare_InvalidKey(t *testing.T) { var b Builder config := testConfig() // Add a random key config["i_should_not_be_valid"] = true warnings, err := b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err == nil { t.Fatal("should have error") } } func TestBuilderPrepare_Region(t *testing.T) { var b Builder config := testConfig() // Test default delete(config, "region") warnings, err := b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err == nil { t.Fatalf("should error") } expected := "sfo1" // Test set config["region"] = expected b = Builder{} warnings, err = b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err != nil { t.Fatalf("should not have error: %s", err) } if b.config.Region != expected { t.Errorf("found %s, expected %s", b.config.Region, expected) } } func TestBuilderPrepare_Size(t *testing.T) { var b Builder config := testConfig() // Test default delete(config, "size") warnings, err := b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err == nil { t.Fatalf("should error") } expected := "1024mb" // Test set config["size"] = expected b = Builder{} warnings, err = b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err != nil { t.Fatalf("should not have error: %s", err) } if b.config.Size != expected { t.Errorf("found %s, expected %s", b.config.Size, expected) } } func TestBuilderPrepare_Image(t *testing.T) { var b Builder config := testConfig() // Test default delete(config, "image") warnings, err := b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err == nil { t.Fatal("should error") } expected := "ubuntu-14-04-x64" // Test set config["image"] = expected b = Builder{} warnings, err = b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err != nil { t.Fatalf("should not have error: %s", err) } if b.config.Image != expected { t.Errorf("found %s, expected %s", b.config.Image, expected) } } func TestBuilderPrepare_StateTimeout(t *testing.T) { var b Builder config := testConfig() // Test default warnings, err := b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err != nil { t.Fatalf("should not have error: %s", err) } if b.config.StateTimeout != 6*time.Minute { t.Errorf("invalid: %s", b.config.StateTimeout) } // Test set config["state_timeout"] = "5m" b = Builder{} warnings, err = b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err != nil { t.Fatalf("should not have error: %s", err) } // Test bad config["state_timeout"] = "tubes" b = Builder{} warnings, err = b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err == nil { t.Fatal("should have error") } } func TestBuilderPrepare_SnapshotTimeout(t *testing.T) { var b Builder config := testConfig() // Test default warnings, err := b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err != nil { t.Fatalf("should not have error: %s", err) } if b.config.SnapshotTimeout != 60*time.Minute { t.Errorf("invalid: %s", b.config.SnapshotTimeout) } // Test set config["snapshot_timeout"] = "15m" b = Builder{} warnings, err = b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err != nil { t.Fatalf("should not have error: %s", err) } // Test bad config["snapshot_timeout"] = "badstring" b = Builder{} warnings, err = b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err == nil { t.Fatal("should have error") } } func TestBuilderPrepare_PrivateNetworking(t *testing.T) { var b Builder config := testConfig() // Test default warnings, err := b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err != nil { t.Fatalf("should not have error: %s", err) } if b.config.PrivateNetworking != false { t.Errorf("invalid: %t", b.config.PrivateNetworking) } // Test set config["private_networking"] = true b = Builder{} warnings, err = b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err != nil { t.Fatalf("should not have error: %s", err) } if b.config.PrivateNetworking != true { t.Errorf("invalid: %t", b.config.PrivateNetworking) } } func TestBuilderPrepare_SnapshotName(t *testing.T) { var b Builder config := testConfig() // Test default warnings, err := b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err != nil { t.Fatalf("should not have error: %s", err) } if b.config.SnapshotName == "" { t.Errorf("invalid: %s", b.config.SnapshotName) } // Test set config["snapshot_name"] = "foobarbaz" b = Builder{} warnings, err = b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err != nil { t.Fatalf("should not have error: %s", err) } // Test set with template config["snapshot_name"] = "{{timestamp}}" b = Builder{} warnings, err = b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err != nil { t.Fatalf("should not have error: %s", err) } _, err = strconv.ParseInt(b.config.SnapshotName, 0, 0) if err != nil { t.Fatalf("failed to parse int in template: %s", err) } } func TestBuilderPrepare_DropletName(t *testing.T) { var b Builder config := testConfig() // Test default warnings, err := b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err != nil { t.Fatalf("should not have error: %s", err) } if b.config.DropletName == "" { t.Errorf("invalid: %s", b.config.DropletName) } // Test normal set config["droplet_name"] = "foobar" b = Builder{} warnings, err = b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err != nil { t.Fatalf("should not have error: %s", err) } // Test with template config["droplet_name"] = "foobar-{{timestamp}}" b = Builder{} warnings, err = b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err != nil { t.Fatalf("should not have error: %s", err) } // Test with bad template config["droplet_name"] = "foobar-{{" b = Builder{} warnings, err = b.Prepare(config) if len(warnings) > 0 { t.Fatalf("bad: %#v", warnings) } if err == nil { t.Fatal("should have error") } }