Automate Atlas connectivity tests

GODRIVER-145

Change-Id: I04a25f03fc87dde23bbccfd555a8a5e7e78a4fd6

Author: connieyu

.evergreen/config.yml

@@ -326,6 +326,37 @@ functions:
           export PATH="${GCC_PATH}:${GO_DIST}/bin:$PATH"
           MONGO_GO_DRIVER_COMPRESSOR="${MONGO_GO_DRIVER_COMPRESSOR}" make -s evg-test-auth
 
+  run-atlas-test:
+    - command: shell.exec
+      type: test
+      params:
+        working_dir: src/go.mongodb.org/mongo-driver
+        script: |
+          # DO NOT ECHO WITH XTRACE (which PREPARE_SHELL does)
+          if [ "Windows_NT" = "$OS" ]; then
+            export GOPATH=$(cygpath -w $(dirname $(dirname $(dirname `pwd`))))
+            export GOCACHE=$(cygpath -w "$(pwd)/.cache")
+          else
+            export GOPATH=$(dirname $(dirname $(dirname `pwd`)))
+            export GOCACHE="$(pwd)/.cache"
+          fi;
+          export GOPATH="$GOPATH"
+          export GOROOT="${GO_DIST}"
+          export GOCACHE="$GOCACHE"
+          export PATH="${GCC_PATH}:${GO_DIST}/bin:$PATH"
+          export ATLAS_FREE="${atlas_free_tier_uri}"
+          export ATLAS_REPLSET="${atlas_replica_set_uri}"
+          export ATLAS_SHARD="${atlas_sharded_uri}"
+          export ATLAS_TLS11="${atlas_tls_v11_uri}"
+          export ATLAS_TLS12="${atlas_tls_v12_uri}"
+          export ATLAS_FREE_SRV="${atlas_free_tier_uri_srv}"
+          export ATLAS_REPLSET_SRV="${atlas_replica_set_uri_srv}"
+          export ATLAS_SHARD_SRV="${atlas_sharded_uri_srv}"
+          export ATLAS_TLS11_SRV="${atlas_tls_v11_uri_srv}"
+          export ATLAS_TLS12_SRV="${atlas_tls_v12_uri_srv}"
+          make -s evg-test-atlas
+
+
 pre:
   - func: fetch-source
   - func: prepare-resources
@@ -655,6 +686,10 @@ tasks:
             targets: "build"
             BUILD_ENV: "GOARCH=ppc64le"
 
+    - name: "atlas-test"
+      commands:
+        - func: "run-atlas-test"
+
 axes:
   - id: version
     display_name: MongoDB Version
@@ -753,6 +788,15 @@ buildvariants:
   tasks:
     - name: ".compile-check"
 
+- name: atlas-test
+  display_name: "Atlas test"
+  run_on:
+    - ubuntu1604-build
+  expansions:
+    GO_DIST: "/opt/golang/go1.12"
+  tasks:
+    - name: "atlas-test"
+
 - matrix_name: "tests-legacy-auth-ssl"
   matrix_spec: { version: ["2.6", "3.0"], os-ssl-legacy: "*" }
   display_name: "${version} ${os-ssl-legacy}"

Makefile

@@ -10,6 +10,7 @@ EXAMPLES_PKGS = $(shell etc/list_pkgs.sh ./examples)
 EXAMPLES_TEST_PKGS = $(shell etc/list_test_pkgs.sh ./examples)
 PKGS = $(BSON_PKGS) $(MONGO_PKGS) $(UNSTABLE_PKGS) $(TAG_PKG) $(EXAMPLES_PKGS)
 TEST_PKGS = $(BSON_TEST_PKGS) $(MONGO_TEST_PKGS) $(UNSTABLE_TEST_PKGS) $(TAG_PKG) $(EXAMPLES_TEST_PKGS)
+ATLAS_URIS = "$(ATLAS_FREE)" "$(ATLAS_REPLSET)" "$(ATLAS_SHARD)" "$(ATLAS_TLS11)" "$(ATLAS_TLS12)" "$(ATLAS_FREE_SRV)" "$(ATLAS_REPLSET_SRV)" "$(ATLAS_SHARD_SRV)" "$(ATLAS_TLS11_SRV)" "$(ATLAS_TLS12_SRV)"
 
 TEST_TIMEOUT = 600
 
@@ -125,6 +126,10 @@ evg-test:
 evg-test-auth:
 	go run -tags gssapi ./x/mongo/driver/examples/count/main.go -uri $(MONGODB_URI)
 
+.PHONY: evg-test-atlas
+evg-test-atlas:
+	go run ./mongo/testatlas/main.go $(ATLAS_URIS)
+
 # benchmark specific targets and support
 perf:driver-test-data.tar.gz
 	tar -zxf $< $(if $(eq $(UNAME_S),Darwin),-s , --transform=s)/data/perf/

mongo/testatlas/main.go

@@ -0,0 +1,64 @@
+// Copyright (C) MongoDB, Inc. 2017-present.
+//
+// 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
+
+package main
+
+import (
+	"context"
+	"flag"
+	"log"
+
+	"go.mongodb.org/mongo-driver/bson"
+	"go.mongodb.org/mongo-driver/mongo"
+	"go.mongodb.org/mongo-driver/mongo/options"
+)
+
+func main() {
+	flag.Parse()
+	uris := flag.Args()
+
+	for _, uri := range uris {
+		ctx := context.Background()
+
+		client, err := mongo.Connect(ctx, options.Client().ApplyURI(uri))
+		if err != nil {
+			log.Fatalf("failed creating and connecting to client: %v", err)
+		}
+
+		db := client.Database("test")
+		defer func() {
+			err := db.Drop(ctx)
+			if err != nil {
+				log.Fatalf("failed dropping database: %v", err)
+			}
+
+			err = client.Disconnect(ctx)
+			if err != nil {
+				log.Fatalf("failed disconnecting from client: %v", err)
+			}
+		}()
+
+		coll := db.Collection("test")
+
+		err = db.RunCommand(
+			ctx,
+			bson.D{{"isMaster", 1}},
+		).Err()
+		if err != nil {
+			log.Fatalf("failed executing isMaster command: %v", err)
+		}
+
+		_, err = coll.InsertOne(ctx, bson.D{{"x", 1}})
+		if err != nil {
+			log.Fatalf("failed executing insertOne command: %v", err)
+		}
+
+		res := coll.FindOne(ctx, bson.D{{"x", 1}})
+		if res.Err() != nil {
+			log.Fatalf("failed executing findOne command: %v", res.Err())
+		}
+	}
+}