| 2468 | } |
| 2469 | |
| 2470 | func TestGetActiveUserCount(t *testing.T) { |
| 2471 | t.Parallel() |
| 2472 | if testing.Short() { |
| 2473 | t.SkipNow() |
| 2474 | } |
| 2475 | |
| 2476 | db, _ := dbtestutil.NewDB(t) |
| 2477 | ctx := testutil.Context(t, testutil.WaitLong) |
| 2478 | |
| 2479 | // Seed users: 2 active humans, 1 active service account, |
| 2480 | // 1 dormant, 1 deleted. Only the 2 active humans should |
| 2481 | // be counted for license seat purposes. |
| 2482 | _ = dbgen.User(t, db, database.User{ |
| 2483 | Status: database.UserStatusActive, |
| 2484 | }) |
| 2485 | _ = dbgen.User(t, db, database.User{ |
| 2486 | Status: database.UserStatusActive, |
| 2487 | }) |
| 2488 | _ = dbgen.User(t, db, database.User{ |
| 2489 | Status: database.UserStatusActive, |
| 2490 | IsServiceAccount: true, |
| 2491 | }) |
| 2492 | _ = dbgen.User(t, db, database.User{ |
| 2493 | Status: database.UserStatusDormant, |
| 2494 | }) |
| 2495 | _ = dbgen.User(t, db, database.User{ |
| 2496 | Status: database.UserStatusActive, |
| 2497 | Deleted: true, |
| 2498 | }) |
| 2499 | |
| 2500 | count, err := db.GetActiveUserCount(ctx, false) |
| 2501 | require.NoError(t, err) |
| 2502 | require.Equal(t, int64(2), count) |
| 2503 | } |
| 2504 | |
| 2505 | func TestUserChangeLoginType(t *testing.T) { |
| 2506 | t.Parallel() |