Fixed tests

rethinkdb · Jul 12, 2016 · d970d3c · d970d3c
1 parent b1cbac6
commit d970d3c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/example_connect_test.go b/example_connect_test.go
@@ -2,11 +2,21 @@ package gorethink_test
 
 import (
 	"log"
+	"os"
 
 	r "github.com/dancannon/gorethink"
 )
 
 var session *r.Session
+var url string
+
+func init() {
+	// If the test is being run by wercker look for the rethink url
+	url = os.Getenv("RETHINKDB_URL")
+	if url == "" {
+		url = "localhost:28015"
+	}
+}
 
 func ExampleConnect() {
 	var err error

diff --git a/example_test.go b/example_test.go
@@ -8,22 +8,22 @@ import (
 )
 
 func Example() {
-	session, err := r.Connect(ConnectOpts{
+	session, err := r.Connect(r.ConnectOpts{
 		Address: url,
 	})
 	if err != nil {
 		log.Fatalln(err)
 	}
 
-	res, err := Expr("Hello World").Run(session)
+	res, err := r.Expr("Hello World").Run(session)
 	if err != nil {
 		log.Fatalln(err)
 	}
 
 	var response string
 	err = res.One(&response)
 	if err != nil {
-		Log.Fatalln(err)
+		log.Fatalln(err)
 	}
 
 	fmt.Println(response)