Qbeast-io#207 Fixes for NormalizedWeightIntegrationTest

src/test/scala/io/qbeast/spark/index/NormalizedWeightIntegrationTest.scala

@@ -3,9 +3,9 @@ package io.qbeast.spark.index
 import io.qbeast.TestClasses.Client3
 import io.qbeast.spark.QbeastIntegrationTestSpec
 import io.qbeast.spark.delta.DeltaQbeastSnapshot
-import io.qbeast.spark.utils.TagUtils
 import org.apache.spark.sql.delta.DeltaLog
 import org.apache.spark.sql.{Dataset, SparkSession}
+import io.qbeast.spark.utils.State
 
 class NormalizedWeightIntegrationTest extends QbeastIntegrationTestSpec {
 
@@ -22,7 +22,6 @@ class NormalizedWeightIntegrationTest extends QbeastIntegrationTestSpec {
   "CubeNormalizedWeights" should
     "write a the right Weight with a single full file" in withQbeastContextSparkAndTmpDir {
       (spark, tmpDir) =>
-        import spark.implicits._
         val cubeSize = 10000
         val df = createDF(cubeSize).repartition(1)
         val names = List("age", "val2")
@@ -36,11 +35,13 @@ class NormalizedWeightIntegrationTest extends QbeastIntegrationTestSpec {
         spark.read.format("qbeast").load(tmpDir).count() shouldBe cubeSize
 
         val deltaLog = DeltaLog.forTable(spark, tmpDir)
-        val files = deltaLog.snapshot.allFiles
-        files.count() shouldBe 1
-        files.map(_.tags(TagUtils.maxWeight).toInt).collect()(0) shouldBe <=(Int.MaxValue)
-        files.map(_.tags(TagUtils.state)).collect()(0) shouldBe "FLOODED"
-
+        val snapshot = DeltaQbeastSnapshot(deltaLog.snapshot)
+        val index = snapshot.loadLatestIndexStatus
+        index.cubesStatuses.size shouldBe 1
+        val status = index.cubesStatuses.values.head
+        status.files.size shouldBe 1
+        val block = status.files.head
+        block.state shouldBe State.FLOODED
     }
 
   it should