change hasSBOMList to add filter based on client usage (#2205)

* change hasSBOMList to filter out deps.dev, add filter based on client usage Signed-off-by: pxp928 <[email protected]> * udpate backend tests with graphql preloads Signed-off-by: pxp928 <[email protected]> * remove filtering on deps.dev sboms Signed-off-by: pxp928 <[email protected]> * add unit test for preloads helper Signed-off-by: pxp928 <[email protected]> --------- Signed-off-by: pxp928 <[email protected]>
guacsec · Oct 22, 2024 · 8e84bbe · 8e84bbe
1 parent 7bebe65
commit 8e84bbe
Show file tree

Hide file tree

Showing 8 changed files with 403 additions and 202 deletions.
diff --git a/internal/testing/backend/hasSBOM_test.go b/internal/testing/backend/hasSBOM_test.go
@@ -21,11 +21,13 @@ import (
  "context"
  "testing"
 
+ "github.com/99designs/gqlgen/graphql"
  "github.com/google/go-cmp/cmp"
  "github.com/guacsec/guac/internal/testing/ptrfrom"
  "github.com/guacsec/guac/internal/testing/testdata"
  "github.com/guacsec/guac/pkg/assembler/graphql/model"
  "github.com/stretchr/testify/assert"
+ "github.com/vektah/gqlparser/v2/ast"
 )
 
 type testDependency struct {
@@ -287,8 +289,81 @@ var includedTestExpectedSBOM = &model.HasSbom{
 
 // End of Test resources
 
+func getGraphQLPreloadCtx() context.Context {
+ ctx := graphql.WithOperationContext(context.Background(), &graphql.OperationContext{})
+ ctx = graphql.WithFieldContext(ctx, &graphql.FieldContext{
+ Field: graphql.CollectedField{
+ Selections: ast.SelectionSet{
+ &ast.Field{
+ Alias: "totalCount",
+ Name: "totalCount",
+ },
+ &ast.Field{
+ Alias: "edges",
+ Name: "edges",
+ SelectionSet: ast.SelectionSet{
+ &ast.Field{
+ Alias: "cursor",
+ Name: "cursor",
+ },
+ &ast.Field{
+ Alias: "node",
+ Name: "node",
+ SelectionSet: ast.SelectionSet{
+ &ast.Field{
+ Alias: "includedSoftware",
+ Name: "includedSoftware",
+ },
+ },
+ },
+ &ast.Field{
+ Alias: "node",
+ Name: "node",
+ SelectionSet: ast.SelectionSet{
+ &ast.Field{
+ Alias: "includedDependencies",
+ Name: "includedDependencies",
+ },
+ },
+ },
+ &ast.Field{
+ Alias: "node",
+ Name: "node",
+ SelectionSet: ast.SelectionSet{
+ &ast.Field{
+ Alias: "includedOccurrences",
+ Name: "includedOccurrences",
+ },
+ },
+ },
+ },
+ },
+ &ast.Field{
+ Alias: "pageInfo",
+ Name: "pageInfo",
+ SelectionSet: ast.SelectionSet{
+ &ast.Field{
+ Alias: "startCursor",
+ Name: "startCursor",
+ },
+ &ast.Field{
+ Alias: "endCursor",
+ Name: "endCursor",
+ },
+ &ast.Field{
+ Alias: "hasNextPage",
+ Name: "hasNextPage",
+ },
+ },
+ },
+ },
+ },
+ })
+ return ctx
+}
+
 func TestHasSBOM(t *testing.T) {
- ctx := context.Background()
+ ctx := getGraphQLPreloadCtx()
  b := setupTest(t)
  type call struct {
  Sub model.PackageOrArtifactInput
@@ -2780,7 +2855,7 @@ func TestHasSBOM(t *testing.T) {
 }
 
 func TestIngestHasSBOMs(t *testing.T) {
- ctx := context.Background()
+ ctx := getGraphQLPreloadCtx()
  b := setupTest(t)
  type call struct {
  Sub model.PackageOrArtifactInputs
@@ -3127,7 +3202,7 @@ func TestIngestHasSBOMs(t *testing.T) {
 }
 
 func TestDeleteHasSBOM(t *testing.T) {
- ctx := context.Background()
+ ctx := getGraphQLPreloadCtx()
  b := setupTest(t)
  type call struct {
  Sub model.PackageOrArtifactInputs