Add support for return ... blocks in acctest

README.md

@@ -35,6 +35,7 @@ The tool currently supports blocks with the following start and end lines:
 |```terraform        |`,  |
 |return fmt.Sprintf(`|`,  |
 |return fmt.Sprintf(`|`)  |
+|return `            |`   |
 
 ### Extract Terraform Blocks
 

lib/blocks/blockreader.go

@@ -64,6 +64,8 @@ func ReaderIgnore(br *Reader, number int, line string) error {
 func IsStartLine(line string) bool {
 	if strings.HasSuffix(line, "return fmt.Sprintf(`\n") { // acctest
 		return true
+	} else if strings.HasSuffix(line, "return `\n") { // acctest
+		return true
 	} else if strings.HasPrefix(line, "```hcl") { // documentation
 		return true
 	} else if strings.HasPrefix(line, "```terraform") { // documentation
@@ -78,6 +80,8 @@ func IsStartLine(line string) bool {
 func IsFinishLine(line string) bool {
 	if accTestFinishLineWithLeadingSpacesMatcher.MatchString(line) { // acctest
 		return true
+	} else if strings.HasSuffix(line, "`\n") { // acctest
+		return true
 	} else if strings.HasPrefix(line, "```") { // documentation
 		return true
 	}

lib/blocks/testdata/test1.go

@@ -12,6 +12,14 @@ resource "aws_s3_bucket" "simple" {
 `)
 }
 
+func testReturnStringSimple() string {
+	return `
+resource "aws_s3_bucket" "simple2" {
+  bucket = "tf-test-bucket-simple2"
+}
+`
+}
+
 func testReturnSprintfWithParameters(randInt int) string {
 	return fmt.Sprintf(`
 resource "aws_s3_bucket" "with-parameters" {

lib/blocks/testdata/test1_results.yaml

@@ -3,6 +3,10 @@ expected_results:
     resource "aws_s3_bucket" "simple" {
       bucket = "tf-test-bucket-simple"
     }
+  - |
+    resource "aws_s3_bucket" "simple2" {
+      bucket = "tf-test-bucket-simple2"
+    }
   - |
     resource "aws_s3_bucket" "with-parameters" {
       bucket = "tf-test-bucket-with-parameters-%d"