poc: working base test for showing how to create a pgtestdb connection that supports pgxpool injection

Author: superlinkx

Diff for: packages/go/bhlog/go.mod

@@ -17,3 +17,5 @@
 module github.com/specterops/bloodhound/bhlog
 
 go 1.23
+
+require github.com/peterldowns/pgtestdb v0.1.1

Diff for: packages/go/bhlog/go.sum

@@ -0,0 +1,2 @@
+github.com/peterldowns/pgtestdb v0.1.1 h1:+hBCD1DcbKeg5Sfg0G+5WNIy/Cm0ORgwMkF4ygihrmU=
+github.com/peterldowns/pgtestdb v0.1.1/go.mod h1:yVWInWV0dxvmLdL2ao3nXDzWZ9+G6EhJ4gRwvI1Ozeg=

Diff for: packages/go/bhlog/random_test.go

@@ -0,0 +1,43 @@
+package bhlog_test
+
+import (
+	"context"
+	"testing"
+
+	_ "github.com/jackc/pgx/v5"
+	_ "github.com/jackc/pgx/v5/stdlib"
+	"github.com/peterldowns/pgtestdb"
+	"github.com/specterops/bloodhound/dawgs"
+	"github.com/specterops/bloodhound/dawgs/drivers/pg"
+	"github.com/specterops/bloodhound/graphschema"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestRandom(t *testing.T) {
+	var ctx = context.Background()
+
+	conf := pgtestdb.Config{
+		DriverName: "pgx",
+		User:       "bhe",
+		Password:   "weneedbetterpasswords",
+		Host:       "localhost",
+		Port:       "55432",
+		Options:    "sslmode=disable",
+	}
+
+	connConf := pgtestdb.Custom(t, conf, pgtestdb.NoopMigrator{})
+
+	pool, err := pg.NewPool(connConf.URL())
+	require.Nil(t, err)
+
+	graphDB, err := dawgs.Open(ctx, pg.DriverName, dawgs.Config{
+		GraphQueryMemoryLimit: 1024 * 1024 * 1024 * 2,
+		ConnectionString:      connConf.URL(),
+		Pool:                  pool,
+	})
+	require.Nil(t, err)
+
+	err = graphDB.AssertSchema(ctx, graphschema.DefaultGraphSchema())
+	assert.Nil(t, err)
+}