Fix test fail

diff --git a/Reply/F_test.go b/Reply/F_test.go
index 666076774ca9497b15de7bc13614f7e8a5603146..cf34800268c750353224833e7f5966796cf38dda 100644 (file)
--- a/Reply/F_test.go
+++ b/Reply/F_test.go
@@ -4,12 +4,11 @@ import (
        "context"
        "database/sql"
        "errors"
-       "fmt"
        "testing"
 
        c "github.com/qydysky/bili_danmu/CV"
        file "github.com/qydysky/part/file"
-       psql "github.com/qydysky/part/sqlite"
+       psql "github.com/qydysky/part/sql"
 )
 
 func TestSaveDanmuToDB(t *testing.T) {
@@ -35,21 +34,18 @@ func TestSaveDanmuToDB(t *testing.T) {
        } else {
                if e := psql.BeginTx[any](db, context.Background(), &sql.TxOptions{}).Do(psql.SqlFunc[any]{
                        Ty:    psql.Queryf,
-                       Query: "select msg from danmu",
+                       Query: "select msg as Msg from danmu",
                        AfterQF: func(_ *any, rows *sql.Rows, txE error) (_ *any, stopErr error) {
-                               var count = 0
-                               for rows.Next() {
-                                       count += 1
-                                       var msg string
-                                       if e := rows.Scan(&msg); e != nil {
-                                               return nil, e
-                                       }
-                                       if msg != "可能走位配合了他的压枪" {
-                                               return nil, errors.New("no msg")
-                                       }
+                               type row struct {
+                                       Msg string
                                }
-                               if count != 1 {
-                                       return nil, fmt.Errorf("no count %d", count)
+
+                               v, err := psql.DealRows(rows, func() row { return row{} })
+                               if err != nil {
+                                       return nil, err
+                               }
+                               if len(v) != 1 || v[0].Msg != "可能走位配合了他的压枪" {
+                                       return nil, errors.New("no msg")
                                }
                                return nil, nil
                        },