在 Go 中執行 main 函數測試時出現問題

php小編西瓜在進行 Go 語言的開發時，經常會遇到在執行 main 函數進行測試時出現問題的情況。這種情況可能會導致程式無法正常運作或出現意料之外的結果。為了解決這個問題，我們需要仔細排查可能的原因，並採取相應的措施來修復程式碼。本文將介紹一些常見的問題和解決方法，幫助開發者更好地應對這類情況。

問題內容

我幾個月前編寫了這個測試，它運行成功，沒有任何錯誤，並且從未執行過init() 函數，因為我只從測試中調用了處理程序函數，但現在它失敗並出現以下錯誤：

/usr/local/opt/go/libexec/bin/go tool test2json -t /private/var/folders/wt/fc27nzn51rdbvwvjzkl05hlc0000gs/T/GoLand/___TestHandleRequest_in_next_api_cmd_getAllRawTransportOrders.test -test.v -test.paniconexit0 -test.run ^\QTestHandleRequest\E$
{"level":"fatal","msg":"no configuration found:  not set","time":"2023-08-27T22:48:03+05:30"}

Process finished with the exit code 1

我知道測試正在呼叫 init() 函數，但為什麼不更早呢。我在這裡缺少一些需要理解的東西嗎？

init.go

#

//go:build !test
// +build !test

package main

import (
    "crypto/tls"
    "encoding/json"
    log "github.com/sirupsen/logrus"
    "os"

    "lib/storage/ssm"

    "next-api/aws/dynamodb"
    "next-api/cmd/utils"
    "next-api/config"
    "next-api/http"
)

func init() {
    // Log as JSON instead of the default ASCII formatter.
    log.SetFormatter(&log.JSONFormatter{})

    // Output to stdout instead of the default stderr
    // Can be any io.Writer, see below for File example
    log.SetOutput(os.Stdout)

    // Only log the warning severity or above.
    log.SetLevel(log.InfoLevel)

    var err error
    paramsRepo := ssm.NewRepository()

    configID, err := utils.GetEnv("CONFIG_ID")
    if err != nil {
        log.Fatal("no configuration found: ", err)
    }

    cfg, err = config.Load(configID, paramsRepo)
    if err != nil {
        log.Fatal("could not load config: ", err)
    }

    urls := http.URLs{}
    json.Unmarshal([]byte(cfg.MABES.URLs), &urls)

    // configure mTLS
    cert, err := tls.X509KeyPair([]byte(cfg.MABES.Certificate), []byte(cfg.MABES.PrivateKey))
    if err != nil {
        log.Fatal("could not create MABES certificate: ", err)
    }

    MABES, err = http.NewMABESClient(urls, http.WithTLS(cert))
    if err != nil {
        log.Fatal("could not create sofaClient client.", err)
    }

    NEXT = dynamodb.NewRepository(cfg.NEXT.Table)
}

main.go

package main

import (
    "encoding/json"
    "fmt"
    "github.com/aws/aws-lambda-go/events"
    "github.com/aws/aws-lambda-go/lambda"
    log "github.com/sirupsen/logrus"
    gohttp "net/http"

    "next-api/api"
    "next-api/config"
    "next-api/errors"
    "next-api/http"
    "next-api/http/problem"
    "next-api/mabes"
    "next-api/next"
)

var (
    cfg   config.Config
    MABES mabes.Client
    NEXT  next.Repository
)

func main() {
    lambda.Start(handler)
}

func handler(request events.APIGatewayProxyRequest) (events.APIGatewayProxyResponse, error) {
    facilityID := request.PathParameters["facilityID"]
    if facilityID == "" {
        return events.APIGatewayProxyResponse{
            StatusCode: gohttp.StatusBadRequest,
            Headers:    http.Headers("application/problem+json"),
            Body:       problem.ServerError(fmt.Errorf("missing facilityID parameter")).ToJSONResponse(),
        }, nil
    }

    transportOrders, err := api.GetAllRawTransportOrders(facilityID, request.QueryStringParameters, MABES, NEXT)
    b, err := json.Marshal(transportOrders)
    if err != nil {
        log.Errorln(err)

        if errors.Is(errors.NEXTRead, err) {
            return events.APIGatewayProxyResponse{
                StatusCode: gohttp.StatusOK,
                Headers:    http.Headers("application/json"),
                Body:       fmt.Sprintf(`{"data":%s}`, string(b)),
            }, nil
        }

        return events.APIGatewayProxyResponse{
            StatusCode: gohttp.StatusInternalServerError,
            Headers:    http.Headers("application/problem+json"),
            Body:       problem.ServerError(err).ToJSONResponse(),
        }, nil
    }

    return events.APIGatewayProxyResponse{
        StatusCode: gohttp.StatusOK,
        Headers:    http.Headers("application/json"),
        Body:       fmt.Sprintf(`{"data":%s}`, string(b)),
    }, nil
}

main_test.go

package main

import (
    "github.com/aws/aws-lambda-go/events"
    "github.com/stretchr/testify/assert"
    "net/http"
    "strings"
    "testing"
    "time"

    "next-api/api"
    "next-api/mabes"
    "next-api/next"
)

func TestHandleRequest(t *testing.T) {
    t.Run("should return the TransportOrders", func(t *testing.T) {
        req := events.APIGatewayProxyRequest{
            PathParameters: map[string]string{
                "facility": "11",
            },
        }
        timeStamp, _ := time.Parse(time.RFC3339, time.Now().Format(time.RFC3339))
        sofa1 := next.SofaTransferModel{}
        sofa1.Phase = "1"
        sofa1.PartKey.PartCode = "abcd"
        sofa1.PartKey.DispatcherFacility = "11"
        sofa1.PartKey.UsageCode = "01"
        sofa1.Storage.Number = "00315"
        sofa1.Shipment.RequestedArrivalDate = timeStamp.String()
        transportOrders := make([]next.SofaTransferModel, 0)
        transportOrders = append(transportOrders, sofa1)

        getAllRawTransportOrdersCalls := 0
        api.GetAllRawTransportOrders = func(facilityID string, params map[string]string, MABES mabes.Client, NEXT next.Repository) (next.SofaTransferObjects, error) {
            getAllRawTransportOrdersCalls++
            return transportOrders, nil
        }

        // when
        res, err := handler(req)
        // then
        assert.Equal(t, 1, getAllRawTransportOrdersCalls)

        assert.NoError(t, err)
        assert.Equal(t, http.StatusOK, res.StatusCode)

        assert.True(t, strings.Contains(res.Body, transportOrders[0].PartKey.PartCode))
        assert.True(t, strings.Contains(res.Body, transportOrders[0].Storage.Number))
    })

}

解決方法

名為init（全部小寫）的頂級函數總是，而總是在測試之前執行。 init 現在失敗的原因，即使以前沒有，現在看來，是環境的變化；即在CONFIG_ID 設定之前，但現在沒有設置，至少在執行測試的shell 中沒有設定.

https://www.php.cn/link/905d8fc4ffb4275a428a84589810f8f4

如果您使用-tags test 執行go test 命令，則將遵守init.go 檔案頂部的建置約束，即// go:build !test，並且該檔案將從測試中省略建造。但預設情況下，沒有「test」建構標籤，因此如果沒有-tags test 參數，這些建置約束將被忽略，並且init.go 檔案將包含在測試建置中。 p>

以上是在 Go 中執行 main 函數測試時出現問題的詳細內容。更多資訊請關注PHP中文網其他相關文章！