test_deps_files = \
    heron_java_proto_files() + [
      "//heron/spi/src/java:statefulstorage-spi-java",
      "//third_party/java:powermock",
      "//third_party/java:mockito",
      "//third_party/java:junit4"
    ]


localfs_deps_files = [
    "//heron/common/src/java:basics-java",
    "//heron/statefulstorages/src/java:localfs-statefulstorage-java",
]

java_library(
  name = "localfs-tests",
  srcs = glob(["**/localfs/*.java", "**/StatefulStorageTestContext.java"]),
  deps = test_deps_files + localfs_deps_files,
)

java_tests(
  test_classes = [
    "org.apache.heron.statefulstorage.localfs.LocalFileSystemStorageTest",
  ],
  runtime_deps = [":localfs-tests"],
  size = "small",
)

hdfs_deps_files = [
  "//heron/statefulstorages/src/java:hdfs-statefulstorage-java",
  "@org_apache_hadoop_hadoop_core//jar",
  "@commons_logging_commons_logging//jar",
]

java_library(
  name = "hdfs-tests",
  srcs = glob(["**/hdfs/*.java", "**/StatefulStorageTestContext.java"]),
  deps = test_deps_files + hdfs_deps_files,
)

java_tests(
  test_classes = [
    "org.apache.heron.statefulstorage.hdfs.HDFSStorageTest",
  ],
  runtime_deps = [ ":hdfs-tests" ],
  size = "small",
)

dlog_deps_files = [
  "//heron/statefulstorages/src/java:dlog-statefulstorage-java",
  "@com_google_guava_guava//jar",
  "@org_apache_distributedlog_core//jar",
  "@io_netty_netty_all//jar",
]

java_library(
  name = "dlog-tests",
  srcs = glob(["**/dlog/*.java", "**/StatefulStorageTestContext.java"]),
  deps = test_deps_files + dlog_deps_files,
)

java_test(
  name = "DlogStorageTest",
  srcs = glob(["**/dlog/*.java", "**/StatefulStorageTestContext.java"]),
  size = "small",
  deps = test_deps_files + dlog_deps_files,
)