attempt to add test

Kira-Pilot · Kira-Pilot · commit c3f04f5be91c · 2022-11-01T19:36:49.000Z
diff --git a/coderd/audit.go b/coderd/audit.go
@@ -145,10 +145,13 @@ func (api *API) generateFakeAuditLog(rw http.ResponseWriter, r *http.Request) {
 	if params.ResourceID == uuid.Nil {
 		params.ResourceID = uuid.New()
 	}
+	if params.Time.IsZero() {
+		params.Time = time.Now()
+	}
 
 	_, err = api.Database.InsertAuditLog(ctx, database.InsertAuditLogParams{
 		ID:               uuid.New(),
-		Time:             time.Now(),
+		Time:             params.Time,
 		UserID:           user.ID,
 		Ip:               ipNet,
 		UserAgent:        r.UserAgent(),
diff --git a/coderd/audit_test.go b/coderd/audit_test.go
@@ -3,6 +3,7 @@ package coderd_test
 import (
 	"context"
 	"testing"
+	"time"
 
 	"github.com/google/uuid"
 	"github.com/stretchr/testify/require"
@@ -54,12 +55,14 @@ func TestAuditLogsFilter(t *testing.T) {
 		err := client.CreateTestAuditLog(ctx, codersdk.CreateTestAuditLogRequest{
 			Action:       codersdk.AuditActionCreate,
 			ResourceType: codersdk.ResourceTypeTemplate,
+			Time:         time.Date(2022, 8, 15, 14, 30, 45, 100, time.UTC), // 2022-8-15 14:30:45
 		})
 		require.NoError(t, err)
 		err = client.CreateTestAuditLog(ctx, codersdk.CreateTestAuditLogRequest{
 			Action:       codersdk.AuditActionCreate,
 			ResourceType: codersdk.ResourceTypeUser,
 			ResourceID:   userResourceID,
+			Time:         time.Date(2022, 8, 16, 14, 30, 45, 100, time.UTC), // 2022-8-16 14:30:45
 		})
 		require.NoError(t, err)
 
@@ -127,6 +130,11 @@ func TestAuditLogsFilter(t *testing.T) {
 				SearchQuery:    "action:invalid",
 				ExpectedResult: 3,
 			},
+			{
+				Name:           "FilterWithDateFrom",
+				SearchQuery:    "action:create date_from:2022-08-15 date_to:2022-08-15",
+				ExpectedResult: 1,
+			},
 		}
 
 		for _, testCase := range testCases {
diff --git a/codersdk/audit.go b/codersdk/audit.go
@@ -127,6 +127,7 @@ type CreateTestAuditLogRequest struct {
 	Action       AuditAction  `json:"action,omitempty"`
 	ResourceType ResourceType `json:"resource_type,omitempty"`
 	ResourceID   uuid.UUID    `json:"resource_id,omitempty"`
+	Time         time.Time    `json:"time,omitempty"`
 }
 
 // AuditLogs retrieves audit logs from the given page.

Original file line number	Diff line number	Diff line change
`@@ -127,6 +127,7 @@ type CreateTestAuditLogRequest struct {`
`127`	`127`	Action AuditAction `json:"action,omitempty"`
`128`	`128`	ResourceType ResourceType `json:"resource_type,omitempty"`
`129`	`129`	ResourceID uuid.UUID `json:"resource_id,omitempty"`
	`130`	+ Time time.Time `json:"time,omitempty"`
`130`	`131`	`}`
`131`	`132`
`132`	`133`	`// AuditLogs retrieves audit logs from the given page.`