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
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
|
package blobs
import (
"context"
"fmt"
"testing"
"github.com/Azure/azure-sdk-for-go/profiles/latest/storage/mgmt/storage"
"github.com/tombuildsstuff/giovanni/storage/2019-12-12/blob/containers"
"github.com/tombuildsstuff/giovanni/storage/internal/auth"
"github.com/tombuildsstuff/giovanni/testhelpers"
)
func TestAppendBlobLifecycle(t *testing.T) {
client, err := testhelpers.Build(t)
if err != nil {
t.Fatal(err)
}
ctx := context.TODO()
resourceGroup := fmt.Sprintf("acctestrg-%d", testhelpers.RandomInt())
accountName := fmt.Sprintf("acctestsa%s", testhelpers.RandomString())
containerName := fmt.Sprintf("cont-%d", testhelpers.RandomInt())
fileName := "append-blob.txt"
testData, err := client.BuildTestResources(ctx, resourceGroup, accountName, storage.BlobStorage)
if err != nil {
t.Fatal(err)
}
defer client.DestroyTestResources(ctx, resourceGroup, accountName)
containersClient := containers.NewWithEnvironment(client.Environment)
containersClient.Client = client.PrepareWithStorageResourceManagerAuth(containersClient.Client)
_, err = containersClient.Create(ctx, accountName, containerName, containers.CreateInput{})
if err != nil {
t.Fatal(fmt.Errorf("Error creating: %s", err))
}
defer containersClient.Delete(ctx, accountName, containerName)
storageAuth := auth.NewSharedKeyLiteAuthorizer(accountName, testData.StorageAccountKey)
blobClient := NewWithEnvironment(client.Environment)
blobClient.Client = client.PrepareWithAuthorizer(blobClient.Client, storageAuth)
t.Logf("[DEBUG] Putting Append Blob..")
if _, err := blobClient.PutAppendBlob(ctx, accountName, containerName, fileName, PutAppendBlobInput{}); err != nil {
t.Fatalf("Error putting append blob: %s", err)
}
t.Logf("[DEBUG] Retrieving Properties..")
props, err := blobClient.GetProperties(ctx, accountName, containerName, fileName, GetPropertiesInput{})
if err != nil {
t.Fatalf("Error retrieving properties: %s", err)
}
if props.ContentLength != 0 {
t.Fatalf("Expected Content-Length to be 0 but it was %d", props.ContentLength)
}
t.Logf("[DEBUG] Appending First Block..")
appendInput := AppendBlockInput{
Content: &[]byte{
12,
48,
93,
76,
29,
10,
},
}
if _, err := blobClient.AppendBlock(ctx, accountName, containerName, fileName, appendInput); err != nil {
t.Fatalf("Error appending first block: %s", err)
}
t.Logf("[DEBUG] Re-Retrieving Properties..")
props, err = blobClient.GetProperties(ctx, accountName, containerName, fileName, GetPropertiesInput{})
if err != nil {
t.Fatalf("Error retrieving properties: %s", err)
}
if props.ContentLength != 6 {
t.Fatalf("Expected Content-Length to be 6 but it was %d", props.ContentLength)
}
t.Logf("[DEBUG] Appending Second Block..")
appendInput = AppendBlockInput{
Content: &[]byte{
92,
62,
64,
47,
83,
77,
},
}
if _, err := blobClient.AppendBlock(ctx, accountName, containerName, fileName, appendInput); err != nil {
t.Fatalf("Error appending Second block: %s", err)
}
t.Logf("[DEBUG] Re-Retrieving Properties..")
props, err = blobClient.GetProperties(ctx, accountName, containerName, fileName, GetPropertiesInput{})
if err != nil {
t.Fatalf("Error retrieving properties: %s", err)
}
if props.ContentLength != 12 {
t.Fatalf("Expected Content-Length to be 12 but it was %d", props.ContentLength)
}
t.Logf("[DEBUG] Acquiring Lease..")
leaseDetails, err := blobClient.AcquireLease(ctx, accountName, containerName, fileName, AcquireLeaseInput{
LeaseDuration: -1,
})
if err != nil {
t.Fatalf("Error acquiring Lease: %s", err)
}
t.Logf("[DEBUG] Lease ID is %q", leaseDetails.LeaseID)
t.Logf("[DEBUG] Appending Third Block..")
appendInput = AppendBlockInput{
Content: &[]byte{
64,
35,
28,
93,
11,
23,
},
LeaseID: &leaseDetails.LeaseID,
}
if _, err := blobClient.AppendBlock(ctx, accountName, containerName, fileName, appendInput); err != nil {
t.Fatalf("Error appending Third block: %s", err)
}
t.Logf("[DEBUG] Re-Retrieving Properties..")
props, err = blobClient.GetProperties(ctx, accountName, containerName, fileName, GetPropertiesInput{
LeaseID: &leaseDetails.LeaseID,
})
if err != nil {
t.Fatalf("Error retrieving properties: %s", err)
}
if props.ContentLength != 18 {
t.Fatalf("Expected Content-Length to be 18 but it was %d", props.ContentLength)
}
t.Logf("[DEBUG] Breaking Lease..")
breakLeaseInput := BreakLeaseInput{
LeaseID: leaseDetails.LeaseID,
}
if _, err := blobClient.BreakLease(ctx, accountName, containerName, fileName, breakLeaseInput); err != nil {
t.Fatalf("Error breaking lease: %s", err)
}
t.Logf("[DEBUG] Deleting Lease..")
if _, err := blobClient.Delete(ctx, accountName, containerName, fileName, DeleteInput{}); err != nil {
t.Fatalf("Error deleting: %s", err)
}
}
|
