| 77 | } |
| 78 | |
| 79 | func TestQDev_TaskDataWithAccountId(t *testing.T) { |
| 80 | // Test new-style scope with AccountId and multiple S3Prefixes |
| 81 | month := 1 |
| 82 | taskData := &tasks.QDevTaskData{ |
| 83 | Options: &tasks.QDevOptions{ |
| 84 | ConnectionId: 1, |
| 85 | AccountId: "034362076319", |
| 86 | BasePath: "user-report", |
| 87 | Year: 2026, |
| 88 | Month: &month, |
| 89 | }, |
| 90 | S3Client: &tasks.QDevS3Client{ |
| 91 | Bucket: "test-bucket", |
| 92 | }, |
| 93 | S3Prefixes: []string{ |
| 94 | "user-report/AWSLogs/034362076319/KiroLogs/by_user_analytic/us-east-1/2026/01", |
| 95 | "user-report/AWSLogs/034362076319/KiroLogs/user_report/us-east-1/2026/01", |
| 96 | }, |
| 97 | } |
| 98 | |
| 99 | assert.Equal(t, "034362076319", taskData.Options.AccountId) |
| 100 | assert.Equal(t, "user-report", taskData.Options.BasePath) |
| 101 | assert.Equal(t, 2026, taskData.Options.Year) |
| 102 | assert.Equal(t, &month, taskData.Options.Month) |
| 103 | assert.Len(t, taskData.S3Prefixes, 2) |
| 104 | assert.Contains(t, taskData.S3Prefixes[0], "by_user_analytic") |
| 105 | assert.Contains(t, taskData.S3Prefixes[1], "user_report") |
| 106 | } |
| 107 | |
| 108 | func TestQDev_TaskDataWithoutIdentityClient(t *testing.T) { |
| 109 | // Test that QDevTaskData works without IdentityClient |