導出Golang軟件包進行測試？

Question

在嘗試對使用以下結構的代碼進行測試時：

type DatabaseSt struct {
    DBName            string
    DBConnectionStr   string
    dbConnection      *sql.DB
    InterpolateParams bool

    //Archived Databases
    MinFinancialYear int
    MaxFinancialYear int
}

//DatabaseContext The context to use if the use of a database is needed.
type DatabaseContext struct {
    *Context
    Database DatabaseSt
}

我偶然發現了這篇中型文章，聲稱您可以在測試代碼中導出帶有其內部結構的Golang軟件包。 不幸的是，我不確定它們的最終含義是什么：

僅當我們運行go test時，export_test.go才被包括在內，因此它不會污染您的API，並且用戶從不訪問它們（不像Java的@VisibleForTesting），並且它搭建了一個橋，讓未導出的橋可以在math_test中訪問

更糟糕的是，它的復制無法快速實現：

/ *在這里， context是包含我想完全訪問* /的結構的包

我基本上需要能夠設置該DatabaseSt的dbConnection進行測試，而無需修改源代碼。

Answer 1

添加以下名為export_test.go的文件：

package context

func SetDbConnection(DatabaseSt *ds, db *sql.DB) {
    ds.dbConnection = db
}

從同一目錄中的其他測試文件中使用它，如下所示：

package context_test

import "context"

func FooTest(t *testing.T) {
     ...
     context.SetDbConnection(ds, db)
     ...
}

或者，在上下文包中編寫測試，以便您可以完全訪問成員：

package context

func FooTest(t *testing.T) {
     ...
     ds.dbConnection = db
     ...
}