From 89c80192218de328cb9b77c6607fea143e65d4b1 Mon Sep 17 00:00:00 2001
From: shyam12860 <shyam.subramaniyam@mongodb.com>
Date: Wed, 23 Apr 2025 14:18:59 -0500
Subject: [PATCH] dumprestore1.js using github copilot

dumprestore1.js is quite simple so it did well.
Split various test cases into separate functions.

Ideally TestDumpRestore1 should not use the mongodump class directly,
but it's still testing the thing we'd expect.

Used Claude 3.7 for this.
---
 test/golang/dumprestore1_test.go | 94 ++++++++++++++++++++++++++++++++
 1 file changed, 94 insertions(+)
 create mode 100644 test/golang/dumprestore1_test.go

diff --git a/test/golang/dumprestore1_test.go b/test/golang/dumprestore1_test.go
new file mode 100644
index 000000000..6cd21de9a
--- /dev/null
+++ b/test/golang/dumprestore1_test.go
@@ -0,0 +1,94 @@
+package legacy_test
+
+import (
+	"context"
+	"os/exec"
+	"testing"
+
+	"github.com/mongodb/mongo-tools/common/testtype"
+	"github.com/mongodb/mongo-tools/common/testutil"
+	"github.com/stretchr/testify/require"
+	"go.mongodb.org/mongo-driver/bson"
+)
+
+func TestDumpRestore1(t *testing.T) {
+	testtype.SkipUnlessTestType(t, testtype.IntegrationTestType)
+	ctx := context.Background()
+
+	session, err := testutil.GetBareSession()
+	require.NoError(t, err, "should connect to server")
+
+	// Setup test database
+	dbName := "foo"
+	collName := "foo"
+	db := session.Database(dbName)
+	coll := db.Collection(collName)
+
+	// Ensure the collection is empty
+	err = coll.Drop(ctx)
+	require.NoError(t, err)
+
+	// Insert one document
+	_, err = coll.InsertOne(ctx, bson.M{"a": 22})
+	require.NoError(t, err)
+
+	// Verify one document inserted
+	count, err := coll.CountDocuments(ctx, bson.M{})
+	require.NoError(t, err)
+	require.Equal(t, int64(1), count)
+
+	// Create temp dir for dump files
+	dumpDir, cleanup := testutil.MakeTempDir(t)
+	defer cleanup()
+
+	// Run mongodump
+	args := append(testutil.GetBareArgs(), "--out", dumpDir)
+	cmd := exec.Command("mongodump", args...)
+	output, err := cmd.CombinedOutput()
+	require.NoError(t, err, "mongodump failed: %s", output)
+
+	// Drop collection to verify restore works
+	err = coll.Drop(ctx)
+	require.NoError(t, err)
+
+	// Verify collection is empty
+	count, err = coll.CountDocuments(ctx, bson.M{})
+	require.NoError(t, err)
+	require.Equal(t, int64(0), count)
+
+	// Run mongorestore
+	args = append([]string{"mongorestore"}, append(testutil.GetBareArgs(), "--dir", dumpDir)...)
+	cmd = exec.Command(args[0], args[1:]...)
+	output, err = cmd.CombinedOutput()
+	require.NoError(t, err, "mongorestore failed: %s", output)
+
+	// Verify document was restored
+	var doc bson.M
+	err = coll.FindOne(ctx, bson.M{}).Decode(&doc)
+	require.NoError(t, err, "should find restored document")
+	require.Equal(t, int32(22), doc["a"])
+}
+
+// Test that mongodump returns failure when --collection is used without --db
+
+func TestMongodumpFailsWithoutDB(t *testing.T) {
+	args := append(testutil.GetBareArgs(), "--collection", "col")
+	cmd := exec.Command("mongodump", args...)
+	_, err := cmd.CombinedOutput()
+	require.Error(t, err, "mongodump should fail when --collection is used without --db")
+}
+
+// Ensure that --db and --collection are provided when filename is "-" (stdin)
+func TestMongorestoreFailsWithoutDBForStdin(t *testing.T) {
+	args := []string{"mongorestore", "--collection", "coll", "--dir", "-"}
+	cmd := exec.Command(args[0], args[1:]...)
+	_, err := cmd.CombinedOutput()
+	require.Error(t, err, "mongorestore should fail when --collection is provided without --db for stdin")
+}
+
+func TestMongorestoreFailsWithoutCollectionForStdin(t *testing.T) {
+	args := []string{"mongorestore", "--db", "db", "--dir", "-"}
+	cmd := exec.Command(args[0], args[1:]...)
+	_, err := cmd.CombinedOutput()
+	require.Error(t, err, "mongorestore should fail when --db is provided without --collection for stdin")
+}