spanner: fix TestIntegration_DashboardPermissionFilter (#102603)

pull/101479/head
Georges Chaudy 4 months ago committed by GitHub
parent 57889a43a7
commit 19d1ded3bf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
  1. 34
      pkg/services/sqlstore/permissions/dashboard_test.go

@ -760,9 +760,21 @@ func setupTest(t *testing.T, numFolders, numDashboards int, permissions []access
})
}
_, err := sess.InsertMulti(&dashes)
if err != nil {
return err
// Insert dashboards in batches
batchSize := 500
if db.IsTestDBSpanner() {
batchSize = 30 // spanner has a limit of 950 parameters per query
}
for i := 0; i < len(dashes); i += batchSize {
end := i + batchSize
if end > len(dashes) {
end = len(dashes)
}
_, err := sess.InsertMulti(dashes[i:end])
if err != nil {
return err
}
}
role := &accesscontrol.Role{
@ -772,7 +784,7 @@ func setupTest(t *testing.T, numFolders, numDashboards int, permissions []access
Updated: time.Now(),
Created: time.Now(),
}
_, err = sess.Insert(role)
_, err := sess.Insert(role)
if err != nil {
return err
}
@ -794,10 +806,18 @@ func setupTest(t *testing.T, numFolders, numDashboards int, permissions []access
permissions[i].Updated = time.Now()
permissions[i].Kind, permissions[i].Attribute, permissions[i].Identifier = permissions[i].SplitScope()
}
if len(permissions) > 0 {
_, err = sess.InsertMulti(&permissions)
if err != nil {
return err
for i := 0; i < len(permissions); i += batchSize {
end := i + batchSize
if end > len(permissions) {
end = len(permissions)
}
_, err = sess.InsertMulti(permissions[i:end])
if err != nil {
return err
}
}
}

Loading…
Cancel
Save