| 832 | } |
| 833 | |
| 834 | func WorkspaceAgentLog(t testing.TB, db database.Store, orig database.WorkspaceAgentLog) database.WorkspaceAgentLog { |
| 835 | log, err := db.InsertWorkspaceAgentLogs(genCtx, database.InsertWorkspaceAgentLogsParams{ |
| 836 | AgentID: takeFirst(orig.AgentID, uuid.New()), |
| 837 | CreatedAt: takeFirst(orig.CreatedAt, dbtime.Now()), |
| 838 | LogSourceID: takeFirst(orig.LogSourceID, uuid.New()), |
| 839 | OutputLength: int32(len(orig.Output)), // nolint: gosec // integer overflow is not a concern here |
| 840 | Level: []database.LogLevel{takeFirst(orig.Level, database.LogLevelInfo)}, |
| 841 | Output: []string{takeFirst(orig.Output, "Test agent log")}, |
| 842 | }) |
| 843 | require.NoError(t, err, "insert workspace agent log") |
| 844 | require.Len(t, log, 1, "incorrect number of agent logs returned") |
| 845 | return log[0] |
| 846 | } |
| 847 | |
| 848 | func WorkspaceBuild(t testing.TB, db database.Store, orig database.WorkspaceBuild) database.WorkspaceBuild { |
| 849 | t.Helper() |