foundation/storage_test.go at master · digithun/foundation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
/*
 * Copyright (c) 2019. Octofox.io
 */

package foundation

import (
	"github.com/aws/aws-sdk-go/aws"
	"github.com/aws/aws-sdk-go/aws/credentials"
	"github.com/aws/aws-sdk-go/aws/endpoints"
	"github.com/octofoxio/foundation/errors"
	"github.com/stretchr/testify/assert"
	"io/ioutil"
	"os"
	"testing"
)

func TestNewLocalFileStorage(t *testing.T) {
	wd, _ := os.Getwd()
	local := NewLocalFileStorage(wd)
	u, err := local.GetObjectURL("./storage.go")
	assert.NoError(t, err)
	assert.NotEmpty(t, u)
	t.Log(u)

	readCloser, err := local.GetObjectReader("./storage.go")
	t.Log(err)
	defer func() { _ = readCloser.Close() }()
	b, err := ioutil.ReadAll(readCloser)
	t.Log(err)
	t.Log(string(b))
}

type localAWSEndpointResolver struct{}

func (l *localAWSEndpointResolver) EndpointFor(service, region string, opts ...func(*endpoints.Options)) (endpoints.ResolvedEndpoint, error) {
	var endpoint = ""
	switch service {
	case "sns":
		endpoint = "http://localhost:4575"
	case "sqs":
		endpoint = "http://localhost:4576"
	}

	return endpoints.ResolvedEndpoint{
		URL: endpoint,
	}, nil
}

// Local AWS config
// require host to run localstack locally
// please use docker run -p "4567-4584:4567-4584" -p "8080:8080" -e DOCKER_HOST=unix:///var/run/docker.sock localstack/localstack
func NewLocalAWSConfig() *aws.Config {
	var resolver = localAWSEndpointResolver{}
	return &aws.Config{
		EndpointResolver: &resolver,
		Region:           aws.String("ap-southeast-1"),
	}
}
func TestS3FileStorage_RemoveObject(t *testing.T) {
	fs := NewS3FileStorage("", NewLocalAWSConfig())
	{
		p, err := fs.getURLByPath("test-bucket", "ap-southeast-1", "billpayment/image.png")
		assert.NoError(t, err)
		assert.EqualValues(t, "https://test-bucket.s3.ap-southeast-1.amazonaws.com/billpayment/image.png", p)
	}
	{

		p, err := fs.getURLByPath("test-bucket", "ap-southeast-1", "/billpayment/image.png")
		assert.NoError(t, err)
		assert.EqualValues(t, "https://test-bucket.s3.ap-southeast-1.amazonaws.com/billpayment/image.png", p)
	}
}

func TestS3StorageIntegration(t *testing.T) {
	// only on credential provide
	// and require create a bucket name foundation-test, to run full test
	// * Access Key ID:     AWS_ACCESS_KEY_ID
	// * Secret Access Key: AWS_SECRET_ACCESS_KEY
	if os.Getenv("AWS_ACCESS_KEY_ID") == "" {
		t.Skip()
	}
	awsConfig := &aws.Config{
		Region: aws.String("ap-southeast-1"),
		Credentials: credentials.NewChainCredentials([]credentials.Provider{
			&credentials.EnvProvider{},
		}),
	}
	ss := NewS3FileStorage("foundation-test", awsConfig)

	err := ss.PutObject(".foundationrc", []byte("just for fun"))
	assert.NoError(t, err)

	rcFile, err := ss.GetObjectURL(".foundationrc")
	assert.NoError(t, err)
	t.Log("get object url: " + rcFile)

	t.Log("test invalid key")
	{
		rcFile, err := ss.GetObjectURL(".foundationrcc")
		assert.Error(t, err)
		assert.NotEmpty(t, rcFile)
		assert.IsType(t, &errors.Error{}, err)
		assert.EqualValues(t, err.(*errors.Error).Type(), errors.ErrorTypeNotfound)
		t.Log("error: ", err.Error())
	}

	t.Log("test get object")
	{
		output, err := ss.GetObject(".foundationrc")
		assert.NoError(t, err)
		assert.NotEmpty(t, output)
		t.Logf("body: %s", string(output))
	}

	t.Log("test get invalid object")
	{
		output, err := ss.GetObject(".foundationrcc")
		assert.Error(t, err)
		assert.Empty(t, output)
	}
}