(t *testing.T, db database.Store)
| 217 | } |
| 218 | |
| 219 | func genData(t *testing.T, db database.Store) []database.User { |
| 220 | t.Helper() |
| 221 | var users []database.User |
| 222 | // Make some users |
| 223 | for _, status := range database.AllUserStatusValues() { |
| 224 | for _, loginType := range database.AllLoginTypeValues() { |
| 225 | for _, deleted := range []bool{false, true} { |
| 226 | randName := testutil.MustRandString(t, 32) |
| 227 | usr := dbgen.User(t, db, database.User{ |
| 228 | Username: randName, |
| 229 | Email: randName + "@notcoder.com", |
| 230 | LoginType: loginType, |
| 231 | Status: status, |
| 232 | Deleted: deleted, |
| 233 | }) |
| 234 | _ = dbgen.ExternalAuthLink(t, db, database.ExternalAuthLink{ |
| 235 | UserID: usr.ID, |
| 236 | ProviderID: "fake", |
| 237 | OAuthAccessToken: "access-" + usr.ID.String(), |
| 238 | OAuthRefreshToken: "refresh-" + usr.ID.String(), |
| 239 | }) |
| 240 | provider := dbgen.AIProvider(t, db, database.AIProvider{ |
| 241 | Name: "ai-provider-" + usr.ID.String(), |
| 242 | Settings: sql.NullString{String: "settings-" + usr.ID.String(), Valid: true}, |
| 243 | }) |
| 244 | _ = dbgen.AIProviderKey(t, db, database.AIProviderKey{ |
| 245 | ProviderID: provider.ID, |
| 246 | APIKey: "provider-key-" + usr.ID.String(), |
| 247 | }) |
| 248 | now := time.Now() |
| 249 | _, err := db.UpsertUserAIProviderKey(context.Background(), database.UpsertUserAIProviderKeyParams{ |
| 250 | ID: uuid.New(), |
| 251 | UserID: usr.ID, |
| 252 | AIProviderID: provider.ID, |
| 253 | APIKey: "user-ai-provider-key-" + usr.ID.String(), |
| 254 | CreatedAt: now, |
| 255 | UpdatedAt: now, |
| 256 | }) |
| 257 | require.NoError(t, err) |
| 258 | |
| 259 | // Deleted users cannot have user_links or user_secrets. |
| 260 | if !deleted { |
| 261 | // Fun fact: our schema allows _all_ login types to have |
| 262 | // a user_link. Even though I'm not sure how it could occur |
| 263 | // in practice, making sure to test all combinations here. |
| 264 | _ = dbgen.UserLink(t, db, database.UserLink{ |
| 265 | UserID: usr.ID, |
| 266 | LoginType: usr.LoginType, |
| 267 | OAuthAccessToken: "access-" + usr.ID.String(), |
| 268 | OAuthRefreshToken: "refresh-" + usr.ID.String(), |
| 269 | }) |
| 270 | |
| 271 | _ = dbgen.UserSecret(t, db, database.UserSecret{ |
| 272 | UserID: usr.ID, |
| 273 | Name: "secret-" + usr.ID.String(), |
| 274 | Value: "value-" + usr.ID.String(), |
| 275 | EnvName: "", |
| 276 | FilePath: "", |
no test coverage detected