@@ -6,11 +6,17 @@ import (
66 "fmt"
77 "log"
88 "os"
9+ "strings"
910 "time"
1011
1112 _ "github.com/googleapis/go-sql-spanner" // Spanner driver
1213 "github.com/ory/dockertest/v3"
1314 "github.com/ory/dockertest/v3/docker"
15+
16+ database "cloud.google.com/go/spanner/admin/database/apiv1"
17+ dbpb "cloud.google.com/go/spanner/admin/database/apiv1/databasepb"
18+ instance "cloud.google.com/go/spanner/admin/instance/apiv1"
19+ inspb "cloud.google.com/go/spanner/admin/instance/apiv1/instancepb"
1420)
1521
1622const (
@@ -22,7 +28,6 @@ const (
2228 SPANNER_DATABASE = "test-db"
2329)
2430
25- // newSpanner spins up a Cloud Spanner emulator and connects to it using the Go SQL driver.
2631func newSpanner (opts ... OptionsFunc ) (* sql.DB , func (), error ) {
2732 option := & options {}
2833 for _ , f := range opts {
@@ -39,7 +44,7 @@ func newSpanner(opts ...OptionsFunc) (*sql.DB, func(), error) {
3944 Repository : SPANNER_IMAGE ,
4045 Tag : SPANNER_VERSION ,
4146 ExposedPorts : []string {
42- "9010/tcp" ,
47+ "9010/tcp" , "9020/tcp" ,
4348 },
4449 Labels : map [string ]string {"goose_test" : "1" },
4550 },
@@ -54,27 +59,28 @@ func newSpanner(opts ...OptionsFunc) (*sql.DB, func(), error) {
5459
5560 hostPort := resource .GetPort ("9010/tcp" )
5661 emulatorHost := fmt .Sprintf ("localhost:%s" , hostPort )
57-
58- // Set environment variable so that the Spanner driver connects to the emulator.
5962 os .Setenv ("SPANNER_EMULATOR_HOST" , emulatorHost )
6063
61- // Use gcloud CLI or spanner admin client to create instance & database.
64+ // Provision instance + database inside emulator
6265 err = pool .Retry (func () error {
63- ctx , cancel := context .WithTimeout (context .Background (), 10 * time .Second )
66+ ctx , cancel := context .WithTimeout (context .Background (), 20 * time .Second )
6467 defer cancel ()
6568
69+ if err := createSpannerResources (ctx ); err != nil {
70+ return err
71+ }
72+
6673 dsn := fmt .Sprintf ("projects/%s/instances/%s/databases/%s" , SPANNER_PROJECT , SPANNER_INSTANCE , SPANNER_DATABASE )
6774 db , err := sql .Open ("spanner" , dsn )
6875 if err != nil {
6976 return err
7077 }
7178 defer db .Close ()
72-
7379 return db .PingContext (ctx )
7480 })
7581 if err != nil {
7682 _ = pool .Purge (resource )
77- return nil , nil , fmt .Errorf ("could not ping spanner emulator: %v" , err )
83+ return nil , nil , fmt .Errorf ("could not initialize spanner emulator: %v" , err )
7884 }
7985
8086 dsn := fmt .Sprintf ("projects/%s/instances/%s/databases/%s" , SPANNER_PROJECT , SPANNER_INSTANCE , SPANNER_DATABASE )
@@ -94,3 +100,42 @@ func newSpanner(opts ...OptionsFunc) (*sql.DB, func(), error) {
94100
95101 return db , cleanup , nil
96102}
103+
104+ func createSpannerResources (ctx context.Context ) error {
105+ instClient , err := instance .NewInstanceAdminClient (ctx )
106+ if err != nil {
107+ return fmt .Errorf ("create instance client failed: %w" , err )
108+ }
109+ defer instClient .Close ()
110+
111+ dbClient , err := database .NewDatabaseAdminClient (ctx )
112+ if err != nil {
113+ return fmt .Errorf ("create database client failed: %w" , err )
114+ }
115+ defer dbClient .Close ()
116+
117+ instReq := & inspb.CreateInstanceRequest {
118+ Parent : "projects/" + SPANNER_PROJECT ,
119+ InstanceId : SPANNER_INSTANCE ,
120+ Instance : & inspb.Instance {
121+ Config : "projects/" + SPANNER_PROJECT + "/instanceConfigs/emulator-config" ,
122+ DisplayName : "Test Instance" ,
123+ NodeCount : 1 ,
124+ },
125+ }
126+ if _ , err = instClient .CreateInstance (ctx , instReq ); err != nil &&
127+ ! strings .Contains (err .Error (), "AlreadyExists" ) {
128+ return fmt .Errorf ("create instance failed: %w" , err )
129+ }
130+
131+ dbReq := & dbpb.CreateDatabaseRequest {
132+ Parent : fmt .Sprintf ("projects/%s/instances/%s" , SPANNER_PROJECT , SPANNER_INSTANCE ),
133+ CreateStatement : "CREATE DATABASE `" + SPANNER_DATABASE + "`" ,
134+ }
135+ if _ , err = dbClient .CreateDatabase (ctx , dbReq ); err != nil &&
136+ ! strings .Contains (err .Error (), "AlreadyExists" ) {
137+ return fmt .Errorf ("create database failed: %w" , err )
138+ }
139+
140+ return nil
141+ }
0 commit comments