wip

FerretDB · Dec 18, 2023 · 6624561 · 6624561
1 parent 8c43dd6
commit 6624561
Show file tree

Hide file tree

Showing 4 changed files with 275 additions and 95 deletions.
diff --git a/integration/users/drop_user_test.go b/integration/users/drop_user_test.go
@@ -0,0 +1,124 @@
+// 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"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+
+	"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"
+)
+
+func TestDropUser(t *testing.T) {
+	t.Parallel()
+
+	ctx, collection := setup.Setup(t)
+	db := collection.Database()
+	client := db.Client()
+	users := client.Database("admin").Collection("system.users")
+
+	// TODO https://github.com/FerretDB/FerretDB/issues/1492
+	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)
+	}
+
+	err := db.RunCommand(ctx, bson.D{
+		{"createUser", "a_user"},
+		{"roles", bson.A{}},
+		{"pwd", "password"},
+	}).Err()
+	assert.NoError(t, err)
+
+	testCases := map[string]struct { //nolint:vet // for readability
+		payload    bson.D
+		err        *mongo.CommandError
+		altMessage string
+		expected   bson.D
+		skip       string
+	}{
+		"NotFound": {
+			payload: bson.D{
+				{"dropUser", "not_found_user"},
+			},
+			err: &mongo.CommandError{
+				Code:    11,
+				Name:    "UserNotFound",
+				Message: "User 'not_found_user@TestDropUser' not found",
+			},
+		},
+		"Success": {
+			payload: bson.D{
+				{"dropUser", "a_user"},
+			},
+			expected: bson.D{
+				{
+					"ok", float64(1),
+				},
+			},
+		},
+	}
+
+	for name, tc := range testCases {
+		name, tc := name, tc
+		t.Run(name, func(t *testing.T) {
+			if tc.skip != "" {
+				t.Skip(tc.skip)
+			}
+
+			t.Parallel()
+
+			var res bson.D
+			err := db.RunCommand(ctx, tc.payload).Decode(&res)
+			if tc.err != nil {
+				integration.AssertEqualAltCommandError(t, *tc.err, tc.altMessage, err)
+				return
+			}
+
+			require.NoError(t, err)
+
+			actual := integration.ConvertDocument(t, res)
+			actual.Remove("$clusterTime")
+			actual.Remove("operationTime")
+
+			expected := integration.ConvertDocument(t, tc.expected)
+			testutil.AssertEqual(t, expected, actual)
+
+			payload := integration.ConvertDocument(t, tc.payload)
+			assertUserNotFound(ctx, t, users, db.Name(), payload)
+		})
+	}
+}
+
+// assertUserNotFound checks it the user doesn't exist in the admin.system.users collection.
+func assertUserNotFound(ctx context.Context, t testing.TB, users *mongo.Collection, dbName string, payload *types.Document) {
+	t.Helper()
+	err := users.FindOne(ctx, bson.D{{"user", must.NotFail(payload.Get("dropUser"))}}).Err()
+	require.Equal(t, mongo.ErrNoDocuments, err, `should return "no documents" error`)
+}
diff --git a/internal/handler/handlererrors/error.go b/internal/handler/handlererrors/error.go
@@ -39,6 +39,9 @@ const (
 	// ErrFailedToParse indicates user input parsing failure.
 	ErrFailedToParse = ErrorCode(9) // FailedToParse
 
+	// ErrUserNotFound indicates an user was not found for the accessed database.
+	ErrUserNotFound = ErrorCode(11) // UserNotFound
+
 	// ErrUnauthorized indicates that cursor is not authorized to access another namespace.
 	ErrUnauthorized = ErrorCode(13) // Unauthorized
 

diff --git a/internal/handler/handlererrors/errorcode_string.go b/internal/handler/handlererrors/errorcode_string.go