Skip to content

Commit

Permalink
fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
linzhou-db committed Oct 7, 2024
1 parent a9970b3 commit 2ac5909
Showing 1 changed file with 3 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ import org.apache.commons.io.IOUtils
import org.scalatest.{BeforeAndAfterAll, FunSuite}
import scalapb.json4s.JsonFormat

import io.delta.sharing.server.DeltaSharingService.DELTA_SHARING_INCLUDE_END_STREAM_ACTION
import io.delta.sharing.server.common.JsonUtils
import io.delta.sharing.server.common.actions.{ColumnMappingTableFeature, DeletionVectorDescriptor, DeletionVectorsTableFeature, DeltaAddFile, DeltaFormat, DeltaProtocol, DeltaSingleAction}
import io.delta.sharing.server.config.ServerConfig
Expand Down Expand Up @@ -158,7 +159,7 @@ class DeltaSharingServiceSuite extends FunSuite with BeforeAndAfterAll {
deltaSharingCapabilities += s";responseformat=$responseFormat"
deltaSharingCapabilities += readerFeatures
if (includeEndStreamAction) {
deltaSharingCapabilities += s";endstreamaction=true"
deltaSharingCapabilities += s";$DELTA_SHARING_INCLUDE_END_STREAM_ACTION=true"
}
connection.setRequestProperty("delta-sharing-capabilities", deltaSharingCapabilities)

Expand Down Expand Up @@ -187,7 +188,7 @@ class DeltaSharingServiceSuite extends FunSuite with BeforeAndAfterAll {
val responseCapabilities = connection.getHeaderField("delta-sharing-capabilities")
var expectedHeader = s"responseformat=$responseFormat"
if (includeEndStreamAction) {
expectedHeader += s";endstreamaction=true"
expectedHeader += s";$DELTA_SHARING_INCLUDE_END_STREAM_ACTION=true"
}
assert(responseCapabilities == expectedHeader, s"Incorrect header: $responseCapabilities")
}
Expand Down

0 comments on commit 2ac5909

Please sign in to comment.