more green

Author: bbasata

statecheck/expect_known_value_test.go

@@ -1338,10 +1338,8 @@ func TestExpectKnownValue_CheckState_String(t *testing.T) {
 	t.Parallel()
 
 	r.Test(t, r.TestCase{
-		ProviderFactories: map[string]func() (*schema.Provider, error){
-			"test": func() (*schema.Provider, error) { //nolint:unparam // required signature
-				return testProvider(), nil
-			},
+		ProtoV6ProviderFactories: map[string]func() (tfprotov6.ProviderServer, error){
+			"test": providerserver.NewProviderServer(anTestProvider),
 		},
 		Steps: []r.TestStep{
 			{
@@ -1364,10 +1362,8 @@ func TestExpectKnownValue_CheckState_String_Custom(t *testing.T) {
 	t.Parallel()
 
 	r.Test(t, r.TestCase{
-		ProviderFactories: map[string]func() (*schema.Provider, error){
-			"test": func() (*schema.Provider, error) { //nolint:unparam // required signature
-				return testProvider(), nil
-			},
+		ProtoV6ProviderFactories: map[string]func() (tfprotov6.ProviderServer, error){
+			"test": providerserver.NewProviderServer(anTestProvider),
 		},
 		Steps: []r.TestStep{
 			{
@@ -1420,10 +1416,8 @@ func TestExpectKnownValue_CheckState_String_KnownValueWrongType(t *testing.T) {
 	t.Parallel()
 
 	r.Test(t, r.TestCase{
-		ProviderFactories: map[string]func() (*schema.Provider, error){
-			"test": func() (*schema.Provider, error) { //nolint:unparam // required signature
-				return testProvider(), nil
-			},
+		ProtoV6ProviderFactories: map[string]func() (tfprotov6.ProviderServer, error){
+			"test": providerserver.NewProviderServer(anTestProvider),
 		},
 		Steps: []r.TestStep{
 			{
@@ -1447,10 +1441,8 @@ func TestExpectKnownValue_CheckState_String_KnownValueWrongValue(t *testing.T) {
 	t.Parallel()
 
 	r.Test(t, r.TestCase{
-		ProviderFactories: map[string]func() (*schema.Provider, error){
-			"test": func() (*schema.Provider, error) { //nolint:unparam // required signature
-				return testProvider(), nil
-			},
+		ProtoV6ProviderFactories: map[string]func() (tfprotov6.ProviderServer, error){
+			"test": providerserver.NewProviderServer(anTestProvider),
 		},
 		Steps: []r.TestStep{
 			{

statecheck/test_provider_test.go

@@ -32,6 +32,16 @@ var anTestProvider = testprovider.Provider{
 								Type:     tftypes.Number,
 								Optional: true,
 							},
+							{
+								Name:     "string_attribute",
+								Type:     tftypes.String,
+								Optional: true,
+							},
+							{
+								Name:     "string_computed_attribute",
+								Type:     tftypes.String,
+								Computed: true,
+							},
 						},
 					},
 				},