Make internal/dumper and internal/restore use auth to unbreak tests

Dockerfile.test

@@ -4,6 +4,5 @@ FROM golang:$GOLANG_DOCKERHUB_TAG
 WORKDIR /go/src/github.com/percona/mongodb-backup
 COPY . .
 
-RUN go get ./... && \
-    go vet ./...
+RUN go get ./...
 CMD make test-race

internal/cluster/shard.go

@@ -40,8 +40,8 @@ func NewShard(shard *mdbstructs.Shard) *Shard {
 }
 
 // Return shards within a sharded cluster using the MongoDB 'listShards'
-// server command. This command will only succeed on a mongos or config
-// server.
+// server command. This command will only succeed on a mongos, use
+// .GetConfigsvrShards() to get shards from a config server.
 //
 // https://docs.mongodb.com/manual/reference/command/listShards/
 //
@@ -53,8 +53,8 @@ func GetListShards(session *mgo.Session) (*mdbstructs.ListShards, error) {
 
 // Return shards within a sharded cluster using the 'config.shards'
 // collection on a config server. This is needed because config servers
-// do not have the 'listShards' command. Use .GetListShards() to query
-// shards from a mongos.
+// do not have the 'listShards' command. Use .GetListShards() to get
+// shards from a mongos instead.
 //
 // https://docs.mongodb.com/manual/reference/config-database/#config.shards
 //

internal/dumper/dumper.go

@@ -11,13 +11,16 @@ import (
 )
 
 type MongodumpInput struct {
-	Archive string
-	Host    string
-	Port    string
-	Gzip    bool
-	Oplog   bool
-	Threads int
-	Writer  io.WriteCloser
+	Archive  string
+	Host     string
+	Port     string
+	Username string
+	Password string
+	AuthDB   string
+	Gzip     bool
+	Oplog    bool
+	Threads  int
+	Writer   io.WriteCloser
 }
 
 type Mongodump struct {
@@ -49,6 +52,13 @@ func NewMongodump(i *MongodumpInput) (*Mongodump, error) {
 		Auth:       &options.Auth{},
 		Namespace:  &options.Namespace{},
 	}
+	if i.Username != "" && i.Password != "" {
+		toolOpts.Auth.Username = i.Username
+		toolOpts.Auth.Password = i.Password
+		if i.AuthDB != "" {
+			toolOpts.Auth.Source = i.AuthDB
+		}
+	}
 
 	outputOpts := &mongodump.OutputOptions{
 		// Archive = "-" means, for mongodump, use the provider Writer

internal/dumper/dumper_test.go

@@ -6,6 +6,8 @@ import (
 	"log"
 	"os"
 	"testing"
+
+	"github.com/percona/mongodb-backup/internal/testutils"
 )
 
 var (
@@ -28,12 +30,14 @@ func TestWriteToFile(t *testing.T) {
 	diag("Using temporary file %q", tmpFile.Name())
 
 	mi := &MongodumpInput{
-		Host:    "localhost",
-		Port:    "17001",
-		Gzip:    false,
-		Oplog:   false,
-		Threads: 1,
-		Writer:  tmpFile,
+		Host:     testutils.MongoDBHost,
+		Port:     testutils.MongoDBPrimaryPort,
+		Username: testutils.MongoDBUser,
+		Password: testutils.MongoDBPassword,
+		Gzip:     false,
+		Oplog:    false,
+		Threads:  1,
+		Writer:   tmpFile,
 	}
 
 	mdump, err := NewMongodump(mi)

internal/restore/restore.go

@@ -12,14 +12,17 @@ import (
 )
 
 type MongoRestoreInput struct {
-	Archive string
-	DryRun  bool // Used only for testing
-	Host    string
-	Port    string
-	Gzip    bool
-	Oplog   bool
-	Threads int
-	Reader  io.ReadCloser
+	Archive  string
+	DryRun   bool // Used only for testing
+	Host     string
+	Port     string
+	Username string
+	Password string
+	AuthDB   string
+	Gzip     bool
+	Oplog    bool
+	Threads  int
+	Reader   io.ReadCloser
 }
 
 type MongoRestore struct {
@@ -52,6 +55,13 @@ func NewMongoRestore(i *MongoRestoreInput) (*MongoRestore, error) {
 		Namespace:  &options.Namespace{},
 		URI:        &options.URI{},
 	}
+	if i.Username != "" && i.Password != "" {
+		toolOpts.Auth.Username = i.Username
+		toolOpts.Auth.Password = i.Password
+		if i.AuthDB != "" {
+			toolOpts.Auth.Source = i.AuthDB
+		}
+	}
 
 	inputOpts := &mongorestore.InputOptions{
 		Gzip:                   i.Gzip,

internal/restore/restore_test.go

@@ -2,20 +2,24 @@ package restore
 
 import (
 	"testing"
+
+	"github.com/percona/mongodb-backup/internal/testutils"
 )
 
 func TestRestore(t *testing.T) {
 
 	input := &MongoRestoreInput{
 		// this file was generated with the dump pkg
-		Archive: "testdata/dump_test.000622955",
-		DryRun:  true,
-		Host:    "localhost",
-		Port:    "17001",
-		Gzip:    false,
-		Oplog:   false,
-		Threads: 1,
-		Reader:  nil,
+		Archive:  "testdata/dump_test.000622955",
+		DryRun:   true,
+		Host:     testutils.MongoDBHost,
+		Port:     testutils.MongoDBPrimaryPort,
+		Username: testutils.MongoDBUser,
+		Password: testutils.MongoDBPassword,
+		Gzip:     false,
+		Oplog:    false,
+		Threads:  1,
+		Reader:   nil,
 	}
 
 	r, err := NewMongoRestore(input)

scripts/init-cluster.sh

@@ -106,7 +106,8 @@ for MONGODB_PORT in ${TEST_MONGODB_PRIMARY_PORT} ${TEST_MONGODB_CONFIGSVR1_PORT}
 					roles: [
 						{ db: "admin", role: "backup" },
 						{ db: "admin", role: "clusterMonitor" },
-						{ db: "admin", role: "restore" }
+						{ db: "admin", role: "restore" },
+						{ db: "test", role: "readWrite" }
 					]
 				})' \
 				admin