Test audit methods

Emyrk · Emyrk · commit 6191561dfa25 · 2023-02-02T15:49:04.000-06:00
diff --git a/coderd/authzquery/audit_test.go b/coderd/authzquery/audit_test.go
@@ -0,0 +1,32 @@
+package authzquery_test
+
+import (
+	"testing"
+
+	"github.com/coder/coder/coderd/database/dbgen"
+
+	"github.com/coder/coder/coderd/database"
+	"github.com/coder/coder/coderd/rbac"
+)
+
+func (suite *MethodTestSuite) TestAuditLogs() {
+	suite.Run("InsertAuditLog", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			return methodCase(inputs(database.InsertAuditLogParams{
+				ResourceType: database.ResourceTypeOrganization,
+				Action:       database.AuditActionCreate,
+			}),
+				asserts(rbac.ResourceAuditLog, rbac.ActionCreate))
+		})
+	})
+	suite.Run("GetAuditLogsOffset", func() {
+		suite.RunMethodTest(func(t *testing.T, db database.Store) MethodCase {
+			_ = dbgen.AuditLog(t, db, database.AuditLog{})
+			_ = dbgen.AuditLog(t, db, database.AuditLog{})
+			return methodCase(inputs(database.GetAuditLogsOffsetParams{
+				Limit: 10,
+			}),
+				asserts(rbac.ResourceAuditLog, rbac.ActionRead))
+		})
+	})
+}
diff --git a/coderd/database/dbgen/generator.go b/coderd/database/dbgen/generator.go
@@ -6,6 +6,7 @@ import (
 	"database/sql"
 	"encoding/hex"
 	"fmt"
+	"net"
 	"testing"
 	"time"
 
@@ -21,6 +22,34 @@ import (
 // All methods take in a 'seed' object. Any provided fields in the seed will be
 // maintained. Any fields omitted will have sensible defaults generated.
 
+func AuditLog(t *testing.T, db database.Store, seed database.AuditLog) database.AuditLog {
+	log, err := db.InsertAuditLog(context.Background(), database.InsertAuditLogParams{
+		ID:             takeFirst(seed.ID, uuid.New()),
+		Time:           takeFirst(seed.Time, time.Now()),
+		UserID:         takeFirst(seed.UserID, uuid.New()),
+		OrganizationID: takeFirst(seed.OrganizationID, uuid.New()),
+		Ip: pqtype.Inet{
+			IPNet: takeFirstIP(seed.Ip.IPNet, net.IPNet{}),
+			Valid: takeFirst(seed.Ip.Valid, false),
+		},
+		UserAgent: sql.NullString{
+			String: takeFirst(seed.UserAgent.String, ""),
+			Valid:  takeFirst(seed.UserAgent.Valid, false),
+		},
+		ResourceType:     takeFirst(seed.ResourceType, database.ResourceTypeOrganization),
+		ResourceID:       takeFirst(seed.ResourceID, uuid.New()),
+		ResourceTarget:   takeFirst(seed.ResourceTarget, uuid.NewString()),
+		Action:           takeFirst(seed.Action, database.AuditActionCreate),
+		Diff:             takeFirstBytes(seed.Diff, []byte("{}")),
+		StatusCode:       takeFirst(seed.StatusCode, 200),
+		AdditionalFields: takeFirstBytes(seed.Diff, []byte("{}")),
+		RequestID:        takeFirst(seed.RequestID, uuid.New()),
+		ResourceIcon:     takeFirst(seed.ResourceIcon, ""),
+	})
+	require.NoError(t, err, "insert audit log")
+	return log
+}
+
 func Template(t *testing.T, db database.Store, seed database.Template) database.Template {
 	template, err := db.InsertTemplate(context.Background(), database.InsertTemplateParams{
 		ID:                           takeFirst(seed.ID, uuid.New()),
diff --git a/coderd/database/dbgen/generator_test.go b/coderd/database/dbgen/generator_test.go
@@ -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()
diff --git a/coderd/database/dbgen/take.go b/coderd/database/dbgen/take.go
@@ -1,5 +1,13 @@
 package dbgen
 
+import "net"
+
+func takeFirstIP(values ...net.IPNet) net.IPNet {
+	return takeFirstF(values, func(v net.IPNet) bool {
+		return len(v.IP) != 0 && len(v.Mask) != 0
+	})
+}
+
 // takeFirstBytes implements takeFirst for []byte.
 // []byte is not a comparable type.
 func takeFirstBytes(values ...[]byte) []byte {
