Added option to spin up a server with replica set

.github/PULL_REQUEST_TEMPLATE.md

@@ -0,0 +1,13 @@
+# Description
+
+<!-- [TS-X](https://strikesecurity.atlassian.net/browse/TS-X) -->
+
+- N/A
+
+# Screenshots
+
+- N/A
+
+# Notes
+
+- N/A

.github/delete-merged-branch-config.yml

@@ -0,0 +1,4 @@
+exclude: 
+  - staging
+  - master
+delete_closed_pr: false

.github/skaffold-backend.yaml

@@ -0,0 +1,36 @@
+apiVersion: skaffold/v4beta1
+kind: Config
+metadata:
+  name: REPOSITORY_NAME
+build:
+  platforms: ["linux/amd64"]
+  tagPolicy:
+    gitCommit:
+      variant: CommitSha
+      prefix: skaffold-preview-PR_NUMBER
+  artifacts:
+  - image: REPOSITORY_NAME
+    docker:
+      dockerfile: Dockerfile
+      buildArgs:
+        env: stg
+        APPLICATION_NAME: REPOSITORY_NAME
+        github_token: GH_TOKEN
+        github_user: strike-cloud
+deploy:
+  helm:
+    releases:
+    - name: "REPOSITORY_NAME-prPR_NUMBER"
+      namespace: apps
+      repo: https://charts.santieich.com
+      chartPath: helm-apps/ephemeral-app-template/stg
+      valuesFiles:
+        - cloud/stg/values.yaml
+      setValueTemplates:
+        image: "{{.IMAGE_FULLY_QUALIFIED_REPO_NAME}}"
+      setValues:
+        appName: "REPOSITORY_NAME-prPR_NUMBER"
+
+    flags:
+      upgrade:
+        - --install

.github/skaffold-frontend.yaml

@@ -0,0 +1,35 @@
+apiVersion: skaffold/v4beta1
+kind: Config
+metadata:
+  name: APP_NAME
+build:
+  platforms: ["linux/amd64"]
+  tagPolicy:
+    gitCommit:
+      variant: CommitSha
+      prefix: skaffold-preview-PR_NUMBER
+  artifacts:
+  - image: APP_NAME
+    docker:
+      dockerfile: Dockerfile.APP_SHORT_NAME
+      buildArgs:
+        env: stg
+        APPLICATION_NAME: APP_SHORT_NAME
+        github_token: GH_TOKEN
+deploy:
+  helm:
+    releases:
+    - name: "APP_NAME-prPR_NUMBER"
+      namespace: apps
+      chartPath: helm-apps/ephemeral-app-template/stg
+      valuesFiles:
+        - cloud/APP_SHORT_NAME/stg/values.yaml
+      setValueTemplates:
+        image: "{{.IMAGE_FULLY_QUALIFIED_UNDERSCORE_APP_NAME}}"
+      setValues:
+        appName: "APP_NAME-prPR_NUMBER"
+        appVersion: skaffold-preview-PR_NUMBER
+
+    flags:
+      upgrade:
+        - --install

README.md

@@ -57,6 +57,26 @@ func TestSomething(t *testing.T) {
 }
 ```
 
+Spin up a replica set server:
+
+```go
+var mongoServer memongo.Server;
+
+func TestMain(m *testing.M) {
+  mongoServer, err = memongo.StartWitOptions(&memongo.Options{MongoVersion: "4.2.1", ShouldUseReplica: true})
+  if (err != nil) {
+    log.Fatal(err)
+  }
+  defer mongoServer.Stop()
+
+  os.Exit(m.Run())
+}
+
+func TestSomething(t *testing.T) {
+  connectAndDoStuff(mongoServer.URI(), memongo.RandomDatabase())
+}
+```
+
 # How it works
 
 Behind the scenes, when you run `Start()`, a few things are happening:

config.go

@@ -9,7 +9,6 @@ import (
 	"path"
 	"runtime"
 	"strconv"
-	"strings"
 	"time"
 
 	"github.com/benweissmann/memongo/memongolog"
@@ -18,6 +17,10 @@ import (
 
 // Options is the configuration options for a launched MongoDB binary
 type Options struct {
+	// ShouldUseReplica indicates whether a replica should be used. If this is not specified,
+	// no replica will be used and mongo server will be run as standalone.
+	ShouldUseReplica bool
+
 	// Port to run MongoDB on. If this is not specified, a random (OS-assigned)
 	// port will be used
 	Port int
@@ -103,23 +106,13 @@ func (opts *Options) fillDefaults() error {
 	}
 
 	if opts.Port == 0 {
-		// MongoDB after version 4 correctly reports what port it's running on if
-		// we tell it to run on port 0, which is ideal -- we just start it on port
-		// 0, the OS assigns a port, and mongo reports in the logs what port it
-		// got.
-		//
-		// For earlier versions, mongo just print "waiting for connections on port 0"
-		// which is unhelpful. So we start up a server and see what port we get,
-		// then shut down that server
-		if opts.MongoVersion == "" || parseMongoMajorVersion(opts.MongoVersion) < 4 {
-			port, err := getFreePort()
-			if err != nil {
-				return fmt.Errorf("error finding a free port: %s", err)
-			}
-
-			opts.Port = port
+		port, err := getFreePort()
+		if err != nil {
+			return fmt.Errorf("error finding a free port: %s", err)
 		}
 
+		opts.Port = port
+
 		if opts.StartupTimeout == 0 {
 			opts.StartupTimeout = 10 * time.Second
 		}
@@ -146,20 +139,6 @@ func (opts *Options) getOrDownloadBinPath() (string, error) {
 	return binPath, nil
 }
 
-func parseMongoMajorVersion(version string) int {
-	strParts := strings.Split(version, ".")
-	if len(strParts) == 0 {
-		return 0
-	}
-
-	maj, err := strconv.Atoi(strParts[0])
-	if err != nil {
-		return 0
-	}
-
-	return maj
-}
-
 func getFreePort() (int, error) {
 	// Based on: https://github.com/phayes/freeport/blob/master/freeport.go
 	addr, err := net.ResolveTCPAddr("tcp", "localhost:0")

go.mod

@@ -1,17 +1,16 @@
 module github.com/benweissmann/memongo
 
+go 1.14
+
 require (
 	github.com/acobaugh/osrelease v0.0.0-20181218015638-a93a0a55a249
-	github.com/go-stack/stack v1.8.0 // indirect
-	github.com/golang/snappy v0.0.1 // indirect
-	github.com/google/go-cmp v0.3.0 // indirect
-	github.com/nats-io/gnatsd v1.4.1
-	github.com/spf13/afero v1.2.1
-	github.com/stretchr/testify v1.3.0
-	github.com/tidwall/pretty v1.0.0 // indirect
-	github.com/xdg/scram v0.0.0-20180814205039-7eeb5667e42c // indirect
+	github.com/golang/snappy v0.0.2 // indirect
+	github.com/google/go-cmp v0.5.4 // indirect
+	github.com/spf13/afero v1.5.1
+	github.com/stretchr/testify v1.6.1
+	github.com/tidwall/pretty v1.0.2 // indirect
 	github.com/xdg/stringprep v1.0.0 // indirect
-	go.mongodb.org/mongo-driver v1.0.3
-	golang.org/x/crypto v0.0.0-20190618222545-ea8f1a30c443 // indirect
-	golang.org/x/sync v0.0.0-20190423024810-112230192c58 // indirect
+	go.mongodb.org/mongo-driver v1.4.4
+	golang.org/x/crypto v0.0.0-20201221181555-eec23a3978ad // indirect
+	golang.org/x/sync v0.0.0-20201207232520-09787c993a3a // indirect
 )