|
4 | 4 | "regexp" |
5 | 5 | "testing" |
6 | 6 |
|
| 7 | +"github.com/google/uuid" |
7 | 8 | "github.com/hashicorp/terraform-plugin-sdk/v2/helper/resource" |
8 | 9 | "github.com/hashicorp/terraform-plugin-sdk/v2/terraform" |
9 | 10 | "github.com/stretchr/testify/require" |
@@ -214,3 +215,83 @@ func TestAITask(t *testing.T) { |
214 | 215 | }) |
215 | 216 | }) |
216 | 217 | } |
| 218 | + |
| 219 | +funcTestTaskDatasource(t*testing.T) { |
| 220 | +t.Run("Exists",func(t*testing.T) { |
| 221 | +t.Setenv("CODER_TASK_ID","7d8d4c2e-fb57-44f9-a183-22509819c2e7") |
| 222 | +t.Setenv("CODER_TASK_PROMPT","some task prompt") |
| 223 | +resource.Test(t, resource.TestCase{ |
| 224 | +ProviderFactories:coderFactory(), |
| 225 | +IsUnitTest:true, |
| 226 | +Steps: []resource.TestStep{{ |
| 227 | +Config:` |
| 228 | +provider "coder" {} |
| 229 | +data "coder_task" "me" {} |
| 230 | +`, |
| 231 | +Check:func(s*terraform.State)error { |
| 232 | +require.Len(t,s.Modules,1) |
| 233 | +require.Len(t,s.Modules[0].Resources,1) |
| 234 | +resource:=s.Modules[0].Resources["data.coder_task.me"] |
| 235 | +require.NotNil(t,resource) |
| 236 | + |
| 237 | +taskID:=resource.Primary.Attributes["id"] |
| 238 | +require.Equal(t,"7d8d4c2e-fb57-44f9-a183-22509819c2e7",taskID) |
| 239 | + |
| 240 | +taskPromptValue:=resource.Primary.Attributes["prompt"] |
| 241 | +require.Equal(t,"some task prompt",taskPromptValue) |
| 242 | + |
| 243 | +enabledValue:=resource.Primary.Attributes["enabled"] |
| 244 | +require.Equal(t,"true",enabledValue) |
| 245 | +returnnil |
| 246 | +}, |
| 247 | +}}, |
| 248 | +}) |
| 249 | +}) |
| 250 | + |
| 251 | +t.Run("NotExists",func(t*testing.T) { |
| 252 | +resource.Test(t, resource.TestCase{ |
| 253 | +ProviderFactories:coderFactory(), |
| 254 | +IsUnitTest:true, |
| 255 | +Steps: []resource.TestStep{{ |
| 256 | +Config:` |
| 257 | +provider "coder" {} |
| 258 | +data "coder_task" "me" {} |
| 259 | +`, |
| 260 | +Check:func(s*terraform.State)error { |
| 261 | +require.Len(t,s.Modules,1) |
| 262 | +require.Len(t,s.Modules[0].Resources,1) |
| 263 | +resource:=s.Modules[0].Resources["data.coder_task.me"] |
| 264 | +require.NotNil(t,resource) |
| 265 | + |
| 266 | +taskID:=resource.Primary.Attributes["id"] |
| 267 | +require.NotEmpty(t,taskID) |
| 268 | +require.NotEqual(t,uuid.Nil.String(),taskID) |
| 269 | +_,err:=uuid.Parse(taskID) |
| 270 | +require.NoError(t,err) |
| 271 | + |
| 272 | +taskPromptValue:=resource.Primary.Attributes["prompt"] |
| 273 | +require.Empty(t,taskPromptValue) |
| 274 | + |
| 275 | +enabledValue:=resource.Primary.Attributes["enabled"] |
| 276 | +require.Equal(t,"false",enabledValue) |
| 277 | +returnnil |
| 278 | +}, |
| 279 | +}}, |
| 280 | +}) |
| 281 | +}) |
| 282 | + |
| 283 | +t.Run("InvalidTaskID",func(t*testing.T) { |
| 284 | +t.Setenv("CODER_TASK_ID","not a valid UUID") |
| 285 | +resource.Test(t, resource.TestCase{ |
| 286 | +ProviderFactories:coderFactory(), |
| 287 | +IsUnitTest:true, |
| 288 | +Steps: []resource.TestStep{{ |
| 289 | +Config:` |
| 290 | +provider "coder" {} |
| 291 | +data "coder_task" "me" {} |
| 292 | +`, |
| 293 | +ExpectError:regexp.MustCompile(`invalid UUID`), |
| 294 | +}}, |
| 295 | +}) |
| 296 | +}) |
| 297 | +} |