testing

Author: Laure-di

internal/services/inference/model_data_source.go

@@ -15,7 +15,6 @@ import (
 func DataSourceModel() *schema.Resource {
 	dsSchema := datasource.SchemaFromResourceSchema(ResourceModel().Schema)
 
-	//datasource.FixDatasourceSchemaFlags(dsSchema, true, "name")
 	datasource.AddOptionalFieldsToSchema(dsSchema, "url", "name")
 	dsSchema["name"].ConflictsWith = []string{"model_id"}
 	dsSchema["model_id"] = &schema.Schema{
@@ -39,12 +38,13 @@ func DataSourceModelRead(ctx context.Context, d *schema.ResourceData, m interfac
 	}
 
 	modelID, ok := d.GetOk("model_id")
+	pageSize := uint32(1000)
 	if !ok {
 		modelName := d.Get("name").(string)
 		modelList, err := api.ListModels(&inference.ListModelsRequest{
 			Region:    region,
-			Name:      scw.StringPtr(modelName),
 			ProjectID: types.ExpandStringPtr(d.Get("project_id")),
+			PageSize:  &pageSize,
 		}, scw.WithContext(ctx))
 		if err != nil {
 			return diag.FromErr(err)

internal/services/inference/model_data_source_test.go

@@ -1 +1,32 @@
-package inference
+package inference_test
+
+import (
+	"fmt"
+	"github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource"
+	"github.com/scaleway/terraform-provider-scaleway/v2/internal/acctest"
+	"testing"
+)
+
+func TestAccDataSourceModel_Basic(t *testing.T) {
+	tt := acctest.NewTestTools(t)
+	defer tt.Cleanup()
+
+	modelName := "mistral/pixtral-12b-2409:bf16"
+	resource.ParallelTest(t, resource.TestCase{
+		PreCheck:          func() { acctest.PreCheck(t) },
+		ProviderFactories: tt.ProviderFactories,
+		Steps: []resource.TestStep{
+			{
+				Config: fmt.Sprintf(`
+				data "scaleway_inference_model" "my-model" {
+					name = "%s"
+				}
+
+`, modelName),
+				Check: resource.ComposeTestCheckFunc(
+					resource.TestCheckResourceAttr("data.scaleway_inference_model.my-model", "name", modelName),
+				),
+			},
+		},
+	})
+}