wip

FerretDB · henvic · Jan 4, 2024 · Dec 12, 2023 · Dec 12, 2023 · Dec 14, 2023
commit ac21f142fe0085cf330867f30bb254d273af1b95
@@ -0,0 +1,83 @@
+// Copyright 2021 FerretDB Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package users
+
+import (
+	"context"
+	"fmt"
+	"testing"
+
+	"github.com/FerretDB/FerretDB/integration"
+	"github.com/FerretDB/FerretDB/integration/setup"
+	"github.com/FerretDB/FerretDB/internal/types"
+	"github.com/FerretDB/FerretDB/internal/util/must"
+	"github.com/FerretDB/FerretDB/internal/util/testutil"
+	"github.com/stretchr/testify/assert"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+)
+
+func TestDropAllUsersFromDatabase(t *testing.T) {
+	t.Parallel()
+
+	ctx, collection := setup.Setup(t)
+	db := collection.Database()
+	client := collection.Database().Client()
+	users := client.Database("admin").Collection("system.users")
+
+	if setup.IsMongoDB(t) {
+		assert.NoError(t, collection.Database().RunCommand(ctx, bson.D{
+			{"dropAllUsersFromDatabase", 1},
+		}).Err())
+	} else {
+		// Erase any previously saved user in the database.
+		_, err := users.DeleteMany(ctx, bson.D{{"db", db.Name()}})
+		assert.NoError(t, err)
+	}
+
+	quantity := 5
+	for i := 1; i <= quantity; i++ {
+		err := db.RunCommand(ctx, bson.D{
+			{"createUser", fmt.Sprintf("user_%d", i)},
+			{"roles", bson.A{}},
+			{"pwd", "password"},
+		}).Err()
+		assert.NoError(t, err)
+	}
+
+	assertDropAllUsersFromDatabase(t, ctx, db, users, quantity)
+
+	// FIXME: calling assertDropAllUsersFromDatabase a second time with quantity = 0
+	// for FerretDB is triggering a "socket was unexpectedly closed: EOF" error for some reason.
+	// assertDropAllUsersFromDatabase(t, ctx, db, users, 0)
+}
+
+func assertDropAllUsersFromDatabase(t *testing.T, ctx context.Context, db *mongo.Database, users *mongo.Collection, quantity int) {
+	var res bson.D
+	err := db.RunCommand(ctx, bson.D{
+		{"dropAllUsersFromDatabase", 1},
+	}).Decode(&res)
+
+	assert.NoError(t, err)
+
+	actual := integration.ConvertDocument(t, res)
+	actual.Remove("$clusterTime")
+	actual.Remove("operationTime")
+
+	expected := must.NotFail(types.NewDocument("n", int32(quantity), "ok", float64(1)))
+	testutil.AssertEqual(t, expected, actual)
+
+	assert.Equal(t, mongo.ErrNoDocuments, users.FindOne(ctx, bson.D{{"db", db.Name()}}).Err())
+}
@@ -16,9 +16,14 @@
 
 import (
 	"context"
+	"errors"
 
+	"github.com/FerretDB/FerretDB/internal/backends"
 	"github.com/FerretDB/FerretDB/internal/handler/common"
+	"github.com/FerretDB/FerretDB/internal/types"
+	"github.com/FerretDB/FerretDB/internal/util/iterator"
 	"github.com/FerretDB/FerretDB/internal/util/lazyerrors"
+	"github.com/FerretDB/FerretDB/internal/util/must"
 	"github.com/FerretDB/FerretDB/internal/wire"
 )
 
@@ -29,6 +34,56 @@
 		return nil, lazyerrors.Error(err)
 	}
 
-	// TODO https://github.com/FerretDB/FerretDB/issues/1492
-	return nil, common.Unimplemented(document, document.Command())
+	// NOTE: In MongoDB, the comment field isn't saved in the database, but used for log and profiling.
+	common.Ignored(document, h.L, "writeConcern", "comment")
+
+	dbName, err := common.GetRequiredParam[string](document, "$db")
+	if err != nil {
+		return nil, err
+	}
+
+	// Users are saved in the "admin" database.
+	adminDB, err := h.b.Database("admin")
+	if err != nil {
+		return nil, lazyerrors.Error(err)
+	}
+
+	users, err := adminDB.Collection("system.users")
+	if err != nil {
+		return nil, lazyerrors.Error(err)
+	}
+
+	qr, err := users.Query(ctx, &backends.QueryParams{
+		Filter: must.NotFail(types.NewDocument("db", dbName)),
+	})
+	if err != nil {
+		return nil, lazyerrors.Error(err)
+	}
+
+	var ids []any
+	defer qr.Iter.Close()
+	for {
+		_, v, err := qr.Iter.Next()
+		if errors.Is(err, iterator.ErrIteratorDone) {
+			break
+		}
+		ids = append(ids, must.NotFail(v.Get("_id")))
+	}
+
+	res, err := users.DeleteAll(ctx, &backends.DeleteAllParams{
+		IDs: ids,
+	})
+	if err != nil {
+		return nil, lazyerrors.Error(err)
+	}
+
+	var reply wire.OpMsg
+	must.NoError(reply.SetSections(wire.OpMsgSection{
+		Documents: []*types.Document{must.NotFail(types.NewDocument(
+			"n", res.Deleted,
+			"ok", float64(1),
+		))},
+	}))
+
+	return &reply, nil
 }