Maison > développement back-end > Golang > Comment transmettre l'objet protobuf sous forme de ligne à sqlmock.AddRow dans Golang ?

Comment transmettre l'objet protobuf sous forme de ligne à sqlmock.AddRow dans Golang ?

PHPz
Libérer: 2024-02-11 11:03:07
avant
559 Les gens l'ont consulté

如何将 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)
    ....
}
Copier après la connexion

这是我的单元测试

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")
    ....
}
Copier après la connexion

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

解决方法

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

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!

Déclaration de ce site Web
Le contenu de cet article est volontairement contribué par les internautes et les droits d'auteur appartiennent à l'auteur original. Ce site n'assume aucune responsabilité légale correspondante. Si vous trouvez un contenu suspecté de plagiat ou de contrefaçon, veuillez contacter admin@php.cn
Tutoriels populaires
Plus>
Derniers téléchargements
Plus>
effets Web
Code source du site Web
Matériel du site Web
Modèle frontal