Skip to content

Commit 735d07c

Browse files
authored
Merge pull request #7 from knights-analytics/s3-test-imports
Prevent s3 tests for importing google gs dependencies
2 parents b3d0ac5 + 7e46526 commit 735d07c

5 files changed

Lines changed: 35 additions & 34 deletions

File tree

auth/jwt.go

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -3,25 +3,25 @@ package auth
33
import (
44
"bytes"
55
"encoding/json"
6+
"io"
7+
"os"
8+
69
"github.com/pkg/errors"
710
"github.com/viant/afs/option"
811
"github.com/viant/afs/storage"
9-
"golang.org/x/oauth2/google"
1012
"golang.org/x/oauth2/jwt"
11-
"io/ioutil"
12-
"os"
1313
)
1414

15-
//JWTProvider represetns JWT based auth provider
15+
// JWTProvider represents JWT based auth provider
1616
type JWTProvider interface {
1717
JWTConfig(scopes ...string) (config *jwt.Config, projectID string, err error)
1818
}
1919

20-
//JwtConfig represents google service account secrets
20+
// JwtConfig represents google service account secrets
2121
type JwtConfig struct {
22-
//google cloud credential
22+
// google cloud credential
2323
ClientEmail string `json:"client_email,omitempty"`
24-
TokenURL string `json:"token_uri,omitempty"`
24+
TokenURL string `json:"token_url,omitempty"`
2525
PrivateKey string `json:"private_key,omitempty"`
2626
PrivateKeyID string `json:"private_key_id,omitempty"`
2727
ProjectID string `json:"project_id,omitempty"`
@@ -32,7 +32,7 @@ type JwtConfig struct {
3232
jwtClientConfig *jwt.Config
3333
}
3434

35-
//JWTConfig returns new JWT config for supplied scopes
35+
// JWTConfig returns new JWT config for supplied scopes
3636
func (c *JwtConfig) JWTConfig(scopes ...string) (config *jwt.Config, projectID string, err error) {
3737
if c.jwtClientConfig != nil {
3838
return c.jwtClientConfig, c.ProjectID, nil
@@ -46,13 +46,13 @@ func (c *JwtConfig) JWTConfig(scopes ...string) (config *jwt.Config, projectID s
4646
TokenURL: c.TokenURL,
4747
}
4848
if result.TokenURL == "" {
49-
result.TokenURL = google.JWTTokenURL
49+
result.TokenURL = "https://oauth2.googleapis.com/token"
5050
}
5151
c.jwtClientConfig = result
5252
return result, c.ProjectID, nil
5353
}
5454

55-
//NewJwtConfig returns new secrets from location
55+
// NewJwtConfig returns new secrets from location
5656
func NewJwtConfig(options ...storage.Option) (*JwtConfig, error) {
5757
location := &option.Location{}
5858
var JSONPayload = make([]byte, 0)
@@ -67,7 +67,7 @@ func NewJwtConfig(options ...storage.Option) (*JwtConfig, error) {
6767
return nil, errors.Wrap(err, "failed to open auth config")
6868
}
6969
defer func() { _ = file.Close() }()
70-
if JSONPayload, err = ioutil.ReadAll(file); err != nil {
70+
if JSONPayload, err = io.ReadAll(file); err != nil {
7171
return nil, err
7272
}
7373
}

s3/example_test.go

Lines changed: 17 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -3,17 +3,17 @@ package s3_test
33
import (
44
"context"
55
"fmt"
6+
"io"
7+
"log"
8+
"os"
9+
"path"
10+
"strings"
11+
612
"github.com/aws/aws-sdk-go/aws"
713
"github.com/viant/afs"
814
"github.com/viant/afs/option"
915
"github.com/viant/afsc/auth"
10-
"github.com/viant/afsc/gs"
1116
"github.com/viant/afsc/s3"
12-
"io/ioutil"
13-
"log"
14-
"os"
15-
"path"
16-
"strings"
1717
)
1818

1919
func ExampleAfsService() {
@@ -33,7 +33,7 @@ func ExampleAfsService() {
3333
log.Fatal(err)
3434
}
3535
defer reader.Close()
36-
data, err := ioutil.ReadAll(reader)
36+
data, err := io.ReadAll(reader)
3737
if err != nil {
3838
log.Fatal(err)
3939
}
@@ -42,25 +42,25 @@ func ExampleAfsService() {
4242
}
4343

4444
func ExampleNew() {
45-
service := gs.New()
45+
service := s3.New()
4646
ctx := context.Background()
4747
reader, err := service.OpenURL(ctx, "s3://my-bucket/folder/asset")
4848
if err != nil {
4949
log.Fatal(err)
5050
}
5151
defer reader.Close()
52-
data, err := ioutil.ReadAll(reader)
52+
data, err := io.ReadAll(reader)
5353
if err != nil {
5454
log.Fatal(err)
5555
}
5656
fmt.Printf("data: %s\n", data)
5757
}
5858

59-
//Example_Storager storager usage example (uses path rather then URLs)
59+
// Example_Storager storager usage example (uses path rather then URLs)
6060
func Example_Storager() {
6161

6262
ctx := context.Background()
63-
service, err := gs.NewStorager(ctx, "s3://myBucket/")
63+
service, err := s3.NewStorager(ctx, "s3://myBucket/")
6464
if err != nil {
6565
log.Fatal(err)
6666
}
@@ -73,7 +73,7 @@ func Example_Storager() {
7373
if err != nil {
7474
log.Fatal(err)
7575
}
76-
data, err := ioutil.ReadAll(reader)
76+
data, err := io.ReadAll(reader)
7777
fmt.Printf("data: %s\n", data)
7878

7979
has, _ := service.Exists(ctx, location)
@@ -102,11 +102,11 @@ func ExampleNewAuthConfig() {
102102
}
103103

104104
ctx := context.Background()
105-
//add default import _ "github.com/viant/afsc/s3"
105+
// add default import _ "github.com/viant/afsc/s3"
106106

107107
service := afs.New()
108108
reader, err := service.OpenURL(ctx, "s3://my-bucket/myfolder/asset.txt", authConfig)
109-
data, err := ioutil.ReadAll(reader)
109+
data, err := io.ReadAll(reader)
110110
if err != nil {
111111
log.Fatal(err)
112112
}
@@ -116,11 +116,11 @@ func ExampleNewAuthConfig() {
116116

117117
func ExampleAwsConfig() {
118118
var awsConfig *aws.Config
119-
//get config
119+
// get config
120120
ctx := context.Background()
121121
service := afs.New()
122122
reader, err := service.OpenURL(ctx, "s3://my-bucket/myfolder/asset.txt", awsConfig)
123-
data, err := ioutil.ReadAll(reader)
123+
data, err := io.ReadAll(reader)
124124
if err != nil {
125125
log.Fatal(err)
126126
}
@@ -139,7 +139,7 @@ func ExampleNewCustomKey() {
139139
log.Fatal(err)
140140
}
141141
reader, err := service.OpenURL(ctx, "s3://mybucket/folder/secret1.txt", customKey)
142-
data, err := ioutil.ReadAll(reader)
142+
data, err := io.ReadAll(reader)
143143
if err != nil {
144144
log.Fatal(err)
145145
}

s3/manager.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ type manager struct {
2323

2424
func (m *manager) provider(ctx context.Context, baseURL string, options ...storage.Option) (storage.Storager, error) {
2525
options = m.Options(options)
26-
return newStorager(ctx, baseURL, options...)
26+
return NewStorager(ctx, baseURL, options...)
2727
}
2828

2929
func (m *manager) copyInMemory(ctx context.Context, sourceURL, destURL string, options []storage.Option) error {

s3/storager.go

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,10 @@ package s3
22

33
import (
44
"context"
5+
"log"
6+
"os"
7+
"time"
8+
59
"github.com/aws/aws-sdk-go/aws"
610
"github.com/aws/aws-sdk-go/aws/session"
711
"github.com/aws/aws-sdk-go/service/s3"
@@ -10,9 +14,6 @@ import (
1014
"github.com/viant/afs/storage"
1115
"github.com/viant/afs/url"
1216
"github.com/viant/afsc/logger"
13-
"log"
14-
"os"
15-
"time"
1617
)
1718

1819
const (
@@ -116,7 +117,7 @@ func getAwsConfig(options []storage.Option) (config *aws.Config, err error) {
116117
return config, err
117118
}
118119

119-
func newStorager(ctx context.Context, baseURL string, options ...storage.Option) (*storager, error) {
120+
func NewStorager(ctx context.Context, baseURL string, options ...storage.Option) (*storager, error) {
120121
result := &storager{
121122
bucket: url.Host(baseURL),
122123
}

s3/storager_test.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,6 +34,6 @@ func NewTestStorager(ctx context.Context, bucket string) (storage.Storager, erro
3434
if err != nil {
3535
return nil, err
3636
}
37-
return newStorager(ctx, fmt.Sprintf("s3://%s", bucket), authConfig)
37+
return NewStorager(ctx, fmt.Sprintf("s3://%s", bucket), authConfig)
3838

3939
}

0 commit comments

Comments
 (0)