sourcegraph/internal/embeddings/BUILD.bazel
Noah S-C 19d9cfc73b
bazel: native go-mockgen in Bazel (#60386)
Adds a new:
- gazelle generator
- rule + rule targets + catchall target
for generating go-mockgen mocks & testing for their being up-to-date.

Each go_mockgen macro invocation adds targets for generating mocks, copying to the source tree, as well as testing whether the current source tree mocks are up-to-date.

How to use this: `bazel run //dev:go_mockgen` for the catch-all, or `bazel run //some/target:generate_mocks` for an individual package, and `bazel test //some/target:generate_mocks_tests` to test for up-to-date-ness. There is no catch-all for testing

This currently uses a fork of go-mockgen, with an open PR for upstream here: https://github.com/derision-test/go-mockgen/pull/50.

Closes https://github.com/sourcegraph/sourcegraph/issues/60099

## Test plan

Extensive testing during development, including the following cases:
- Deleting a generated file and its entry in a go_library/go_test `srcs` attribute list and then re-running `sg bazel configure`
- Adding a non-existent output directory to mockgen.test.yaml and running the bash one-liner emitted to prepare the workspace for rerunning `sg bazel configure`

The existing config tests a lot of existing paths anyway (creating mocks for a 3rd party library's interface, entries for a given output file in >1 config file etc)
2024-02-16 13:26:48 +00:00

104 lines
2.8 KiB
Python

load("//dev:go_mockgen.bzl", "go_mockgen")
load("//dev:go_defs.bzl", "go_test")
load("@io_bazel_rules_go//go:def.bzl", "go_library")
go_library(
name = "embeddings",
srcs = [
"client.go",
"context_detection.go",
"dot.go",
"dot_amd64.go",
"dot_amd64.s",
"dot_arm64.go",
"dot_arm64.s",
"dot_portable.go",
"index_name.go",
"index_storage.go",
"mocks_temp.go",
"quantize.go",
"schedule.go",
"similarity_search.go",
"types.go",
"uploadstore.go",
],
importpath = "github.com/sourcegraph/sourcegraph/internal/embeddings",
visibility = ["//:__subpackages__"],
deps = [
"//internal/api",
"//internal/codeintel/types",
"//internal/conf/conftypes",
"//internal/conf/deploy",
"//internal/database",
"//internal/embeddings/background/repo",
"//internal/endpoint",
"//internal/env",
"//internal/gitserver",
"//internal/httpcli",
"//internal/lazyregexp",
"//internal/observation",
"//internal/trace",
"//internal/uploadstore",
"//lib/errors",
"@com_github_sourcegraph_conc//:conc",
"@com_github_sourcegraph_conc//pool",
"@io_opentelemetry_go_otel//attribute",
"@org_golang_x_sync//errgroup",
] + select({
"@io_bazel_rules_go//go/platform:amd64": [
"@com_github_klauspost_cpuid_v2//:cpuid",
],
"@io_bazel_rules_go//go/platform:arm64": [
"@com_github_klauspost_cpuid_v2//:cpuid",
],
"//conditions:default": [],
}),
)
go_test(
name = "embeddings_test",
timeout = "moderate",
srcs = [
"context_detection_test.go",
"dot_test.go",
"index_storage_test.go",
"quantize_test.go",
"schedule_test.go",
"similarity_search_test.go",
"types_test.go",
],
data = glob(["testdata/**"]),
embed = [":embeddings"],
tags = [
# Test requires localhost database
"requires-network",
],
deps = [
"//internal/api",
"//internal/codeintel/types",
"//internal/database",
"//internal/database/dbtest",
"//internal/embeddings/background/repo",
"//internal/gitserver",
"//internal/types",
"//internal/uploadstore",
"//lib/errors",
"//lib/iterator",
"@com_github_google_go_cmp//cmp",
"@com_github_keegancsmith_sqlf//:sqlf",
"@com_github_sourcegraph_log//logtest",
"@com_github_stretchr_testify//require",
],
)
go_mockgen(
name = "generate_mocks",
out = "mocks_temp.go",
manifests = [
"//:mockgen.yaml",
"//:mockgen.test.yaml",
"//:mockgen.temp.yaml",
],
deps = [":embeddings"],
)