主页 / coding / 问题 / 79240512
Accepted
ns.altun
Asked: 2024-12-01 05:15:58 +0800 CST

如何在 golang 中模拟 mongodb 驱动程序

  • 772

在我的 Golang 项目中，我需要模拟 mongodb 才能为我的 Repository 函数编写单元测试。我主要需要模拟collection.Indexes.CreateMany或FindOneAndUpdate或Find或cursor.All或InsertOne等。据我所知，使用 mtest lib 是可能的。但我无法模拟创建索引。我需要一些关于如何使用mtest（或是否有更好的选择）在 golang 中模拟 mongo 驱动程序的示例（尤其是collection.FindOneAndUpdate和collection.Indexes.CreateMany）的帮助

mongodb

1 个回答

  1. Best Answer

    假设你有MyRepositoryFunction这样的：

    package mypackage

import (
    "context"
    "fmt"

    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/mongo"
)

func MyRepositoryFunction(ctx context.Context, collection *mongo.Collection) error {
    models := []mongo.IndexModel{
        {Keys: bson.D{{"surname", 1}, {"firstName", 1}}},
        {Keys: bson.D{{"key", -1}}},
    }
    _, err := collection.Indexes().CreateMany(ctx, models)
    if err != nil {
        return fmt.Errorf("failed to created indexes: %w", err)
    }

    filter := bson.D{{"_id", "custom123"}}
    update := bson.D{{"$set", bson.D{
        {"key", 42},
    }}}
    result := collection.FindOneAndUpdate(ctx, filter, update)
    if result.Err() != nil {
        return fmt.Errorf("failed to created indexes: %w", result.Err())
    }

    return nil
}

    你可以mtest 这样使用：

    package mypackage

import (
    "context"
    "testing"

    "github.com/stretchr/testify/assert"
    "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/mongo/integration/mtest"
)

func TestMyRepositoryFunction(t *testing.T) {
    ctx := context.Background()
    mt := mtest.New(t, mtest.NewOptions().ClientType(mtest.Mock))

    mt.Run("MyRepositoryFunction", func(mt *mtest.T) {
        mt.AddMockResponses(
            mtest.CreateSuccessResponse(),
            mtest.CreateSuccessResponse(bson.E{
                "value", bson.M{"_id": "custom123", "key": 24},
            }),
        )

        err := MyRepositoryFunction(ctx, mt.Coll)

        assert.NoError(t, err, "Should have successfully run")
    })
}

    您必须知道函数的调用顺序，并以有线格式提供足够的结果以使客户端满意。此外，您实际上并没有模拟驱动程序 - 这样您将模拟服务器。这可能有点困难，因为您必须了解协议。

    • 1

相关问题