在 golang 中如何进行映射对象的测试

哈喽！今天心血来潮给大家带来了《在 golang 中如何进行映射对象的测试》，想必大家应该对Golang都不陌生吧，那么阅读本文就都不会很困难，以下内容主要涉及到，若是你正在学习Golang，千万别错过这篇文章~希望能帮助到你！

我尝试在我的 go 代码中实现一些单元测试，发现模拟方法的主题相当困难。 我有以下示例，希望您能帮助我:)

在第一层我有以下代码：

package api

import (
    "fmt"
    "core"
)

type createuserdto struct {
    id string
}

func apimethod() {
    fmt.println("some incoming api call wit user")
    incominguserdata := &createuserdto{id: "testid"}
    mapeduser := incominguserdata.mapuser()
    mapeduser.create()
}

func (createuserdto *createuserdto) mapuser() core.user {
    return &core.usercore{id: createuserdto.id}
}

第二层代码如下：

package core

import (
    "fmt"
)

type CoreUser struct{ Id string }

type User interface {
    Create()
}

func (user CoreUser) Create() {
    fmt.Println("Do Stuff")
}

我现在的问题是，如何在不测试核心包的情况下测试 api 包中的每个公共方法。特别是create()方法。

正确答案

根据评论，我整理了一个简单的 github 存储库，以展示我通常如何处理 go 中的结构化项目。存储库目前不考虑测试部分，但使用此项目结构插入它们应该很容易。
让我们从常规文件夹的布局开始：

• controllers
• services
• db
• dto
• models

现在，让我们看看相关文件。

models/user.go

package models

import "gorm.io/gorm"

type user struct {
    *gorm.model
    id string `gorm:"primarykey"`
}

func newuser(id string) *user {
    return &user{id: id}
}

这里有简单的结构定义。

dto/user.go

package dto

import "time"

type userdto struct {
    id      string    `json:"id"`
    addedon time.time `json:"added_on"`
}

func newuserdto(id string) *userdto {
    return &userdto{id: id}
}

我们使用虚拟 addedon 字段来丰富模型结构，该字段仅用于演示目的。

db/user.go

package db

import (
    "gorm.io/gorm"

    "userapp/models"
)

type userdb struct {
    conn *gorm.db
}

type userdbinterface interface {
    saveuser(user *models.user) error
}

func (u *userdb) saveuser(user *models.user) error {
    if dbtrn := u.conn.create(user); dbtrn.error != nil {
        return dbtrn.error
    }
    return nil
}

在这里，我们定义了一个使用 user 存储库的接口。在我们的测试中，我们可以提供一个模拟而不是 userdb 结构的实例。

services/user.go

package services

import (
    "time"

    "userapp/db"
    "userapp/dto"
    "userapp/models"
)

type userservice struct {
    db db.userdbinterface
}

type userserviceinterface interface {
    adduser(inputreq *dto.userdto) (*dto.userdto, error)
}

func newuserservice(db db.userdbinterface) *userservice {
    return &userservice{
        db: db,
    }
}

func (u *userservice) adduser(inputreq *dto.userdto) (*dto.userdto, error) {
    // here you can write complex logic
    user := models.newuser(inputreq.id)

    // invoke db repo
    if err := u.db.saveuser(user); err != nil {
        return nil, err
    }

    inputreq.addedon = time.now()

    return inputreq, nil
}

这是桥接表示层和底层存储库之间连接的层。

controllers/user.go

package controllers

import (
    "encoding/json"
    "io"
    "net/http"

    "userapp/dto"
    "userapp/services"
)

type UserController struct {
    US services.UserServiceInterface
}

func NewUserController(userService services.UserServiceInterface) *UserController {
    return &UserController{
        US: userService,
    }
}

func (u *UserController) Save(w http.ResponseWriter, r *http.Request) {
    reqBody, err := io.ReadAll(r.Body)
    if err != nil {
        panic(err)
    }
    r.Body.Close()

    var userReq dto.UserDTO

    json.Unmarshal(reqBody, &userReq)

    userRes, err := u.US.AddUser(&userReq)
    if err != nil {
        w.WriteHeader(http.StatusInternalServerError)
        json.NewEncoder(w).Encode(err)
        return
    }

    w.WriteHeader(http.StatusCreated)
    json.NewEncoder(w).Encode(userRes)
}

在这里，我们定义了（通过依赖注入）使用 userservice 结构的控制器。

您可以在我的存储库 GitHub 中找到所有内容 如果它澄清了一点，请告诉我。

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~