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
|
package messages
import (
"context"
"fmt"
"testing"
"github.com/Azure/azure-sdk-for-go/profiles/latest/storage/mgmt/storage"
"github.com/tombuildsstuff/giovanni/storage/2017-07-29/queue/queues"
"github.com/tombuildsstuff/giovanni/storage/internal/auth"
"github.com/tombuildsstuff/giovanni/testhelpers"
)
func TestLifeCycle(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())
queueName := fmt.Sprintf("queue-%d", testhelpers.RandomInt())
testData, err := client.BuildTestResources(ctx, resourceGroup, accountName, storage.Storage)
if err != nil {
t.Fatal(err)
}
defer client.DestroyTestResources(ctx, resourceGroup, accountName)
storageAuth := auth.NewSharedKeyLiteAuthorizer(accountName, testData.StorageAccountKey)
queuesClient := queues.NewWithEnvironment(client.Environment)
queuesClient.Client = client.PrepareWithAuthorizer(queuesClient.Client, storageAuth)
messagesClient := NewWithEnvironment(client.Environment)
messagesClient.Client = client.PrepareWithAuthorizer(messagesClient.Client, storageAuth)
_, err = queuesClient.Create(ctx, accountName, queueName, map[string]string{})
if err != nil {
t.Fatalf("Error creating queue: %s", err)
}
defer queuesClient.Delete(ctx, accountName, queueName)
input := PutInput{
Message: "ohhai",
}
putResp, err := messagesClient.Put(ctx, accountName, queueName, input)
if err != nil {
t.Fatalf("Error putting message in queue: %s", err)
}
messageId := (*putResp.QueueMessages)[0].MessageId
popReceipt := (*putResp.QueueMessages)[0].PopReceipt
_, err = messagesClient.Update(ctx, accountName, queueName, messageId, UpdateInput{
PopReceipt: popReceipt,
Message: "Updated message",
VisibilityTimeout: 65,
})
if err != nil {
t.Fatalf("Error updating: %s", err)
}
for i := 0; i < 5; i++ {
input := PutInput{
Message: fmt.Sprintf("Message %d", i),
}
_, err := messagesClient.Put(ctx, accountName, queueName, input)
if err != nil {
t.Fatalf("Error putting message %d in queue: %s", i, err)
}
}
peakedMessages, err := messagesClient.Peek(ctx, accountName, queueName, 3)
if err != nil {
t.Fatalf("Error peaking messages: %s", err)
}
for _, v := range *peakedMessages.QueueMessages {
t.Logf("Message: %q", v.MessageId)
}
retrievedMessages, err := messagesClient.Get(ctx, accountName, queueName, 6, GetInput{})
if err != nil {
t.Fatalf("Error retrieving messages: %s", err)
}
for _, v := range *retrievedMessages.QueueMessages {
t.Logf("Message: %q", v.MessageId)
_, err = messagesClient.Delete(ctx, accountName, queueName, v.MessageId, v.PopReceipt)
if err != nil {
t.Fatalf("Error deleting message from queue: %s", err)
}
}
}
|
