如何在 Go 中为允许单元测试的休息服务设置 Mongo DB？

Question

我正在使用 MongoDB 和 mux 路由器设置休息服务。我遇到了关于如何最好地设置它以允许在单独的数据库中进行单元/集成测试的问题。

我曾尝试在 Init() 函数中设置数据库，但这在尝试使用测试数据库设置单元测试时给我带来了问题。

以下是我现在所处位置的一些示例。我试图在发布之前使用一种方法连接到数据库，因此在我的测试中我可以连接到不同的测试数据库。

type user struct {

    name string `json:"name"`

    age  int    `json:"age"`

}

type database struct {

    db *mongo.Database

}

func ConnectToDB() (*database, error) {

    client, err := mongo.NewClient(options.Client().ApplyURI("mongodb://localhost:27017"))

    if err != nil {

        return nil, err

    }

    if err := client.Connect(context.Background()); err != nil {

        return nil, err

    }

    database := &database{

        db: client.Database("PMBaseGo"),

    }

    return database, nil

}

func PostUser(w http.ResponseWriter, r *http.Request) {

    w.Header().Set("Content-Type", "application/json")

    //Retrieving request body

    var user user

    _ = json.NewDecoder(r.Body).Decode(&user)

    //Posting Company.

    err := PostUserToDB(user)

    //Error Handling

    if err != nil {

        fmt.Println(err)

        w.WriteHeader(500)

    }

}

func (d *database) connPostUserToDB(user user) error {

    _, err := d.db.Collection("companies").InsertOne(context.Background(), user)

    if err != nil {

        return err

    }

    return nil

}

func main() {

    _, _ = ConnectToDB()

    r := mux.NewRouter()

    r.HandleFunc("/user", PostUser).Methods("POST")

    fmt.Println("Application Running...")

    log.Fatal(http.ListenAndServe("localhost:8081", r))

}

我现在遇到的问题是试图调用函数PostUserToDB中的方法PostUser。

我开始认为问题在于我将如何连接到数据库。

Answer 1

使您的处理程序成为包含处理程序依赖项的结构上的方法：type server struct {    srv *service}type service struct {    db *mongo.Database}func (s *server) PostUser(w http.ResponseWriter, r *http.Request) {    // Create user struct from request data...    if err := s.srv.CreateUser(u); err != nil {        // Error response.    }    // Success response.}func (s *service) CreateUser(u *user) error {    // ...    _, err := d.db.Collection("foo").InsertOne(r.Context(), user)    return err}func main() {    s := &server{        srv: &service{            db: getMongoDatabase(),        },    }    r := mux.NewRouter()    r.HandleFunc("/user", s.PostUser).Methods("POST")    log.Fatal(http.ListenAndServe("localhost:8081", r))}在您的测试代码中，您可以简单地注入一个不同的数据库，让您的测试在该数据库上运行。