| 1759 | } |
| 1760 | |
| 1761 | func WorkspaceAgentMemoryResourceMonitor(t testing.TB, db database.Store, seed database.WorkspaceAgentMemoryResourceMonitor) database.WorkspaceAgentMemoryResourceMonitor { |
| 1762 | monitor, err := db.InsertMemoryResourceMonitor(genCtx, database.InsertMemoryResourceMonitorParams{ |
| 1763 | AgentID: takeFirst(seed.AgentID, uuid.New()), |
| 1764 | Enabled: takeFirst(seed.Enabled, true), |
| 1765 | State: takeFirst(seed.State, database.WorkspaceAgentMonitorStateOK), |
| 1766 | Threshold: takeFirst(seed.Threshold, 100), |
| 1767 | CreatedAt: takeFirst(seed.CreatedAt, dbtime.Now()), |
| 1768 | UpdatedAt: takeFirst(seed.UpdatedAt, dbtime.Now()), |
| 1769 | DebouncedUntil: takeFirst(seed.DebouncedUntil, time.Time{}), |
| 1770 | }) |
| 1771 | require.NoError(t, err, "insert workspace agent memory resource monitor") |
| 1772 | return monitor |
| 1773 | } |
| 1774 | |
| 1775 | func WorkspaceAgentVolumeResourceMonitor(t testing.TB, db database.Store, seed database.WorkspaceAgentVolumeResourceMonitor) database.WorkspaceAgentVolumeResourceMonitor { |
| 1776 | monitor, err := db.InsertVolumeResourceMonitor(genCtx, database.InsertVolumeResourceMonitorParams{ |