extract: add subtest using targetvar

Author: bschaatsbergen

statecheck/extract_bool_value_test.go

@@ -4,6 +4,7 @@
 package statecheck_test
 
 import (
+	"fmt"
 	"regexp"
 	"testing"
 
@@ -42,6 +43,13 @@ func TestExtractBoolValue_Basic(t *testing.T) {
 			},
 		},
 	})
+
+	t.Run("CheckTargetVar", func(t *testing.T) {
+		// Now use the testAccAssertBoolEquals function to verify the value of targetVar
+		if err := testAccAssertBoolEquals(true, targetVar); err != nil {
+			t.Errorf("Error in testAccAssertBoolEquals: %v", err)
+		}
+	})
 }
 
 func TestExtractBoolValue_KnownValueWrongType(t *testing.T) {
@@ -136,3 +144,11 @@ func TestExtractBoolValue_ResourceNotFound(t *testing.T) {
 		},
 	})
 }
+
+// testAccAssertBoolEquals compares the expected and target bool values.
+func testAccAssertBoolEquals(expected bool, targetVar bool) error {
+	if targetVar != expected {
+		return fmt.Errorf("expected targetVar to be %v, got %v", expected, targetVar)
+	}
+	return nil
+}

statecheck/extract_string_value_test.go

@@ -4,6 +4,7 @@
 package statecheck_test
 
 import (
+	"fmt"
 	"testing"
 
 	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
@@ -41,4 +42,19 @@ func TestExtractStringValue_Basic(t *testing.T) {
 			},
 		},
 	})
+
+	t.Run("CheckTargetVar", func(t *testing.T) {
+		// Now use the testAccAssertBoolEquals function to verify the value of targetVar
+		if err := testAccAssertStringEquals("str", targetVar); err != nil {
+			t.Errorf("Error in testAccAssertBoolEquals: %v", err)
+		}
+	})
+}
+
+// testAccAssertStringEquals compares the expected and target string values.
+func testAccAssertStringEquals(expected string, targetVar string) error {
+	if targetVar != expected {
+		return fmt.Errorf("expected targetVar to be %v, got %v", expected, targetVar)
+	}
+	return nil
 }