Reduce duplicate code using a test class

PiperOrigin-RevId: 652947486
tensorflow · Jul 16, 2024 · 51cf3a7 · 51cf3a7
1 parent 9161c60
commit 51cf3a7
Show file tree

Hide file tree

Showing 2 changed files with 188 additions and 199 deletions.
diff --git a/tensorflow_serving/core/BUILD b/tensorflow_serving/core/BUILD
@@ -428,8 +428,12 @@ cc_test(
     size = "small",
     srcs = ["servable_state_monitor_test.cc"],
     deps = [
+        ":manager",
+        ":servable_id",
+        ":servable_state",
         ":servable_state_monitor",
         "//tensorflow_serving/core/test_util:test_main",
+        "//tensorflow_serving/util:event_bus",
         "@org_tensorflow//tensorflow/core:lib",
         "@org_tensorflow//tensorflow/core/kernels/batching_util:fake_clock_env",
     ],