Skip to content

Commit

Permalink
print detailed error
Browse files Browse the repository at this point in the history
  • Loading branch information
@fuyufjh
fuyufjh committed May 13, 2024
1 parent 4497a25 commit a555559
Showing 1 changed file with 25 additions and 47 deletions.
72 changes: 25 additions & 47 deletions ...ve-connector-test/src/test/java/com/risingwave/connector/sink/SinkStreamObserverTest.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -165,6 +165,13 @@ public void testOnNext_writeValidation() {
.setSinkParam(fileSinkParam))
.build();

// Encoded StreamChunk: 1 'test'
byte[] data1 =
new byte[] {
8, 1, 18, 1, 1, 26, 20, 8, 2, 18, 6, 8, 1, 18, 2, 1, 1, 26, 8, 8, 1, 18, 4, 0,
0, 0, 1, 26, 42, 8, 6, 18, 6, 8, 1, 18, 2, 1, 1, 26, 20, 8, 1, 18, 16, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4, 26, 8, 8, 1, 18, 4, 116, 101, 115, 116
};
ConnectorServiceProto.SinkWriterStreamRequest firstWrite =
ConnectorServiceProto.SinkWriterStreamRequest.newBuilder()
.setWriteBatch(
Expand All @@ -175,21 +182,7 @@ public void testOnNext_writeValidation() {
.setStreamChunkPayload(
ConnectorServiceProto.SinkWriterStreamRequest
.WriteBatch.StreamChunkPayload.newBuilder()
.setBinaryData(
ByteString.copyFrom(
new byte[] {
8, 1, 18, 1, 1, 26, 20,
8, 2, 18, 6, 8, 1, 18,
2, 1, 1, 26, 8, 8, 1,
18, 4, 0, 0, 0, 1, 26,
42, 8, 6, 18, 6, 8, 1,
18, 2, 1, 1, 26, 20, 8,
1, 18, 16, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 4, 26, 8, 8, 1,
18, 4, 116, 101, 115,
116
}))
.setBinaryData(ByteString.copyFrom(data1))
.build()))
.build();

Expand All @@ -202,6 +195,13 @@ public void testOnNext_writeValidation() {
.build())
.build();

// Encoded StreamChunk: 2 'test'
byte[] data2 =
new byte[] {
8, 1, 18, 1, 1, 26, 20, 8, 2, 18, 6, 8, 1, 18, 2, 1, 1, 26, 8, 8, 1, 18, 4, 0,
0, 0, 2, 26, 42, 8, 6, 18, 6, 8, 1, 18, 2, 1, 1, 26, 20, 8, 1, 18, 16, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 4, 26, 8, 8, 1, 18, 4, 116, 101, 115, 116
};
ConnectorServiceProto.SinkWriterStreamRequest secondWrite =
ConnectorServiceProto.SinkWriterStreamRequest.newBuilder()
.setWriteBatch(
Expand All @@ -212,21 +212,7 @@ public void testOnNext_writeValidation() {
.setStreamChunkPayload(
ConnectorServiceProto.SinkWriterStreamRequest
.WriteBatch.StreamChunkPayload.newBuilder()
.setBinaryData(
ByteString.copyFrom(
new byte[] {
8, 1, 18, 1, 1, 26, 20,
8, 2, 18, 6, 8, 1, 18,
2, 1, 1, 26, 8, 8, 1,
18, 4, 0, 0, 0, 2, 26,
42, 8, 6, 18, 6, 8, 1,
18, 2, 1, 1, 26, 20, 8,
1, 18, 16, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 4, 26, 8, 8, 1,
18, 4, 116, 101, 115,
116
}))
.setBinaryData(ByteString.copyFrom(data2))
.build()))
.build();

Expand All @@ -240,21 +226,7 @@ public void testOnNext_writeValidation() {
.setStreamChunkPayload(
ConnectorServiceProto.SinkWriterStreamRequest
.WriteBatch.StreamChunkPayload.newBuilder()
.setBinaryData(
ByteString.copyFrom(
new byte[] {
8, 1, 18, 1, 1, 26, 20,
8, 2, 18, 6, 8, 1, 18,
2, 1, 1, 26, 8, 8, 1,
18, 4, 0, 0, 0, 2, 26,
42, 8, 6, 18, 6, 8, 1,
18, 2, 1, 1, 26, 20, 8,
1, 18, 16, 0, 0, 0, 0,
0, 0, 0, 0, 0, 0, 0, 0,
0, 0, 0, 4, 26, 8, 8, 1,
18, 4, 116, 101, 115,
116
}))
.setBinaryData(ByteString.copyFrom(data2))
.build()))
.build();

Expand All @@ -266,7 +238,10 @@ public void testOnNext_writeValidation() {
sinkWriterStreamObserver.onNext(firstWrite);
} catch (RuntimeException e) {
exceptionThrown = true;
Assert.assertTrue(e.getMessage().toLowerCase().contains("batch id"));
if (!e.getMessage().toLowerCase().contains("batch id")) {
e.printStackTrace();
Assert.fail("Expected `batch id`, but got " + e.getMessage());
}
}
if (!exceptionThrown) {
Assert.fail("Expected exception not thrown: `invalid batch id`");
Expand All @@ -282,7 +257,10 @@ public void testOnNext_writeValidation() {
sinkWriterStreamObserver.onNext(secondWriteWrongEpoch);
} catch (RuntimeException e) {
exceptionThrown = true;
Assert.assertTrue(e.getMessage().toLowerCase().contains("invalid epoch"));
if (!e.getMessage().toLowerCase().contains("invalid epoch")) {
e.printStackTrace();
Assert.fail("Expected `invalid epoch`, but got " + e.getMessage());
}
}
if (!exceptionThrown) {
Assert.fail(
Expand Down

0 comments on commit a555559

Please sign in to comment.