fix

Author: wankai123

src/test/java/org/apache/skywalking/banyandb/v1/client/ITBanyanDBMeasureQueryTests.java

@@ -36,6 +36,7 @@
 import org.apache.skywalking.banyandb.database.v1.BanyandbDatabase.TagSpec;
 import org.apache.skywalking.banyandb.database.v1.BanyandbDatabase.TagType;
 import org.apache.skywalking.banyandb.measure.v1.BanyandbMeasure;
+import org.apache.skywalking.banyandb.model.v1.BanyandbModel;
 import org.apache.skywalking.banyandb.v1.client.grpc.exception.BanyanDBException;
 import org.apache.skywalking.banyandb.v1.client.metadata.Duration;
 import org.junit.After;
@@ -81,7 +82,7 @@ public void testMeasureQuery() throws BanyanDBException, ExecutionException, Int
             = client.getMeasureServiceStub().write(new StreamObserver<BanyandbMeasure.WriteResponse>() {
             @Override
             public void onNext(BanyandbMeasure.WriteResponse writeResponse) {
-                Assert.assertEquals("STATUS_SUCCEED", writeResponse.getStatus());
+                Assert.assertEquals(BanyandbModel.Status.STATUS_SUCCEED.name(), writeResponse.getStatus());
             }
 
             @Override

src/test/java/org/apache/skywalking/banyandb/v1/client/ITBanyanDBStreamQueryTests.java

@@ -38,14 +38,14 @@
 import org.apache.skywalking.banyandb.database.v1.BanyandbDatabase.Stream;
 import org.apache.skywalking.banyandb.database.v1.BanyandbDatabase.IndexRule;
 import org.apache.skywalking.banyandb.database.v1.BanyandbDatabase.IndexRuleBinding;
+import org.apache.skywalking.banyandb.model.v1.BanyandbModel;
 import org.apache.skywalking.banyandb.stream.v1.BanyandbStream;
 import org.apache.skywalking.banyandb.v1.client.grpc.exception.BanyanDBException;
 import org.apache.skywalking.banyandb.v1.client.util.TimeUtils;
 import org.junit.After;
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
-
 import java.io.IOException;
 import java.time.Instant;
 import java.util.concurrent.ExecutionException;
@@ -110,7 +110,7 @@ public void testStreamQuery_TraceID() throws BanyanDBException, ExecutionExcepti
             = client.getStreamServiceStub().write(new StreamObserver<BanyandbStream.WriteResponse>() {
             @Override
             public void onNext(BanyandbStream.WriteResponse writeResponse) {
-                Assert.assertEquals("STATUS_SUCCEED", writeResponse.getStatus());
+                Assert.assertEquals(BanyandbModel.Status.STATUS_SUCCEED.name(), writeResponse.getStatus());
             }
 
             @Override

src/test/java/org/apache/skywalking/banyandb/v1/client/ITTraceTest.java

@@ -23,14 +23,14 @@
 import io.grpc.stub.StreamObserver;
 import org.apache.skywalking.banyandb.common.v1.BanyandbCommon;
 import org.apache.skywalking.banyandb.database.v1.BanyandbDatabase;
+import org.apache.skywalking.banyandb.model.v1.BanyandbModel;
 import org.apache.skywalking.banyandb.trace.v1.BanyandbTrace;
 import org.apache.skywalking.banyandb.v1.client.grpc.exception.BanyanDBException;
 import org.apache.skywalking.banyandb.v1.client.util.TimeUtils;
 import org.junit.After;
 import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
-
 import java.io.IOException;
 import java.time.Instant;
 import java.time.ZoneOffset;
@@ -136,7 +136,7 @@ public void testTraceQueryByTraceId() throws BanyanDBException, ExecutionExcepti
             = client.getTraceServiceStub().write(new StreamObserver<BanyandbTrace.WriteResponse>() {
             @Override
             public void onNext(BanyandbTrace.WriteResponse writeResponse) {
-                Assert.assertEquals("STATUS_SUCCEED", writeResponse.getStatus());
+                Assert.assertEquals(BanyandbModel.Status.STATUS_SUCCEED.name(), writeResponse.getStatus());
             }
 
             @Override
@@ -154,13 +154,6 @@ public void onCompleted() {
         } finally {
             writeObserver.onCompleted();
         }
-        // Write the trace via bulk processor
-//        CompletableFuture<Void> writeFuture = processor.add(traceWrite);
-//        writeFuture.exceptionally(exp -> {
-//            Assert.fail("Write failed: " + exp.getMessage());
-//            return null;
-//        });
-//        writeFuture.get(10, TimeUnit.SECONDS);
 
         // Create trace query with trace_id condition
         TraceQuery query = new TraceQuery(
@@ -232,7 +225,7 @@ public void testTraceQueryOrderByStartTime() throws BanyanDBException, Execution
             = client.getTraceServiceStub().write(new StreamObserver<BanyandbTrace.WriteResponse>() {
             @Override
             public void onNext(BanyandbTrace.WriteResponse writeResponse) {
-                Assert.assertEquals("STATUS_SUCCEED", writeResponse.getStatus());
+                Assert.assertEquals(BanyandbModel.Status.STATUS_SUCCEED.name(), writeResponse.getStatus());
             }
 
             @Override