完成update测试

dpsv1/client.go

@@ -92,6 +92,8 @@ func (c *Client) Insert(req *ports.InsertRequest) (string, error) {
 	return reply.Statement, nil
 }
 
+// TODO InsertBatch
+
 func (c *Client) Delete(req *ports.DeleteRequest) (string, error) {
 	reply, err := c.commandServiceClient.Delete(context.Background(), &request.DeleteRequest{
 		DatabaseID:            req.DatabaseID,

test/v1/v1_test.go

@@ -72,3 +72,64 @@ func TestInsert(t *testing.T) {
 		assertEqual(now.UnixMilli(), resultMap["time"].(time.Time).Local().UnixMilli(), "时间不一致").
 		assertEqual(tableNum, resultMap["table_num"], "表数量不一致")
 }
+
+func TestUpdate(t *testing.T) {
+	initClient(t, "localhost:30170")
+	defer destroyClient(t)
+
+	tablePrefix := "test." + simpleUUID()[0:8]
+
+	id := simpleUUID()
+	name := simpleUUID()
+	now := time.Now().Local()
+	tableNum := rand.New(rand.NewSource(now.Unix())).Intn(10)
+	newName := simpleUUID()
+	newNow := time.Now().Local()
+	newTableNum := rand.New(rand.NewSource(now.Unix())).Intn(10)
+
+	resultMap := make(map[string]any)
+
+	newToolKit(t).
+		autoMigrate(&ports.AutoMigrateRequest{
+			DatabaseID:            "2b78141779ee432295ca371b91c5cac7",
+			TablePrefixWithSchema: tablePrefix,
+			Version:               "v1",
+			TableModelDescribe:    tableModelDescribe,
+		}).
+		insert(&ports.InsertRequest{
+			DatabaseID:            "2b78141779ee432295ca371b91c5cac7",
+			TablePrefixWithSchema: tablePrefix,
+			Version:               "v1",
+			KeyColumns:            []string{"id"},
+			TableRow: map[string]any{
+				"id":        id,
+				"name":      name,
+				"time":      now,
+				"table_num": tableNum,
+			},
+			UserID: "test",
+		}).
+		update(&ports.UpdateRequest{
+			DatabaseID:            "2b78141779ee432295ca371b91c5cac7",
+			TablePrefixWithSchema: tablePrefix,
+			Version:               "v1",
+			KeyValues:             map[string]string{"id": id},
+			NewTableRow: map[string]any{
+				"id":        id,
+				"name":      newName,
+				"time":      newNow,
+				"table_num": newTableNum,
+			},
+			UserID: "test",
+		}).
+		queryByKeys(&ports.QueryByKeysRequest{
+			DatabaseID:            "2b78141779ee432295ca371b91c5cac7",
+			TablePrefixWithSchema: tablePrefix,
+			Version:               "v1",
+			KeyValues:             map[string]string{"id": id},
+		}, &resultMap).
+		assertEqual(id, resultMap["id"], "ID不一致").
+		assertEqual(newName, resultMap["name"], "名称不一致").
+		assertEqual(newNow.UnixMilli(), resultMap["time"].(time.Time).Local().UnixMilli(), "时间不一致").
+		assertEqual(newTableNum, resultMap["table_num"], "表数量不一致")
+}