Using option.WithEndpoint to connect to bigquery-emulator

Author: mbobrovskyi

database/bigquery/README.md

@@ -4,7 +4,7 @@
 * [Examples](./examples)
 
 ### Usage
-`bigquery://https://www.googleapis.com/bigquery/v2:443?x-migrations-table=schema_migrations&x-statement-timeout=0&credentials_filename=./myproject-XXXXXXXXXXXXX-XXXXXXXXXXXX.json&project_id=myproject-XXXXXXXXXXXXX&dataset_id=mydataset`
+`bigquery://https://bigquery.googleapis.com/bigquery/v2/?x-migrations-table=schema_migrations&x-statement-timeout=0&credentials_filename=./myproject-XXXXXXXXXXXXX-XXXXXXXXXXXX.json&project_id=myproject-XXXXXXXXXXXXX&dataset_id=mydataset`
 
 
 ### System variables reference

database/bigquery/bigquery.go

@@ -14,6 +14,7 @@ import (
 	"google.golang.org/api/option"
 	"io"
 	nurl "net/url"
+	"os"
 	"strconv"
 	"strings"
 	"time"
@@ -148,12 +149,14 @@ func (b *BigQuery) Open(url string) (database.Driver, error) {
 		}
 	}
 
-	//By default, try to get from environment variable GOOGLE_APPLICATION_CREDENTIALS
 	if q.Has("credentials_filename") {
 		opts = append(opts, option.WithCredentialsFile(q.Get("credentials_filename")))
+	} else if os.Getenv("GOOGLE_APPLICATION_CREDENTIALS") != "" {
+		opts = append(opts, option.WithCredentialsFile(os.Getenv("GOOGLE_APPLICATION_CREDENTIALS")))
+	} else {
+		opts = append(opts, option.WithoutAuthentication())
 	}
 
-	// By default, try to get from credentials file
 	projectID := bigquery.DetectProjectID
 	if q.Has("project_id") {
 		projectID = q.Get("project_id")
@@ -175,6 +178,8 @@ func (b *BigQuery) Open(url string) (database.Driver, error) {
 		config.TimeZone = q.Get("time_zone")
 	}
 
+	opts = append(opts, option.WithEndpoint(fmt.Sprintf("%s%s", purl.Host, purl.Path)))
+
 	client, err := bigquery.NewClient(ctx, projectID, opts...)
 	if err != nil {
 		return nil, err

database/bigquery/bigquery_test.go

@@ -7,7 +7,8 @@ import (
 )
 
 const (
-	connectionUrl = "bigquery://https://www.googleapis.com/bigquery/v2:443?x-migrations-table=schema_migrations&x-statement-timeout=0&credentials_filename=./tmp/myproject-XXXXXXXXXXXXX-XXXXXXXXXXXX.json&dataset_id=mydataset"
+	//connectionUrl = "bigquery://https://bigquery.googleapis.com/bigquery/v2/?x-migrations-table=schema_migrations&x-statement-timeout=0&credentials_filename=./tmp/myproject-XXXXXXXXXXXXX-XXXXXXXXXXXX.json&dataset_id=mydataset"
+	connectionUrl = "bigquery://http://0.0.0.0:9050/?x-migrations-table=schema_migrations&project_id=myproject&dataset_id=mydataset"
 )
 
 func openConnection() (database.Driver, error) {
@@ -22,65 +23,75 @@ func openConnection() (database.Driver, error) {
 }
 
 func TestOpen(t *testing.T) {
-	_, err := openConnection()
+	driver, err := openConnection()
 	if err != nil {
 		t.Error(err)
 		return
 	}
+
+	defer driver.Close()
 }
 
-func TestVersion(t *testing.T) {
+func TestClose(t *testing.T) {
 	driver, err := openConnection()
 	if err != nil {
 		t.Error(err)
 		return
 	}
 
-	version, dirty, err := driver.Version()
+	defer driver.Close()
+
+	err = driver.Close()
 	if err != nil {
 		t.Error(err)
 		return
 	}
-
-	t.Log(version, dirty)
 }
 
-func TestDrop(t *testing.T) {
+func TestVersion(t *testing.T) {
 	driver, err := openConnection()
 	if err != nil {
 		t.Error(err)
 		return
 	}
 
-	err = driver.Drop()
+	defer driver.Close()
+
+	version, dirty, err := driver.Version()
 	if err != nil {
 		t.Error(err)
 		return
 	}
+
+	t.Log(version, dirty)
 }
 
-func TestClose(t *testing.T) {
+func TestSetVersion(t *testing.T) {
 	driver, err := openConnection()
 	if err != nil {
 		t.Error(err)
 		return
 	}
 
-	err = driver.Close()
+	defer driver.Close()
+
+	err = driver.SetVersion(-1, false)
 	if err != nil {
 		t.Error(err)
 		return
 	}
 }
 
-func TestSetVersion(t *testing.T) {
+func TestDrop(t *testing.T) {
 	driver, err := openConnection()
 	if err != nil {
 		t.Error(err)
 		return
 	}
 
-	err = driver.SetVersion(-1, false)
+	defer driver.Close()
+
+	err = driver.Drop()
 	if err != nil {
 		t.Error(err)
 		return
@@ -94,6 +105,8 @@ func TestRun(t *testing.T) {
 		return
 	}
 
+	defer driver.Close()
+
 	err = driver.Run(strings.NewReader(`
 		CREATE TABLE IF NOT EXISTS users (
 			first_name STRING,
@@ -112,6 +125,8 @@ func TestRunWithError(t *testing.T) {
 		return
 	}
 
+	defer driver.Close()
+
 	err = driver.Run(strings.NewReader(`
 		CREATE TABLE IF NOT EXISTS users (
 			first_name STRINGa,