more green

Author: bbasata

statecheck/expect_known_value_test.go

@@ -28,10 +28,8 @@ func TestExpectKnownValue_CheckState_ResourceNotFound(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{
 			{
@@ -407,10 +405,13 @@ func TestExpectKnownValue_CheckState_List(t *testing.T) {
 
 	r.Test(t, r.TestCase{
 		ProviderFactories: map[string]func() (*schema.Provider, error){
-			"test": func() (*schema.Provider, error) { //nolint:unparam // required signature
+			"test7": 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{
 			{
 				Config: `resource "test_resource" "one" {
@@ -440,10 +441,13 @@ func TestExpectKnownValue_CheckState_List_KnownValueWrongType(t *testing.T) {
 
 	r.Test(t, r.TestCase{
 		ProviderFactories: map[string]func() (*schema.Provider, error){
-			"test": func() (*schema.Provider, error) { //nolint:unparam // required signature
+			"test7": 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{
 			{
 				Config: `resource "test_resource" "one" {
@@ -471,10 +475,13 @@ func TestExpectKnownValue_CheckState_List_KnownValueWrongValue(t *testing.T) {
 
 	r.Test(t, r.TestCase{
 		ProviderFactories: map[string]func() (*schema.Provider, error){
-			"test": func() (*schema.Provider, error) { //nolint:unparam // required signature
+			"test7": 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{
 			{
 				Config: `resource "test_resource" "one" {
@@ -504,10 +511,8 @@ func TestExpectKnownValue_CheckState_ListPartial(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{
 			{
@@ -538,10 +543,8 @@ func TestExpectKnownValue_CheckState_ListPartial_KnownValueWrongValue(t *testing
 	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{
 			{
@@ -571,10 +574,8 @@ func TestExpectKnownValue_CheckState_ListElements(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{
 			{
@@ -601,10 +602,8 @@ func TestExpectKnownValue_CheckState_ListElements_WrongNum(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

@@ -42,6 +42,11 @@ var anTestProvider = testprovider.Provider{
 								Type:     tftypes.String,
 								Computed: true,
 							},
+							{
+								Name:     "list_attribute",
+								Type:     tftypes.List{ElementType: tftypes.String},
+								Optional: true,
+							},
 						},
 					},
 				},