Changes to fix all the test failures with Bazel

.bazelproject

@@ -0,0 +1,22 @@
+directories:
+  # Add the directories you want added as source here
+  # By default, we've added your entire workspace ('.')
+  .
+
+# Automatically includes all relevant targets under the 'directories' above
+derive_targets_from_directories: true
+
+targets:
+  # If source code isn't resolving, add additional targets that compile it here
+
+additional_languages:
+  # Uncomment any additional languages you want supported
+  # android
+  # dart
+  # go
+  # javascript
+  # kotlin
+  python
+  scala
+  typescript
+  java

.github/image/Dockerfile

@@ -19,6 +19,10 @@ ENV PATH=$PATH:$JAVA_HOME/bin
 RUN curl -L "https://github.com/sbt/sbt/releases/download/v1.8.2/sbt-1.8.2.tgz" | tar -xz -C /usr/local
 ENV PATH="/usr/local/sbt/bin:${PATH}"
 
+# bazel
+RUN curl -fsSL "https://github.com/bazelbuild/bazelisk/releases/download/v1.18.0/bazelisk-linux-amd64" -o /usr/local/bin/bazel
+ENV PATH="/usr/local/bin/bazel:${PATH}"
+
 # thrift
 ARG THRIFT_VERSION=0.21.0
 RUN apk add --no-cache \

.gitignore

@@ -11,6 +11,7 @@
 *.iml
 *.db
 .idea/
+.ijwb/
 **/local_warehouse/
 .eclipse
 **/.vscode/

BUILD.bazel

@@ -0,0 +1 @@
+package(default_visibility = ["//visibility:public"])

WORKSPACE

@@ -23,22 +23,29 @@ http_archive(
         "https://github.com/bazelbuild/rules_java/releases/download/7.0.6/rules_java-7.0.6.tar.gz",
     ],
 )
-load("@rules_java//java:repositories.bzl", "rules_java_dependencies", "rules_java_toolchains")
+
+load("@rules_java//java:repositories.bzl", "remote_jdk17_repos", "rules_java_dependencies", "rules_java_toolchains")
+
 rules_java_dependencies()
+
 rules_java_toolchains()
-load("@rules_java//java:repositories.bzl", "remote_jdk17_repos")
+
 remote_jdk17_repos()
 
 # For JVM support
 http_archive(
     name = "rules_jvm_external",
-    strip_prefix = "rules_jvm_external-6.6",
     sha256 = "3afe5195069bd379373528899c03a3072f568d33bd96fe037bd43b1f590535e7",
-    url = "https://github.com/bazel-contrib/rules_jvm_external/releases/download/6.6/rules_jvm_external-6.6.tar.gz"
+    strip_prefix = "rules_jvm_external-6.6",
+    url = "https://github.com/bazel-contrib/rules_jvm_external/releases/download/6.6/rules_jvm_external-6.6.tar.gz",
 )
+
 load("@rules_jvm_external//:repositories.bzl", "rules_jvm_external_deps")
+
 rules_jvm_external_deps()
+
 load("@rules_jvm_external//:setup.bzl", "rules_jvm_external_setup")
+
 rules_jvm_external_setup()
 
 # For additional rulesets like java_test_suite
@@ -48,9 +55,13 @@ http_archive(
     strip_prefix = "rules_jvm-0.24.0",
     urls = ["https://github.com/bazel-contrib/rules_jvm/releases/download/v0.24.0/rules_jvm-v0.24.0.tar.gz"],
 )
+
 load("@contrib_rules_jvm//:repositories.bzl", "contrib_rules_jvm_deps")
+
 contrib_rules_jvm_deps()
+
 load("@contrib_rules_jvm//:setup.bzl", "contrib_rules_jvm_setup")
+
 contrib_rules_jvm_setup()
 
 # For Scala support
@@ -60,15 +71,37 @@ http_archive(
     strip_prefix = "rules_scala-6.6.0",
     url = "https://github.com/bazelbuild/rules_scala/releases/download/v6.6.0/rules_scala-v6.6.0.tar.gz",
 )
+
 # Initialize Scala with specific version support
 load("@io_bazel_rules_scala//:scala_config.bzl", "scala_config")
+
 scala_config(scala_version = SCALA_VERSION)
+
+load("@io_bazel_rules_scala//scala:scala_maven_import_external.bzl", "scala_maven_import_external")
+
+scala_maven_import_external(
+    name = "scala_compiler_source_2_12_18",
+    artifact = "org.scala-lang:scala-compiler:%s:sources" % SCALA_VERSION,
+    artifact_sha256 = "f79ee80f140218253f2a38c9d73f8a9b552d06afce7a5f61cf08079a388e21df",
+    licenses = ["notice"],
+    server_urls = [
+        "https://repo1.maven.org/maven2",
+        "https://mirror.bazel.build/repo1.maven.org/maven2",
+    ],
+)
+
 load("@io_bazel_rules_scala//scala:scala.bzl", "scala_repositories")
+
 scala_repositories()
+
 load("@io_bazel_rules_scala//scala:toolchains.bzl", "scala_register_toolchains")
+
 scala_register_toolchains()
+
 load("@io_bazel_rules_scala//testing:scalatest.bzl", "scalatest_repositories", "scalatest_toolchain")
+
 scalatest_repositories()
+
 scalatest_toolchain()
 
 # For Protobuf support
@@ -80,10 +113,14 @@ http_archive(
         "https://github.com/bazelbuild/rules_proto/archive/refs/tags/5.3.0-21.7.tar.gz",
     ],
 )
+
 load("@rules_proto//proto:repositories.bzl", "rules_proto_dependencies", "rules_proto_toolchains")
+
 rules_proto_dependencies()
+
 rules_proto_toolchains()
 
 # To load all dependencies used across our modules
 load("//tools/build_rules/dependencies:load_dependencies.bzl", "load_all_dependencies")
-load_all_dependencies()
+
+load_all_dependencies()

aggregator/BUILD.bazel

@@ -17,45 +17,45 @@ scala_library(
     ],
 )
 
+test_deps = [
+    ":lib",
+    "//api:lib",
+    "//api:thrift",
+    # Libraries
+    maven_artifact("org.slf4j:slf4j-api"),
+    maven_artifact("org.apache.logging.log4j:log4j-slf4j-impl"),
+    maven_artifact("com.google.code.gson:gson"),
+    maven_artifact("org.apache.datasketches:datasketches-memory"),
+    maven_artifact("org.apache.datasketches:datasketches-java"),
+    maven_artifact("org.apache.commons:commons-lang3"),
+    maven_artifact("org.apache.commons:commons-math3"),
+    maven_artifact("org.scala-lang.modules:scala-collection-compat_2.12"),
+    # Testing
+    scala_artifact("org.scalatest:scalatest-matchers-core"),
+    scala_artifact("org.scalatest:scalatest-core"),
+    scala_artifact("org.scalatest:scalatest"),
+    scala_artifact("org.scalatest:scalatest-flatspec"),
+    scala_artifact("org.scalatest:scalatest-funsuite"),
+    scala_artifact("org.scalatest:scalatest-shouldmatchers"),
+    scala_artifact("org.scalactic:scalactic"),
+    scala_artifact("org.scalatestplus:mockito-3-4"),
+    scala_artifact("org.mockito:mockito-scala"),
+    maven_artifact("org.mockito:mockito-core"),
+    maven_artifact("org.scalatest:scalatest-compatible"),
+    maven_artifact("junit:junit"),
+    maven_artifact("com.novocode:junit-interface"),
+]
+
 scala_library(
     name = "test-lib",
     srcs = glob(["src/test/**/*.scala"]),
     visibility = ["//visibility:public"],
-    deps = [
-        ":lib",
-        "//api:lib",
-        "//api:thrift",
-        maven_artifact("junit:junit"),
-        maven_artifact("com.novocode:junit-interface"),
-        maven_artifact("org.slf4j:slf4j-api"),
-        maven_artifact("org.apache.logging.log4j:log4j-slf4j-impl"),
-        maven_artifact("com.google.code.gson:gson"),
-        maven_artifact("org.apache.datasketches:datasketches-memory"),
-        maven_artifact("org.apache.datasketches:datasketches-java"),
-        maven_artifact("org.apache.commons:commons-lang3"),
-        maven_artifact("org.apache.commons:commons-math3"),
-        maven_artifact("org.scala-lang.modules:scala-collection-compat_2.12"),
-    ],
+    deps = test_deps,
 )
 
 scala_test_suite(
     name = "test",
     srcs = glob(["src/test/**/*.scala"]),
     visibility = ["//visibility:public"],
-    deps = [
-       ":lib",
-       ":test-lib",
-       "//api:lib",
-       "//api:thrift",
-       maven_artifact("junit:junit"),
-       maven_artifact("com.novocode:junit-interface"),
-       maven_artifact("org.slf4j:slf4j-api"),
-       maven_artifact("org.apache.logging.log4j:log4j-slf4j-impl"),
-       maven_artifact("com.google.code.gson:gson"),
-       maven_artifact("org.apache.datasketches:datasketches-memory"),
-       maven_artifact("org.apache.datasketches:datasketches-java"),
-       maven_artifact("org.apache.commons:commons-lang3"),
-       maven_artifact("org.apache.commons:commons-math3"),
-       maven_artifact("org.scala-lang.modules:scala-collection-compat_2.12"),
-   ],
+    deps = test_deps + [":test-lib"],
 )

aggregator/src/test/scala/ai/chronon/aggregator/test/ApproxHistogramTest.scala

@@ -42,11 +42,12 @@ class ApproxHistogramTest extends AnyFlatSpec {
     assertTrue(ir2.isApprox)
 
     val ir = approxHistogram.merge(ir1, ir2)
-    assertEquals(toHashMap(Map(
-                   "4" -> 8,
-                   "6" -> 6,
-                   "5" -> 5
-                 )),
+    assertEquals(toHashMap(
+                   Map(
+                     "4" -> 8,
+                     "6" -> 6,
+                     "5" -> 5
+                   )),
                  approxHistogram.finalize(ir))
     assertTrue(ir.isApprox)
     assertTrue(ir.histogram.isEmpty)
@@ -65,11 +66,13 @@ class ApproxHistogramTest extends AnyFlatSpec {
 
     val ir = approxHistogram.merge(ir1, ir2)
 
-    assertEquals(toHashMap(Map(
-      "2" -> 4,
-      "4" -> 4,
-      "3" -> 3
-    )), approxHistogram.finalize(ir))
+    assertEquals(toHashMap(
+                   Map(
+                     "2" -> 4,
+                     "4" -> 4,
+                     "3" -> 3
+                   )),
+                 approxHistogram.finalize(ir))
     assertTrue(!ir.isApprox)
     assertTrue(ir.sketch.isEmpty)
   }
@@ -87,11 +90,13 @@ class ApproxHistogramTest extends AnyFlatSpec {
 
     val ir = approxHistogram.merge(ir1, ir2)
 
-    assertEquals(toHashMap(Map(
-      "4" -> 4,
-      "3" -> 3,
-      "2" -> 2
-    )), approxHistogram.finalize(ir))
+    assertEquals(toHashMap(
+                   Map(
+                     "4" -> 4,
+                     "3" -> 3,
+                     "2" -> 2
+                   )),
+                 approxHistogram.finalize(ir))
 
     assertTrue(ir.isApprox)
     assertTrue(ir.histogram.isEmpty)
@@ -110,11 +115,13 @@ class ApproxHistogramTest extends AnyFlatSpec {
 
     val ir = approxHistogram.merge(ir1, ir2)
 
-    assertEquals(toHashMap(Map(
-      "5" -> 5,
-      "2" -> 4,
-      "3" -> 3
-    )), approxHistogram.finalize(ir))
+    assertEquals(toHashMap(
+                   Map(
+                     "5" -> 5,
+                     "2" -> 4,
+                     "3" -> 3
+                   )),
+                 approxHistogram.finalize(ir))
     assertTrue(ir.isApprox)
     assert(ir.histogram.isEmpty)
   }

aggregator/src/test/scala/ai/chronon/aggregator/test/DataGen.scala

@@ -65,7 +65,10 @@ abstract class CStream[+T: ClassTag] {
     }
   }
 
-  def zipChunk[Other](other: CStream[Other], minSize: Int = 0, maxSize: Int = 20, nullRate: Double = 0.1): CStream[Any] = {
+  def zipChunk[Other](other: CStream[Other],
+                      minSize: Int = 0,
+                      maxSize: Int = 20,
+                      nullRate: Double = 0.1): CStream[Any] = {
     def nextKey(): T = next()
     def nextValue(): Other = other.next()
 
@@ -91,9 +94,12 @@ object CStream {
                     count: Int,
                     roundMillis: Int = 1,
                     maxTs: Long = System.currentTimeMillis()): Array[Long] =
-    new CStream.TimeStream(window, roundMillis, maxTs).gen(count).toArray.sorted(new Ordering[Any] {
-      override def compare(x: Any, y: Any): Int = x.asInstanceOf[Long].compareTo(y.asInstanceOf[Long])
-    })
+    new CStream.TimeStream(window, roundMillis, maxTs)
+      .gen(count)
+      .toArray
+      .sorted(new Ordering[Any] {
+        override def compare(x: Any, y: Any): Int = x.asInstanceOf[Long].compareTo(y.asInstanceOf[Long])
+      })
 
   def genPartitions(count: Int, partitionSpec: PartitionSpec): Array[String] = {
     val today = partitionSpec.at(System.currentTimeMillis())
@@ -178,7 +184,7 @@ case class Column(name: String, `type`: DataType, cardinality: Int, chunkSize: I
         }
       case IntType    => new IntStream(cardinality, nullRate)
       case DoubleType => new DoubleStream(cardinality, nullRate)
-      case FloatType => new FloatStream(cardinality, nullRate)
+      case FloatType  => new FloatStream(cardinality, nullRate)
       case LongType =>
         name match {
           case Constants.TimeColumn => new TimeStream(new Window(cardinality, TimeUnit.DAYS))