From 379a985ddef555d0b97097e822e490dcb0689cee Mon Sep 17 00:00:00 2001
From: Mizux Seiha <mizux.dev@gmail.com>
Date: Fri, 13 Nov 2020 16:08:47 +0100
Subject: [PATCH] bazel: Run samples as test

---
 ortools/algorithms/BUILD                          |  2 +-
 ortools/algorithms/samples/code_samples.bzl       | 12 +++++++++++-
 .../constraint_solver/samples/code_samples.bzl    | 14 ++++++++++++++
 ortools/graph/samples/code_samples.bzl            | 15 +++++++++++++++
 ortools/linear_solver/samples/code_samples.bzl    | 12 ++++++++++++
 ortools/sat/samples/code_samples.bzl              | 15 +++++++++++++--
 6 files changed, 66 insertions(+), 4 deletions(-)

diff --git a/ortools/algorithms/BUILD b/ortools/algorithms/BUILD
index 4970c77cc3..e2956c67e3 100644
--- a/ortools/algorithms/BUILD
+++ b/ortools/algorithms/BUILD
@@ -14,7 +14,7 @@ cc_library(
 
 cc_test(
     name = "hungarian_test",
-    size = "medium",
+    size = "small",
     srcs = ["hungarian_test.cc"],
     copts = ["-Iexternal/gtest/include"],
     deps = [
diff --git a/ortools/algorithms/samples/code_samples.bzl b/ortools/algorithms/samples/code_samples.bzl
index 610780294a..1c98e396d2 100644
--- a/ortools/algorithms/samples/code_samples.bzl
+++ b/ortools/algorithms/samples/code_samples.bzl
@@ -3,7 +3,17 @@ def code_sample_cc(sample):
       name = sample,
       srcs = [sample + ".cc"],
       deps = [
-          "//ortools/algorithms:knapsack_solver_lib",
+        "//ortools/algorithms:knapsack_solver_lib",
+      ],
+  )
+
+  native.cc_test(
+      name = sample+"_test",
+      size = "small",
+      srcs = [sample + ".cc"],
+      deps = [
+        ":"+sample,
+        "//ortools/algorithms:knapsack_solver_lib",
       ],
   )
 
diff --git a/ortools/constraint_solver/samples/code_samples.bzl b/ortools/constraint_solver/samples/code_samples.bzl
index 8df694e576..1f221eb843 100644
--- a/ortools/constraint_solver/samples/code_samples.bzl
+++ b/ortools/constraint_solver/samples/code_samples.bzl
@@ -11,3 +11,17 @@ def code_sample_cc(sample):
       ],
   )
 
+  native.cc_test(
+      name = sample+"_test",
+      size = "small",
+      srcs = [sample + ".cc"],
+      deps = [
+        ":"+sample,
+        "//ortools/base",
+        "//ortools/constraint_solver:cp",
+        "//ortools/constraint_solver:routing",
+        "//ortools/constraint_solver:routing_enums_cc_proto",
+        "//ortools/constraint_solver:routing_flags",
+      ],
+  )
+
diff --git a/ortools/graph/samples/code_samples.bzl b/ortools/graph/samples/code_samples.bzl
index 89598ad644..bd6f63bfaa 100644
--- a/ortools/graph/samples/code_samples.bzl
+++ b/ortools/graph/samples/code_samples.bzl
@@ -12,3 +12,18 @@ def code_sample_cc(sample):
       ],
   )
 
+  native.cc_test(
+      name = sample+"_test",
+      size = "small",
+      srcs = [sample + ".cc"],
+      deps = [
+        ":"+sample,
+        "//ortools/base",
+        "//ortools/graph:min_cost_flow",
+        "//ortools/graph:max_flow",
+        "//ortools/graph:shortestpaths",
+        "//ortools/graph:ebert_graph",
+        "//ortools/graph:linear_assignment",
+      ],
+  )
+
diff --git a/ortools/linear_solver/samples/code_samples.bzl b/ortools/linear_solver/samples/code_samples.bzl
index 93f64d295d..b69343fe33 100644
--- a/ortools/linear_solver/samples/code_samples.bzl
+++ b/ortools/linear_solver/samples/code_samples.bzl
@@ -9,3 +9,15 @@ def code_sample_cc(sample):
       ],
   )
 
+  native.cc_test(
+      name = sample+"_test",
+      size = "small",
+      srcs = [sample + ".cc"],
+      deps = [
+        ":"+sample,
+        "//ortools/base",
+        "//ortools/linear_solver",
+        "//ortools/linear_solver:linear_solver_cc_proto",
+      ],
+  )
+
diff --git a/ortools/sat/samples/code_samples.bzl b/ortools/sat/samples/code_samples.bzl
index fb6d9efb7d..c9b5f76446 100644
--- a/ortools/sat/samples/code_samples.bzl
+++ b/ortools/sat/samples/code_samples.bzl
@@ -3,8 +3,19 @@ def code_sample_cc(sample):
       name = sample,
       srcs = [sample + ".cc"],
       deps = [
-          "//ortools/sat:cp_model",
-          "//ortools/sat:cp_model_solver",
+        "//ortools/sat:cp_model",
+        "//ortools/sat:cp_model_solver",
+      ],
+  )
+
+  native.cc_test(
+      name = sample+"_test",
+      size = "small",
+      srcs = [sample + ".cc"],
+      deps = [
+        ":"+sample,
+        "//ortools/sat:cp_model",
+        "//ortools/sat:cp_model_solver",
       ],
   )