2022-05-09 22:05:01 +00:00
|
|
|
package backends_test
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"testing"
|
|
|
|
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
|
2023-08-18 18:55:43 +00:00
|
|
|
"github.com/coder/coder/v2/coderd/database"
|
2023-10-30 17:42:20 +00:00
|
|
|
"github.com/coder/coder/v2/coderd/database/dbmem"
|
2023-08-31 18:32:28 +00:00
|
|
|
"github.com/coder/coder/v2/enterprise/audit"
|
2023-08-18 18:55:43 +00:00
|
|
|
"github.com/coder/coder/v2/enterprise/audit/audittest"
|
|
|
|
"github.com/coder/coder/v2/enterprise/audit/backends"
|
2022-05-09 22:05:01 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
func TestPostgresBackend(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
t.Run("OK", func(t *testing.T) {
|
|
|
|
t.Parallel()
|
|
|
|
|
|
|
|
var (
|
|
|
|
ctx, cancel = context.WithCancel(context.Background())
|
2023-10-30 17:42:20 +00:00
|
|
|
db = dbmem.New()
|
2022-05-09 22:05:01 +00:00
|
|
|
pgb = backends.NewPostgres(db, true)
|
2022-09-02 16:42:28 +00:00
|
|
|
alog = audittest.RandomLog()
|
2022-05-09 22:05:01 +00:00
|
|
|
)
|
|
|
|
defer cancel()
|
|
|
|
|
2023-08-31 18:32:28 +00:00
|
|
|
err := pgb.Export(ctx, alog, audit.BackendDetails{})
|
2022-05-09 22:05:01 +00:00
|
|
|
require.NoError(t, err)
|
|
|
|
|
2022-09-07 16:38:19 +00:00
|
|
|
got, err := db.GetAuditLogsOffset(ctx, database.GetAuditLogsOffsetParams{
|
|
|
|
Offset: 0,
|
|
|
|
Limit: 1,
|
2022-05-09 22:05:01 +00:00
|
|
|
})
|
|
|
|
require.NoError(t, err)
|
|
|
|
require.Len(t, got, 1)
|
2022-09-07 16:38:19 +00:00
|
|
|
require.Equal(t, alog.ID, got[0].ID)
|
2022-05-09 22:05:01 +00:00
|
|
|
})
|
|
|
|
}
|