如何为另一个包中的方法编写模拟代码

IT行业相对于一般传统行业，发展更新速度更快，一旦停止了学习，很快就会被行业所淘汰。所以我们需要踏踏实实的不断学习，精进自己的技术，尤其是初学者。今天golang学习网给大家整理了《如何为另一个包中的方法编写模拟代码》，聊聊，我们一起来看看吧！

我正在为 service.go 设置 service_test.go。

在service.go中，从dao.go调用方法。

所以我需要模拟这个dao方法。

但我不确定如何编写这个模拟方法。

这是存储库结构。

article
  ├ client
  ├ api
  │  ├ main.go
  │  ├ contoroller
  │  │    └ contoroller.go
  │  ├ service
  │  │    └ service.go
  │  ├ dao
  │  │    └ dao.go
  │  ├ go.mod 
  │  ├ go.sum
  │  └ dockerfile
  ├ nginx
  └ docker-compose.yml

service.go

func getarticleservice(db *sql.db) []util.article {
    var articles []util.article

    // i want to mock dao.getarticledao(db) 
    results := dao.getarticledao(db)

    article := util.article{}
    for results.next() {
        err := results.scan(&article.id, &article.uuid, &article.title, &article.content)
        if err != nil {
            panic(err.error())
        } else {
            articles = append(articles, article)
        }
    }
    return articles
}

service_test.go

func TestGetArticleService(t *testing.T) {
    //mock db
    db, _, err := sqlmock.New()

    if err != nil {
        t.Fatalf("an error '%s' was not expected when opening a stub database connection", err)
    }

    defer db.Close()

    //make expected value
    var expectedArticles []util.Article

    expectedArticle1 := util.Article{
        ID:      1,
        UUID:    "bea1b24d-0627-4ea0-aa2b-8af4c6c2a41c",
        TITLE:   "test",
        CONTENT: "test",
    }
    expectedArticles = append(expectedArticles, expectedArticle1)

    expectedArticle2 := util.Article{
        ID:      2,
        UUID:    "844bc620-7336-41a3-9cb4-552a0024ff1c",
        TITLE:   "test2",
        CONTENT: "test2",
    }
    expectedArticles = append(expectedArticles, expectedArticle2)

    //check whether expected value and GetArticleService(db) is equal.
    //Since GetArticleService(db) is executed, I want to mock dao.GetArticleDao(db)
    assert.Equal(t, expectedArticles, GetArticleService(db))
}

我想在 service.go 中模拟 dao.getarticledao(db) 以通过 service_test.go 中的测试。

但我不知道如何模拟这段代码。

解决方案

因此，要做到这一点，您需要对实际代码的定义更加抽象，这允许您替换对第三方模块的调用以进行测试。

类似下面的内容将允许您将 dao 函数替换为具有相同签名的测试模拟函数。一旦定义了硬类型或函数调用，您就永久地绑定了该函数/类型。

func GetArticleService(getArticleDao func(*sql.DB) (string), db *sql.DB) []util.Article {
    var articles []util.Article

    // this is the mock, at this point you can pass in a mock function for testing
    results := getArticleDao(db)

    article := util.Article{}
    for results.Next() {
        err := results.Scan(&article.ID, &article.UUID, &article.TITLE, &article.CONTENT)
        if err != nil {
            panic(err.Error())
        } else {
            articles = append(articles, article)
        }
    }
    return articles
}

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