Implement fake

Emyrk · Emyrk · commit 02555b14ae12 · 2023-10-09T18:37:52.000-05:00
diff --git a/cli/templateversionarchive.go b/cli/templateversionarchive.go
@@ -70,7 +70,6 @@ func (r *RootCmd) setArchiveTemplateVersion(archive bool) *clibase.Cmd {
 				versions = append(versions, version)
 			}
 
-			failed := 0
 			for _, version := range versions {
 				if version.Archived == archive {
 					_, _ = fmt.Fprintln(
@@ -88,10 +87,6 @@ func (r *RootCmd) setArchiveTemplateVersion(archive bool) *clibase.Cmd {
 					inv.Stdout, fmt.Sprintf("Version "+pretty.Sprint(cliui.DefaultStyles.Keyword, version.Name)+" "+pastVerb+" at "+cliui.Timestamp(time.Now())),
 				)
 			}
-
-			if failed > 0 {
-				return xerrors.Errorf("failed on %d template versions", failed)
-			}
 			return nil
 		},
 	}
@@ -159,7 +154,6 @@ func (r *RootCmd) archiveTemplateVersions() *clibase.Cmd {
 				return err
 			}
 
-			failed := 0
 			for _, template := range templates {
 				resp, err := client.ArchiveTemplateVersions(ctx, template.ID, all.Value())
 				if err != nil {
@@ -180,10 +174,6 @@ func (r *RootCmd) archiveTemplateVersions() *clibase.Cmd {
 					)
 				}
 			}
-
-			if failed > 0 {
-				return xerrors.Errorf("failed on %d templates", failed)
-			}
 			return nil
 		},
 	}
diff --git a/coderd/database/dbfake/dbfake.go b/coderd/database/dbfake/dbfake.go
@@ -865,6 +865,10 @@ func (q *FakeQuerier) ArchiveUnusedTemplateVersions(_ context.Context, arg datab
 			// Not the latest
 			continue
 		}
+		// Ignore deleted workspaces.
+		if b.Transition == database.WorkspaceTransitionDelete {
+			continue
+		}
 		latest[b.WorkspaceID] = latestBuild{
 			Number:  b.BuildNumber,
 			Version: b.TemplateVersionID,
@@ -881,6 +885,11 @@ func (q *FakeQuerier) ArchiveUnusedTemplateVersions(_ context.Context, arg datab
 
 	var archived []uuid.UUID
 	for i, v := range q.templateVersions {
+		if arg.TemplateVersionID != uuid.Nil {
+			if v.ID != arg.TemplateVersionID {
+				continue
+			}
+		}
 		if v.Archived {
 			continue
 		}
@@ -5336,7 +5345,16 @@ func (q *FakeQuerier) UnarchiveTemplateVersion(ctx context.Context, arg database
 		return err
 	}
 
-	panic("not implemented")
+	for i, v := range q.data.templateVersions {
+		if v.ID == arg.TemplateVersionID {
+			v.Archived = false
+			v.UpdatedAt = arg.UpdatedAt
+			q.data.templateVersions[i] = v
+			return nil
+		}
+	}
+
+	return sql.ErrNoRows
 }
 
 func (q *FakeQuerier) UpdateAPIKeyByID(_ context.Context, arg database.UpdateAPIKeyByIDParams) error {
diff --git a/coderd/database/queries/templateversions.sql b/coderd/database/queries/templateversions.sql
@@ -195,7 +195,7 @@ FROM
 		-- the build is outdated.
 		NOT EXISTS (
 			-- Return all "used" versions, where "used" is defined as being
-		    -- used by a latest workspace build.
+			-- used by a latest workspace build.
 			SELECT template_version_id FROM (
 				SELECT
 				    DISTINCT ON (workspace_id) template_version_id, transition

Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,6 @@ func (r RootCmd) setArchiveTemplateVersion(archive bool) clibase.Cmd {`
`70`	`70`	`versions = append(versions, version)`
`71`	`71`	`}`
`72`	`72`
`73`		`- failed := 0`
`74`	`73`	`for _, version := range versions {`
`75`	`74`	`if version.Archived == archive {`
`76`	`75`	`_, _ = fmt.Fprintln(`
`@@ -88,10 +87,6 @@ func (r RootCmd) setArchiveTemplateVersion(archive bool) clibase.Cmd {`
`88`	`87`	`inv.Stdout, fmt.Sprintf("Version "+pretty.Sprint(cliui.DefaultStyles.Keyword, version.Name)+" "+pastVerb+" at "+cliui.Timestamp(time.Now())),`
`89`	`88`	`)`
`90`	`89`	`}`
`91`		`-`
`92`		`- if failed > 0 {`
`93`		`- return xerrors.Errorf("failed on %d template versions", failed)`
`94`		`- }`
`95`	`90`	`return nil`
`96`	`91`	`},`
`97`	`92`	`}`
`@@ -159,7 +154,6 @@ func (r RootCmd) archiveTemplateVersions() clibase.Cmd {`
`159`	`154`	`return err`
`160`	`155`	`}`
`161`	`156`
`162`		`- failed := 0`
`163`	`157`	`for _, template := range templates {`
`164`	`158`	`resp, err := client.ArchiveTemplateVersions(ctx, template.ID, all.Value())`
`165`	`159`	`if err != nil {`
`@@ -180,10 +174,6 @@ func (r RootCmd) archiveTemplateVersions() clibase.Cmd {`
`180`	`174`	`)`
`181`	`175`	`}`
`182`	`176`	`}`
`183`		`-`
`184`		`- if failed > 0 {`
`185`		`- return xerrors.Errorf("failed on %d templates", failed)`
`186`		`- }`
`187`	`177`	`return nil`
`188`	`178`	`},`
`189`	`179`	`}`