Add unit tests for generator

Emyrk · Emyrk · commit 564849988a4e · 2023-01-30T15:41:10.000-06:00
diff --git a/coderd/database/databasefake/generator_test.go b/coderd/database/databasefake/generator_test.go
@@ -0,0 +1,98 @@
+package databasefake_test
+
+import (
+	"context"
+	"testing"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/coder/coder/coderd/database"
+	"github.com/coder/coder/coderd/database/databasefake"
+)
+
+func TestGenerator(t *testing.T) {
+	t.Parallel()
+
+	// Reuse the same database for all tests.
+	db := databasefake.New()
+	gen := databasefake.NewGenerator(t, db)
+
+	t.Run("APIKey", func(t *testing.T) {
+		t.Parallel()
+		exp, _ := gen.APIKey(context.Background(), database.APIKey{})
+		require.Equal(t, exp, must(db.GetAPIKeyByID(context.Background(), exp.ID)))
+	})
+
+	t.Run("File", func(t *testing.T) {
+		t.Parallel()
+		exp := gen.File(context.Background(), database.File{})
+		require.Equal(t, exp, must(db.GetFileByID(context.Background(), exp.ID)))
+	})
+
+	t.Run("UserLink", func(t *testing.T) {
+		t.Parallel()
+		exp := gen.UserLink(context.Background(), database.UserLink{})
+		require.Equal(t, exp, must(db.GetUserLinkByLinkedID(context.Background(), exp.LinkedID)))
+	})
+
+	t.Run("WorkspaceResource", func(t *testing.T) {
+		t.Parallel()
+		exp := gen.WorkspaceResource(context.Background(), database.WorkspaceResource{})
+		require.Equal(t, exp, must(db.GetWorkspaceResourceByID(context.Background(), exp.ID)))
+	})
+
+	t.Run("Job", func(t *testing.T) {
+		t.Parallel()
+		exp := gen.Job(context.Background(), database.ProvisionerJob{})
+		require.Equal(t, exp, must(db.GetProvisionerJobByID(context.Background(), exp.ID)))
+	})
+
+	t.Run("Group", func(t *testing.T) {
+		t.Parallel()
+		exp := gen.Group(context.Background(), database.Group{})
+		require.Equal(t, exp, must(db.GetGroupByID(context.Background(), exp.ID)))
+	})
+
+	t.Run("Organization", func(t *testing.T) {
+		t.Parallel()
+		exp := gen.Organization(context.Background(), database.Organization{})
+		require.Equal(t, exp, must(db.GetOrganizationByID(context.Background(), exp.ID)))
+	})
+
+	t.Run("Workspace", func(t *testing.T) {
+		t.Parallel()
+		exp := gen.Workspace(context.Background(), database.Workspace{})
+		require.Equal(t, exp, must(db.GetWorkspaceByID(context.Background(), exp.ID)))
+	})
+
+	t.Run("Template", func(t *testing.T) {
+		t.Parallel()
+		exp := gen.Template(context.Background(), database.Template{})
+		require.Equal(t, exp, must(db.GetTemplateByID(context.Background(), exp.ID)))
+	})
+
+	t.Run("TemplateVersion", func(t *testing.T) {
+		t.Parallel()
+		exp := gen.TemplateVersion(context.Background(), database.TemplateVersion{})
+		require.Equal(t, exp, must(db.GetTemplateVersionByID(context.Background(), exp.ID)))
+	})
+
+	t.Run("WorkspaceBuild", func(t *testing.T) {
+		t.Parallel()
+		exp := gen.WorkspaceBuild(context.Background(), database.WorkspaceBuild{})
+		require.Equal(t, exp, must(db.GetWorkspaceBuildByID(context.Background(), exp.ID)))
+	})
+
+	t.Run("User", func(t *testing.T) {
+		t.Parallel()
+		exp := gen.User(context.Background(), database.User{})
+		require.Equal(t, exp, must(db.GetUserByID(context.Background(), exp.ID)))
+	})
+}
+
+func must[T any](value T, err error) T {
+	if err != nil {
+		panic(err)
+	}
+	return value
+}
