Add tests for NestedBlock

tfwrite/nested_block_test.go

@@ -0,0 +1,42 @@
+package tfwrite
+
+import (
+	"testing"
+)
+
+func TestNestedBlockType(t *testing.T) {
+	cases := []struct {
+		desc string
+		src  string
+		name string
+		want string
+		ok   bool
+	}{
+		{
+			desc: "simple",
+			src: `
+foo {
+  nested {
+	  bar = "baz"
+  }
+}
+`,
+			name: "nested",
+			want: "nested",
+			ok:   true,
+		},
+	}
+
+	for _, tc := range cases {
+		t.Run(tc.desc, func(t *testing.T) {
+			f := parseTestFile(t, tc.src)
+			b := findFirstTestBlock(t, f)
+			nestedBlocks := b.FindNestedBlocksByType(tc.name)
+
+			got := nestedBlocks[0].Type()
+			if got != tc.want {
+				t.Errorf("got = %s, but want = %s", got, tc.want)
+			}
+		})
+	}
+}

tfwrite/resource_test.go

@@ -6,6 +6,40 @@ import (
 	"github.com/google/go-cmp/cmp"
 )
 
+func TestResourceType(t *testing.T) {
+	cases := []struct {
+		desc string
+		src  string
+		want string
+		ok   bool
+	}{
+		{
+			desc: "simple",
+			src: `
+resource "foo_test" "example" {
+  nested {
+	  bar = "baz"
+  }
+}
+`,
+			want: "resource",
+			ok:   true,
+		},
+	}
+
+	for _, tc := range cases {
+		t.Run(tc.desc, func(t *testing.T) {
+			f := parseTestFile(t, tc.src)
+			r := findFirstTestResource(t, f)
+
+			got := r.Type()
+			if got != tc.want {
+				t.Errorf("got = %s, but want = %s", got, tc.want)
+			}
+		})
+	}
+}
+
 func TestResourceSchemaType(t *testing.T) {
 	cases := []struct {
 		desc string