(t *testing.T, db database.DB)
| 163 | } |
| 164 | |
| 165 | func testOrgsWithPagination(t *testing.T, db database.DB) { |
| 166 | ctx := context.Background() |
| 167 | |
| 168 | user, err := db.InsertUser(ctx, &database.InsertUserOptions{Email: "test@rilldata.com"}) |
| 169 | require.NoError(t, err) |
| 170 | require.Equal(t, "test@rilldata.com", user.Email) |
| 171 | |
| 172 | role, err := db.FindOrganizationRole(ctx, database.OrganizationRoleNameAdmin) |
| 173 | require.NoError(t, err) |
| 174 | |
| 175 | // add org and give user permission |
| 176 | org, err := db.InsertOrganization(ctx, &database.InsertOrganizationOptions{Name: "alpha"}) |
| 177 | require.NoError(t, err) |
| 178 | require.Equal(t, "alpha", org.Name) |
| 179 | _, err = db.InsertOrganizationMemberUser(ctx, org.ID, user.ID, role.ID, nil, false) |
| 180 | require.NoError(t, err) |
| 181 | |
| 182 | // add org and give user permission |
| 183 | org, err = db.InsertOrganization(ctx, &database.InsertOrganizationOptions{Name: "beta"}) |
| 184 | require.NoError(t, err) |
| 185 | require.Equal(t, "beta", org.Name) |
| 186 | _, err = db.InsertOrganizationMemberUser(ctx, org.ID, user.ID, role.ID, nil, false) |
| 187 | require.NoError(t, err) |
| 188 | |
| 189 | // add org only |
| 190 | org, err = db.InsertOrganization(ctx, &database.InsertOrganizationOptions{Name: "gamma"}) |
| 191 | require.NoError(t, err) |
| 192 | require.Equal(t, "gamma", org.Name) |
| 193 | |
| 194 | // fetch org without name filter |
| 195 | orgs, err := db.FindOrganizationsForUser(ctx, user.ID, "", 1) |
| 196 | require.NoError(t, err) |
| 197 | require.Equal(t, len(orgs), 1) |
| 198 | require.Equal(t, "alpha", orgs[0].Name) |
| 199 | |
| 200 | // fetch org with name filter |
| 201 | orgs, err = db.FindOrganizationsForUser(ctx, user.ID, orgs[0].Name, 10) |
| 202 | require.NoError(t, err) |
| 203 | require.Equal(t, len(orgs), 1) |
| 204 | require.Equal(t, "beta", orgs[0].Name) |
| 205 | |
| 206 | //cleanup |
| 207 | require.NoError(t, db.DeleteOrganization(ctx, "alpha")) |
| 208 | require.NoError(t, db.DeleteOrganization(ctx, "beta")) |
| 209 | require.NoError(t, db.DeleteOrganization(ctx, "gamma")) |
| 210 | require.NoError(t, db.DeleteUser(ctx, user.ID)) |
| 211 | } |
| 212 | |
| 213 | func testProjects(t *testing.T, db database.DB) { |
| 214 | ctx := context.Background() |
no test coverage detected