1 files changed, 52 insertions, 0 deletions
diff --git a/pkg/segment/test/BUILD b/pkg/segment/test/BUILD
new file mode 100644
index 000000000..9d398d71a
--- /dev/null
+++ b/pkg/segment/test/BUILD
@@ -0,0 +1,52 @@
+load("@io_bazel_rules_go//go:def.bzl", "go_library", "go_test")
+
+package(
+    default_visibility = ["//visibility:private"],
+    licenses = ["notice"],  # Apache 2.0
+)
+
+load("//tools/go_generics:defs.bzl", "go_template_instance")
+
+go_template_instance(
+    name = "int_range",
+    out = "int_range.go",
+    package = "segment",
+    template = "//pkg/segment:generic_range",
+    types = {
+        "T": "int",
+    },
+)
+
+go_template_instance(
+    name = "int_set",
+    out = "int_set.go",
+    package = "segment",
+    template = "//pkg/segment:generic_set",
+    types = {
+        "Key": "int",
+        "Range": "Range",
+        "Value": "int",
+        "Functions": "setFunctions",
+    },
+)
+
+go_library(
+    name = "segment",
+    testonly = 1,
+    srcs = [
+        "int_range.go",
+        "int_set.go",
+        "set_functions.go",
+    ],
+    importpath = "gvisor.googlesource.com/gvisor/pkg/segment/segment",
+    deps = [
+        "//pkg/state",
+    ],
+)
+
+go_test(
+    name = "segment_test",
+    size = "small",
+    srcs = ["segment_test.go"],
+    embed = [":segment"],
+)