synced/coder
1
0
Fork 0
mirror of https://github.com/coder/coder.git synced 2025-07-13 21:36:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: Add more dbgen functions (#6005)

Browse Source
This commit is contained in:
Steven Masley
2023-02-02 17:21:29 -06:00
committed by GitHub
parent 5fe4819669
commit 41e52310bf
3 changed files with 119 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
coderd/database/dbgen/generator_test.go
View File

@ -14,6 +14,14 @@ import (
func TestGenerator(t *testing.T) {
t.Parallel()
t.Run("AuditLog", func(t *testing.T) {
t.Parallel()
db := databasefake.New()
_ = dbgen.AuditLog(t, db, database.AuditLog{})
logs := must(db.GetAuditLogsOffset(context.Background(), database.GetAuditLogsOffsetParams{Limit: 1}))
require.Len(t, logs, 1)
})
t.Run("APIKey", func(t *testing.T) {
t.Parallel()
db := databasefake.New()
@ -56,6 +64,17 @@ func TestGenerator(t *testing.T) {
require.Equal(t, exp, must(db.GetGroupByID(context.Background(), exp.ID)))
})
t.Run("GroupMember", func(t *testing.T) {
t.Parallel()
db := databasefake.New()
g := dbgen.Group(t, db, database.Group{})
u := dbgen.User(t, db, database.User{})
exp := []database.User{u}
dbgen.GroupMember(t, db, database.GroupMember{GroupID: g.ID, UserID: u.ID})
require.Equal(t, exp, must(db.GetGroupMembers(context.Background(), g.ID)))
})
t.Run("Organization", func(t *testing.T) {
t.Parallel()
db := databasefake.New()
@ -70,6 +89,13 @@ func TestGenerator(t *testing.T) {
require.Equal(t, exp, must(db.GetWorkspaceByID(context.Background(), exp.ID)))
})
t.Run("WorkspaceAgent", func(t *testing.T) {
t.Parallel()
db := databasefake.New()
exp := dbgen.WorkspaceAgent(t, db, database.WorkspaceAgent{})
require.Equal(t, exp, must(db.GetWorkspaceAgentByID(context.Background(), exp.ID)))
})
t.Run("Template", func(t *testing.T) {
t.Parallel()
db := databasefake.New()