update

baidubce · Jan 20, 2025 · fd1d190 · fd1d190
1 parent ebda56c
commit fd1d190
Show file tree

Hide file tree

Showing 2 changed files with 55 additions and 55 deletions.
diff --git a/go/appbuilder/dataset_test.go b/go/appbuilder/dataset_test.go
@@ -69,41 +69,41 @@ func TestDatasetError(t *testing.T) {
 	dataset, _ = NewDataset(config)
 
 }
-func TestDataset(t *testing.T) {
-	t.Parallel() // 并发运行
-	// 创建缓冲区来存储日志
-	var logBuffer bytes.Buffer
-
-	// 定义一个日志函数，将日志写入缓冲区
-	log := func(format string, args ...interface{}) {
-		fmt.Fprintf(&logBuffer, format+"\n", args...)
-	}
-
-	// 测试逻辑
-	config, err := NewSDKConfig("", os.Getenv(SecretKeyV3))
-	if err != nil {
-		t.Logf("%s========== FAIL:  %s ==========%s", "\033[31m", t.Name(), "\033[0m")
-		t.Fatalf("new http client config failed: %v", err)
-	}
-	dataset, _ := NewDataset(config)
-	datasetID, err := dataset.Create("测试集合")
-	if err != nil {
-		datasetID = os.Getenv(SecretKeyV3)
-	}
-
-	_, err = dataset.ListDocument(datasetID, 1, 10, "")
-	if err != nil {
-		t.Logf("%s========== FAIL:  %s ==========%s", "\033[31m", t.Name(), "\033[0m")
-		t.Fatalf("list document failed: %v", err)
-	}
-	log("Listed documents for dataset ID: %s", datasetID)
-
-	// 如果测试失败，则输出缓冲区中的日志
-	if t.Failed() {
-		t.Logf("%s========== FAIL:  %s ==========%s", "\033[31m", t.Name(), "\033[0m")
-		fmt.Println(logBuffer.String())
-	} else { // else 紧跟在右大括号后面
-		// 测试通过，打印文件名和测试函数名
-		t.Logf("%s========== OK:  %s ==========%s", "\033[32m", t.Name(), "\033[0m")
-	}
+// func TestDataset(t *testing.T) {
+// 	t.Parallel() // 并发运行
+// 	// 创建缓冲区来存储日志
+// 	var logBuffer bytes.Buffer
+
+// 	// 定义一个日志函数，将日志写入缓冲区
+// 	log := func(format string, args ...interface{}) {
+// 		fmt.Fprintf(&logBuffer, format+"\n", args...)
+// 	}
+
+// 	// 测试逻辑
+// 	config, err := NewSDKConfig("", os.Getenv(SecretKeyV3))
+// 	if err != nil {
+// 		t.Logf("%s========== FAIL:  %s ==========%s", "\033[31m", t.Name(), "\033[0m")
+// 		t.Fatalf("new http client config failed: %v", err)
+// 	}
+// 	dataset, _ := NewDataset(config)
+// 	datasetID, err := dataset.Create("测试集合")
+// 	if err != nil {
+// 		datasetID = os.Getenv(SecretKeyV3)
+// 	}
+
+// 	_, err = dataset.ListDocument(datasetID, 1, 10, "")
+// 	if err != nil {
+// 		t.Logf("%s========== FAIL:  %s ==========%s", "\033[31m", t.Name(), "\033[0m")
+// 		t.Fatalf("list document failed: %v", err)
+// 	}
+// 	log("Listed documents for dataset ID: %s", datasetID)
+
+// 	// 如果测试失败，则输出缓冲区中的日志
+// 	if t.Failed() {
+// 		t.Logf("%s========== FAIL:  %s ==========%s", "\033[31m", t.Name(), "\033[0m")
+// 		fmt.Println(logBuffer.String())
+// 	} else { // else 紧跟在右大括号后面
+// 		// 测试通过，打印文件名和测试函数名
+// 		t.Logf("%s========== OK:  %s ==========%s", "\033[32m", t.Name(), "\033[0m")
+// 	}
 }
diff --git a/java/src/test/java/com/baidubce/appbuilder/DatasetTest.java b/java/src/test/java/com/baidubce/appbuilder/DatasetTest.java
@@ -19,27 +19,27 @@ public void setUp()  {
         System.setProperty("APPBUILDER_TOKEN", System.getenv("APPBUILDER_TOKEN_V3"));
     }
 
-    @Test
-    public void testCreateDataset() throws IOException, AppBuilderServerException {
-        Dataset dataset = new Dataset();
+    // @Test
+    // public void testCreateDataset() throws IOException, AppBuilderServerException {
+    //     Dataset dataset = new Dataset();
 
-        String datasetId = "";
-        try {
-            datasetId = dataset.createDataset("dataset_name");
-            assertNotNull(datasetId);
-        } catch (Exception e) {
-            datasetId = System.getenv("DATASET_ID_V3");
-            dataset.setDatasetId(datasetId);
-        }
+    //     String datasetId = "";
+    //     try {
+    //         datasetId = dataset.createDataset("dataset_name");
+    //         assertNotNull(datasetId);
+    //     } catch (Exception e) {
+    //         datasetId = System.getenv("DATASET_ID_V3");
+    //         dataset.setDatasetId(datasetId);
+    //     }
 
-        String filePath = "src/test/java/com/baidubce/appbuilder/files/test.pdf";
+    //     String filePath = "src/test/java/com/baidubce/appbuilder/files/test.pdf";
 
-        String[] documentIds = dataset.addDocuments(new ArrayList<>(Collections.singletonList(filePath)), false, null, false);
-        assertNotEquals(documentIds.length, 0);
+    //     String[] documentIds = dataset.addDocuments(new ArrayList<>(Collections.singletonList(filePath)), false, null, false);
+    //     assertNotEquals(documentIds.length, 0);
 
-        DocumentListResponse resp = dataset.getDocumentList(1, 20, "");
-        assertNotEquals(resp.getResult().getTotal(), 0);
+    //     DocumentListResponse resp = dataset.getDocumentList(1, 20, "");
+    //     assertNotEquals(resp.getResult().getTotal(), 0);
 
-        dataset.deleteDocuments(documentIds);
-    }
+    //     dataset.deleteDocuments(documentIds);
+    // }
 }