terraform-provider-greenhost/digitalocean/resource_digitalocean_droplet_migrate_test.go

package digitalocean

import (
	"reflect"
	"testing"

	"github.com/hashicorp/terraform-plugin-sdk/v2/terraform"
)

func TestDigitalOceanDropletMigrateState(t *testing.T) {
	cases := map[string]struct {
		StateVersion int
		ID           string
		Attributes   map[string]string
		Expected     map[string]string
	}{
		"v0_1_with_values": {
			StateVersion: 0,
			ID:           "id",
			Attributes: map[string]string{
				"backups":    "true",
				"monitoring": "false",
			},
			Expected: map[string]string{
				"backups":    "true",
				"monitoring": "false",
			},
		},
		"v0_1_without_values": {
			StateVersion: 0,
			ID:           "id",
			Attributes:   map[string]string{},
			Expected: map[string]string{
				"backups":    "false",
				"monitoring": "false",
			},
		},
	}

	for tn, tc := range cases {
		is := &terraform.InstanceState{
			ID:         tc.ID,
			Attributes: tc.Attributes,
		}
		is, err := resourceDigitalOceanDropletMigrateState(tc.StateVersion, is, nil)

		if err != nil {
			t.Fatalf("bad: %q, err: %#v", tn, err)
		}

		if !reflect.DeepEqual(tc.Expected, is.Attributes) {
			t.Fatalf("Bad Droplet Migrate\n\n. Got: %+v\n\n expected: %+v", is.Attributes, tc.Expected)
		}
	}
}