TrueCloudLab/distribution
14
0
Fork 3
Code Issues 2 Pull requests Projects Releases 2 Packages Wiki Activity Actions

Merge pull request #1357 from aibaars/gcs-test

Browse source 
StorageDriver GCS test suite: try google.DefaultTokenSource first
This commit is contained in:
Richard Scothern 2016-01-18 11:29:01 -08:00
commit f4bd6cacd1
1 changed files with 23 additions and 17 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
docs/storage/driver/gcs/gcs_test.go
View file

@ -31,7 +31,7 @@ func init() {
// Skip GCS storage driver tests if environment variable parameters are not provided // Skip GCS storage driver tests if environment variable parameters are not provided
skipGCS = func() string { skipGCS = func() string {
if bucket == "" || credentials == "" { if bucket == "" || credentials == "" {
return "The following environment variables must be set to enable these tests: REGISTRY_STORAGE_GCS_BUCKET, REGISTRY_STORAGE_GCS_CREDS" return "The following environment variables must be set to enable these tests: REGISTRY_STORAGE_GCS_BUCKET, GOOGLE_APPLICATION_CREDENTIALS"
} }
return "" return ""
} }
@ -45,30 +45,36 @@ func init() {
panic(err) panic(err)
} }
defer os.Remove(root) defer os.Remove(root)
var ts oauth2.TokenSource
var email string
var privateKey []byte
_, err = os.Stat(credentials) ts, err = google.DefaultTokenSource(ctx.Background(), storage.ScopeFullControl)
if err == nil {
jsonKey, err := ioutil.ReadFile(credentials)
if err != nil { if err != nil {
panic(fmt.Sprintf("Unable to read credentials from file : %s", err))
}
credentials = string(jsonKey)
}
// Assume that the file contents are within the environment variable since it exists // Assume that the file contents are within the environment variable since it exists
// but does not contain a valid file path // but does not contain a valid file path
jwtConfig, err := google.JWTConfigFromJSON([]byte(credentials), storage.ScopeFullControl) jwtConfig, err := google.JWTConfigFromJSON([]byte(credentials), storage.ScopeFullControl)
if err != nil { if err != nil {
panic(fmt.Sprintf("Error reading JWT config : %s", err)) panic(fmt.Sprintf("Error reading JWT config : %s", err))
} }
email = jwtConfig.Email
privateKey = []byte(jwtConfig.PrivateKey)
if len(privateKey) == 0 {
panic("Error reading JWT config : missing private_key property")
}
if email == "" {
panic("Error reading JWT config : missing client_email property")
}
ts = jwtConfig.TokenSource(ctx.Background())
}
gcsDriverConstructor = func(rootDirectory string) (storagedriver.StorageDriver, error) { gcsDriverConstructor = func(rootDirectory string) (storagedriver.StorageDriver, error) {
parameters := driverParameters{ parameters := driverParameters{
bucket: bucket, bucket: bucket,
rootDirectory: root, rootDirectory: root,
email: jwtConfig.Email, email: email,
privateKey: []byte(jwtConfig.PrivateKey), privateKey: privateKey,
client: oauth2.NewClient(ctx.Background(), jwtConfig.TokenSource(ctx.Background())), client: oauth2.NewClient(ctx.Background(), ts),
} }
return New(parameters) return New(parameters)