Bagaimana untuk menghantar objek protobuf sebagai baris ke sqlmock.AddRow dalam golang?-Golang-php.cn

Bagaimana untuk menghantar objek protobuf sebagai baris ke sqlmock.AddRow dalam golang?

PHPz

Lepaskan： 2024-02-11 11:03:07

ke hadapan

482 orang telah melayarinya

如何将 protobuf 对象作为一行传递给 golang 中的 sqlmock.AddRow？

php小编小新在这里为大家介绍一个问题：如何将 protobuf 对象作为一行传递给 golang 中的 sqlmock.AddRow？在使用 golang 开发中，我们经常会遇到需要将 protobuf 对象转换为数据库行的情况。而 sqlmock 是一个常用的 mock 数据库操作的工具，但是如何将 protobuf 对象传递给 sqlmock 的 AddRow 方法却是一个需要解决的问题。接下来，我们将详细介绍如何解决这个问题，让大家能够更好地应用 protobuf 和 sqlmock。

问题内容

我正在尝试使用 sqlmock 对我的 go 代码进行单元测试。这是我试图测试的原始代码。

func enrollcourse(db *gorm.db, user_id string ,course_id string) error {
    user := &usermodels.user{}
    ref := db.first(user, "uuid = ?", user_id)
    userpb := user.protobuf()
    fmt.printf("user name %+v", user.name)
    ....
}

Salin selepas log masuk

这是我的单元测试

func TestEnrollCourse(t *testing.T){
    db, mock, err := sqlmock.New()
    if err != nil {
        t.Fatalf("an error '%s' was not expected when opening a stub database connection", err)
    }
    defer db.Close()

    rows := sqlmock.NewRows([]string{"user_id","user_name"}).
        AddRow(1, "John")
    mock.ExpectQuery(regexp.QuoteMeta(`SELECT * FROM "users" WHERE uuid = $1`)).WithArgs("user-fd3746c8-d32f-4fb8-8f6a-b6d72dcf2969").WillReturnRows(rows)
    
    gdb, err := gorm.Open(postgres.New(postgres.Config{
                       Conn: db,
               }), &gorm.Config{})

    enrollCourse(gdb, "user-fd3746c8-d32f-4fb8-8f6a-b6d72dcf2969", "english_course")
    ....
}

Salin selepas log masuk

我期待 fmt.printf("user name %+v", user.name) 打印用户名，但它为零。如何正确地将 protobuf 对象传递给 addrow？

解决方法

您用于 AddRow 的数据类型似乎是字符串。在 sqlmock 中使用 protobuf 的最佳方法是借助存根文件中创建的构造函数来创建对象，并将其以字符串形式序列化并存储。然后在实际方法（enrollCourse）中，您可以获取此序列化消息并将 unmarshal 转换为适当的模型。

Atas ialah kandungan terperinci Bagaimana untuk menghantar objek protobuf sebagai baris ke sqlmock.AddRow dalam golang?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!