Skip to content

feat: Add database data generator to make fakedbs easier to populate #5922

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 21 commits into from
Jan 31, 2023
Merged

feat: Add database data generator to make fakedbs easier to populate #5922

Changes from 1 commit
Commits
Show all changes
21 commits
Select commit Hold shift + click to select a range
50ea251
feat: Add database data generator to make fakedbs easier to populate
Emyrk Jan 30, 2023
722194b
Add resources and jobS
Emyrk Jan 30, 2023
38eb88e
Support api keys in generator
Emyrk Jan 30, 2023
d4fe6b7
Begin refactoring api key tests with generator
Emyrk Jan 30, 2023
d423ba2
Refactor api tests to use fake generator
Emyrk Jan 30, 2023
fac054a
Refactor workspace apps test
Emyrk Jan 30, 2023
387c1bd
Remove unused function
Emyrk Jan 30, 2023
50cba39
Refactor provsionerdserver tests
Emyrk Jan 30, 2023
396aa4a
Refactor prom metrics test
Emyrk Jan 30, 2023
5648499
Add unit tests for generator
Emyrk Jan 30, 2023
298ca3e
Remove takeFirstTime
Emyrk Jan 31, 2023
3bfd4d6
Add more randomness to names
Emyrk Jan 31, 2023
7fc8ead
Use fatal over panic
Emyrk Jan 31, 2023
5ba3463
Remove use of panics
Emyrk Jan 31, 2023
bc88e41
Remove all state from generator
Emyrk Jan 31, 2023
1113d8f
Move generate functions and no more generics
Emyrk Jan 31, 2023
5cd9acd
Move to database gen package
Emyrk Jan 31, 2023
e7d2b5f
Fix all refactored tests to use new methods
Emyrk Jan 31, 2023
5e12b78
Remove unused ctx
Emyrk Jan 31, 2023
144ef40
Add comment about unused fields
Emyrk Jan 31, 2023
61ca1ca
rename to dbgen
Emyrk Jan 31, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Refactor workspace apps test
  • Loading branch information
@Emyrk
Emyrk committed Jan 30, 2023
commit fac054aa9e6649828ee386dbb0cba94f8a0004d2
40 changes: 17 additions & 23 deletions coderd/workspaceapps_internal_test.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -16,36 +16,22 @@ import (
func TestAPIKeyEncryption(t *testing.T) {
t.Parallel()

generateAPIKey := func(t *testing.T, db database.Store) (keyID, keySecret string, hashedSecret []byte, data encryptedAPIKeyPayload) {
keyID, keySecret, err := GenerateAPIKeyIDSecret()
require.NoError(t, err)
generateAPIKey := func(t *testing.T, db database.Store) (keyID, keyToken string, hashedSecret []byte, data encryptedAPIKeyPayload) {
gen := databasefake.NewGenerator(t, db)
key, token := gen.APIKey(context.Background(), database.APIKey{})

hashedSecretArray := sha256.Sum256([]byte(keySecret))
data = encryptedAPIKeyPayload{
APIKey: keyID + "-" + keySecret,
APIKey: token,
ExpiresAt: database.Now().Add(24 * time.Hour),
}

return keyID, keySecret, hashedSecretArray[:], data
}
insertAPIKey := func(t *testing.T, db database.Store, keyID string, hashedSecret []byte) {
ctx, cancel := context.WithTimeout(context.Background(), testutil.WaitLong)
defer cancel()

_, err := db.InsertAPIKey(ctx, database.InsertAPIKeyParams{
ID: keyID,
HashedSecret: hashedSecret,
LoginType: database.LoginTypePassword,
Scope: database.APIKeyScopeAll,
})
require.NoError(t, err)
return key.ID, token, key.HashedSecret[:], data
}

t.Run("OK", func(t *testing.T) {
t.Parallel()
db := databasefake.New()
keyID, _, hashedSecret, data := generateAPIKey(t, db)
insertAPIKey(t, db, keyID, hashedSecret)

encrypted, err := encryptAPIKey(data)
require.NoError(t, err)
Expand All @@ -66,8 +52,7 @@ func TestAPIKeyEncryption(t *testing.T) {
t.Run("Expiry", func(t *testing.T) {
t.Parallel()
db := databasefake.New()
keyID, _, hashedSecret, data := generateAPIKey(t, db)
insertAPIKey(t, db, keyID, hashedSecret)
_, _, _, data := generateAPIKey(t, db)

data.ExpiresAt = database.Now().Add(-1 * time.Hour)
encrypted, err := encryptAPIKey(data)
Expand All @@ -84,9 +69,18 @@ func TestAPIKeyEncryption(t *testing.T) {
t.Run("KeyMatches", func(t *testing.T) {
t.Parallel()
db := databasefake.New()
keyID, _, _, data := generateAPIKey(t, db)

gen := databasefake.NewGenerator(t, db)
hashedSecret := sha256.Sum256([]byte("wrong"))
insertAPIKey(t, db, keyID, hashedSecret[:])
// Insert a token with a mismatched hashed secret.
_, token := gen.APIKey(context.Background(), database.APIKey{
HashedSecret: hashedSecret[:],
})

data := encryptedAPIKeyPayload{
APIKey: token,
ExpiresAt: database.Now().Add(24 * time.Hour),
}

encrypted, err := encryptAPIKey(data)
require.NoError(t, err)
Expand Down