extend tests

tests/terraform/references/main.tf

@@ -14,16 +14,12 @@ provider "aws" {
   }
 }
 
-resource "aws_s3_bucket" "unencrypted-bucket" {
-  bucket = "my-unencrypted-bucket"
-}
-
 resource "aws_s3_bucket" "aes-encrypted-bucket" {
   bucket = "my-aes-encrypted-bucket"
 }
 
 resource "aws_s3_bucket_server_side_encryption_configuration" "aes-encrypted-configuration" {
-  bucket_ref = aws_s3_bucket.aes-encrypted-bucket.bucket
+  bucket = aws_s3_bucket.aes-encrypted-bucket.bucket
 
   rule {
     apply_server_side_encryption_by_default {
@@ -37,11 +33,25 @@ resource "aws_s3_bucket" "kms-encrypted-bucket" {
 }
 
 resource "aws_s3_bucket_server_side_encryption_configuration" "kms-encrypted-configuration" {
-  bucket_ref = aws_s3_bucket.kms-encrypted-bucket.bucket
+  bucket = aws_s3_bucket.kms-encrypted-bucket.bucket
 
   rule {
     apply_server_side_encryption_by_default {
       sse_algorithm = "aws:kms"
     }
   }
 }
+
+resource "aws_s3_bucket" "sample-bucket" {
+  bucket = "sample-bucket"
+}
+
+resource "aws_s3_bucket" "log-bucket" {
+  bucket = "log-bucket"
+}
+
+resource "aws_s3_bucket_logging" "example" {
+  bucket        = aws_s3_bucket.sample-bucket.id
+  target_bucket = aws_s3_bucket.log-bucket.id
+  target_prefix = "log/"
+}

tests/test_tfparse.py

@@ -423,24 +423,38 @@ def test_references(tmp_path):
     mod_path = init_module("references", tmp_path)
     parsed = load_from_path(mod_path)
 
-    aes_bucket, kms_bucket, _ = parsed["aws_s3_bucket"]
+    aes_bucket, kms_bucket, log_bucket, sample_bucket = parsed["aws_s3_bucket"]
     config1, config2 = parsed["aws_s3_bucket_server_side_encryption_configuration"]
-    assert config1["bucket_ref"] == "my-aes-encrypted-bucket"
+    assert config1["bucket"] == "my-aes-encrypted-bucket"
     assert config1["__tfmeta"]["references"] == [
         {
             "id": aes_bucket["id"],
             "label": "aws_s3_bucket",
             "name": "aes-encrypted-bucket",
         },
     ]
-    assert config2["bucket_ref"] == "my-kms-encrypted-bucket"
+    assert config2["bucket"] == "my-kms-encrypted-bucket"
     assert config2["__tfmeta"]["references"] == [
         {
             "id": kms_bucket["id"],
             "label": "aws_s3_bucket",
             "name": "kms-encrypted-bucket",
         },
     ]
+    # all reference to other blocks are reported
+    [bucket_logging] = parsed["aws_s3_bucket_logging"]
+    assert bucket_logging["__tfmeta"]["references"] == [
+        {
+            "id": sample_bucket["id"],
+            "label": "aws_s3_bucket",
+            "name": "sample-bucket",
+        },
+        {
+            "id": log_bucket["id"],
+            "label": "aws_s3_bucket",
+            "name": "log-bucket",
+        },
+    ]
 
 
 def test_modules_located_above_root(tmp_path):