update backend tests

Signed-off-by: pxp928 <[email protected]>
guacsec · Oct 10, 2024 · 135f1af · 135f1af
1 parent 4208868
commit 135f1af
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/internal/testing/backend/search_test.go b/internal/testing/backend/search_test.go
@@ -381,7 +381,13 @@ func TestQueryPackagesListForScan(t *testing.T) {
  t.Fatalf("did not get expected ingest error: %v", err)
  }
  }
- got, err := b.QueryPackagesListForScan(ctx, model.PkgSpec{}, test.QueryType, test.lastScan, nil, ptrfrom.Int(10))
+
+ pkgIDsResponse, err := b.FindPackagesThatNeedScanning(ctx, model.PkgSpec{}, test.QueryType, test.lastScan)
+ if err != nil {
+ t.Fatalf("did not get expected query error: %v", err)
+ }
+
+ got, err := b.QueryPackagesListForScan(ctx, pkgIDsResponse, nil, ptrfrom.Int(10))
  if err != nil {
  t.Fatalf("did not get expected query error: %v", err)
  }

diff --git a/pkg/assembler/backends/keyvalue/search.go b/pkg/assembler/backends/keyvalue/search.go
@@ -333,15 +333,14 @@ func (c *demoClient) FindPackagesThatNeedScanning(ctx context.Context, pkgSpec m
  }
  } else {
  pkgIDs = append(pkgIDs, pkgVer.ThisID)
-
  }
  }
  }
  }
  }
  }
  }
- return nil, nil
+ return pkgIDs, nil
 }
 
 func (c *demoClient) QueryPackagesListForScan(ctx context.Context, pkgIDs []string, after *string, first *int) (*model.PackageConnection, error) {