From e2e33aba4dc6b21e2f7c5792f67979e536a88806 Mon Sep 17 00:00:00 2001 From: John Hopper Date: Wed, 18 Dec 2024 09:26:17 -0800 Subject: [PATCH] chore: BED-5168 - prepare for review --- cmd/api/src/api/v2/integration/ingest.go | 3 ++- packages/go/cypher/models/pgsql/translate/translator_test.go | 3 ++- .../go/cypher/models/pgsql/visualization/visualizer_test.go | 3 ++- packages/go/dawgs/drivers/pg/node_test.go | 3 ++- packages/go/dawgs/drivers/pg/pgutil/kindmapper.go | 1 + 5 files changed, 9 insertions(+), 4 deletions(-) diff --git a/cmd/api/src/api/v2/integration/ingest.go b/cmd/api/src/api/v2/integration/ingest.go index 240d3a3dc..ec03c20e0 100644 --- a/cmd/api/src/api/v2/integration/ingest.go +++ b/cmd/api/src/api/v2/integration/ingest.go @@ -18,10 +18,11 @@ package integration import ( "encoding/json" - "github.com/specterops/bloodhound/graphschema" "strings" "time" + "github.com/specterops/bloodhound/graphschema" + "github.com/specterops/bloodhound/dawgs/graph" "github.com/specterops/bloodhound/src/model" "github.com/specterops/bloodhound/src/model/appcfg" diff --git a/packages/go/cypher/models/pgsql/translate/translator_test.go b/packages/go/cypher/models/pgsql/translate/translator_test.go index c14053985..724211293 100644 --- a/packages/go/cypher/models/pgsql/translate/translator_test.go +++ b/packages/go/cypher/models/pgsql/translate/translator_test.go @@ -18,9 +18,10 @@ package translate_test import ( "fmt" - "github.com/specterops/bloodhound/dawgs/drivers/pg/pgutil" "testing" + "github.com/specterops/bloodhound/dawgs/drivers/pg/pgutil" + "github.com/specterops/bloodhound/cypher/models/pgsql" "github.com/specterops/bloodhound/cypher/models/pgsql/test" "github.com/specterops/bloodhound/dawgs/graph" diff --git a/packages/go/cypher/models/pgsql/visualization/visualizer_test.go b/packages/go/cypher/models/pgsql/visualization/visualizer_test.go index fd50b3964..5ee79d401 100644 --- a/packages/go/cypher/models/pgsql/visualization/visualizer_test.go +++ b/packages/go/cypher/models/pgsql/visualization/visualizer_test.go @@ -19,9 +19,10 @@ package visualization import ( "bytes" "context" - "github.com/specterops/bloodhound/dawgs/drivers/pg/pgutil" "testing" + "github.com/specterops/bloodhound/dawgs/drivers/pg/pgutil" + "github.com/specterops/bloodhound/cypher/frontend" "github.com/specterops/bloodhound/cypher/models/pgsql/translate" "github.com/stretchr/testify/require" diff --git a/packages/go/dawgs/drivers/pg/node_test.go b/packages/go/dawgs/drivers/pg/node_test.go index 0275953af..e3d2b21ca 100644 --- a/packages/go/dawgs/drivers/pg/node_test.go +++ b/packages/go/dawgs/drivers/pg/node_test.go @@ -18,9 +18,10 @@ package pg import ( "context" - "github.com/specterops/bloodhound/dawgs/drivers/pg/pgutil" "testing" + "github.com/specterops/bloodhound/dawgs/drivers/pg/pgutil" + "github.com/specterops/bloodhound/dawgs/graph" graph_mocks "github.com/specterops/bloodhound/dawgs/graph/mocks" "github.com/specterops/bloodhound/dawgs/query" diff --git a/packages/go/dawgs/drivers/pg/pgutil/kindmapper.go b/packages/go/dawgs/drivers/pg/pgutil/kindmapper.go index b82d8da2c..66a2eb5c1 100644 --- a/packages/go/dawgs/drivers/pg/pgutil/kindmapper.go +++ b/packages/go/dawgs/drivers/pg/pgutil/kindmapper.go @@ -19,6 +19,7 @@ package pgutil import ( "context" "fmt" + "github.com/specterops/bloodhound/dawgs/graph" )