Commit missed changes

Author: NickLovera

database/mongodb/mongodb.go

@@ -157,7 +157,7 @@ func (m *Mongo) Open(dsn string) (database.Driver, error) {
 	if err != nil {
 		return nil, err
 	}
-	client, err := mongo.Connect(context.TODO(), options.Client().ApplyURI(dsn))
+	client, err := mongo.Connect(options.Client().ApplyURI(dsn))
 	if err != nil {
 		return nil, err
 	}
@@ -263,16 +263,18 @@ func (m *Mongo) Run(migration io.Reader) error {
 }
 
 func (m *Mongo) executeCommandsWithTransaction(ctx context.Context, cmds []bson.D) error {
-	err := m.db.Client().UseSession(ctx, func(sessionContext mongo.SessionContext) error {
-		if err := sessionContext.StartTransaction(); err != nil {
+	err := m.db.Client().UseSession(ctx, func(sessionContext context.Context) error {
+		sess := mongo.SessionFromContext(ctx)
+
+		if err := sess.StartTransaction(); err != nil {
 			return &database.Error{OrigErr: err, Err: "failed to start transaction"}
 		}
 		if err := m.executeCommands(sessionContext, cmds); err != nil {
 			// When command execution is failed, it's aborting transaction
 			// If you tried to call abortTransaction, it`s return error that transaction already aborted
 			return err
 		}
-		if err := sessionContext.CommitTransaction(sessionContext); err != nil {
+		if err := sess.CommitTransaction(sessionContext); err != nil {
 			return &database.Error{OrigErr: err, Err: "failed to commit transaction"}
 		}
 		return nil

database/mongodb/mongodb_test.go

@@ -48,7 +48,7 @@ func isReady(ctx context.Context, c dktest.ContainerInfo) bool {
 		return false
 	}
 
-	client, err := mongo.Connect(ctx, options.Client().ApplyURI(mongoConnectionString(ip, port)))
+	client, err := mongo.Connect(options.Client().ApplyURI(mongoConnectionString(ip, port)))
 	if err != nil {
 		return false
 	}