dustMason-right_aws 2.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- data/History.txt +305 -0
- data/Manifest.txt +60 -0
- data/README.txt +165 -0
- data/Rakefile +112 -0
- data/lib/acf/right_acf_interface.rb +549 -0
- data/lib/acf/right_acf_invalidations.rb +144 -0
- data/lib/acf/right_acf_origin_access_identities.rb +230 -0
- data/lib/acf/right_acf_streaming_interface.rb +229 -0
- data/lib/acw/right_acw_interface.rb +248 -0
- data/lib/as/right_as_interface.rb +698 -0
- data/lib/awsbase/benchmark_fix.rb +39 -0
- data/lib/awsbase/right_awsbase.rb +1174 -0
- data/lib/awsbase/support.rb +35 -0
- data/lib/awsbase/version.rb +9 -0
- data/lib/ec2/right_ec2.rb +458 -0
- data/lib/ec2/right_ec2_ebs.rb +465 -0
- data/lib/ec2/right_ec2_images.rb +413 -0
- data/lib/ec2/right_ec2_instances.rb +785 -0
- data/lib/ec2/right_ec2_monitoring.rb +70 -0
- data/lib/ec2/right_ec2_placement_groups.rb +108 -0
- data/lib/ec2/right_ec2_reserved_instances.rb +174 -0
- data/lib/ec2/right_ec2_security_groups.rb +396 -0
- data/lib/ec2/right_ec2_spot_instances.rb +425 -0
- data/lib/ec2/right_ec2_tags.rb +139 -0
- data/lib/ec2/right_ec2_vpc.rb +583 -0
- data/lib/ec2/right_ec2_windows_mobility.rb +84 -0
- data/lib/elb/right_elb_interface.rb +571 -0
- data/lib/iam/right_iam_access_keys.rb +71 -0
- data/lib/iam/right_iam_groups.rb +195 -0
- data/lib/iam/right_iam_interface.rb +341 -0
- data/lib/iam/right_iam_mfa_devices.rb +67 -0
- data/lib/iam/right_iam_users.rb +251 -0
- data/lib/rds/right_rds_interface.rb +1309 -0
- data/lib/right_aws.rb +88 -0
- data/lib/route_53/right_route_53_interface.rb +630 -0
- data/lib/s3/right_s3.rb +1123 -0
- data/lib/s3/right_s3_interface.rb +1198 -0
- data/lib/sdb/active_sdb.rb +1107 -0
- data/lib/sdb/right_sdb_interface.rb +753 -0
- data/lib/sns/right_sns.rb +205 -0
- data/lib/sns/right_sns_interface.rb +343 -0
- data/lib/sqs/right_sqs.rb +387 -0
- data/lib/sqs/right_sqs_gen2.rb +342 -0
- data/lib/sqs/right_sqs_gen2_interface.rb +523 -0
- data/lib/sqs/right_sqs_interface.rb +593 -0
- data/right_aws.gemspec +91 -0
- data/test/acf/test_helper.rb +2 -0
- data/test/acf/test_right_acf.rb +138 -0
- data/test/awsbase/test_helper.rb +2 -0
- data/test/awsbase/test_right_awsbase.rb +12 -0
- data/test/ec2/test_helper.rb +2 -0
- data/test/ec2/test_right_ec2.rb +108 -0
- data/test/http_connection.rb +87 -0
- data/test/rds/test_helper.rb +2 -0
- data/test/rds/test_right_rds.rb +120 -0
- data/test/s3/test_helper.rb +2 -0
- data/test/s3/test_right_s3.rb +421 -0
- data/test/s3/test_right_s3_stubbed.rb +97 -0
- data/test/sdb/test_active_sdb.rb +357 -0
- data/test/sdb/test_batch_put_attributes.rb +54 -0
- data/test/sdb/test_helper.rb +3 -0
- data/test/sdb/test_right_sdb.rb +253 -0
- data/test/sns/test_helper.rb +2 -0
- data/test/sns/test_right_sns.rb +73 -0
- data/test/sqs/test_helper.rb +2 -0
- data/test/sqs/test_right_sqs.rb +285 -0
- data/test/sqs/test_right_sqs_gen2.rb +264 -0
- data/test/test_credentials.rb +37 -0
- data/test/ts_right_aws.rb +14 -0
- metadata +244 -0
@@ -0,0 +1,54 @@
|
|
1
|
+
# -*- coding: utf-8 -*-
|
2
|
+
require File.dirname(__FILE__) + '/test_helper.rb'
|
3
|
+
|
4
|
+
class TestSdb < Test::Unit::TestCase
|
5
|
+
|
6
|
+
def setup
|
7
|
+
STDOUT.sync = true
|
8
|
+
@domain = 'right_sdb_awesome_test_domain'
|
9
|
+
@attributes = {
|
10
|
+
'a' => { 'foo' => '123' },
|
11
|
+
'b' => { 'bar' => '456' }
|
12
|
+
}
|
13
|
+
# Interface instance
|
14
|
+
@sdb = Rightscale::SdbInterface.new
|
15
|
+
@sdb.delete_domain(@domain)
|
16
|
+
wait(SDB_DELAY, "after removing domain")
|
17
|
+
@sdb.create_domain(@domain)
|
18
|
+
wait(SDB_DELAY, "after recreating domain")
|
19
|
+
end
|
20
|
+
|
21
|
+
SDB_DELAY = 2
|
22
|
+
|
23
|
+
def wait(delay, msg='')
|
24
|
+
print "waiting #{delay} seconds #{msg}"
|
25
|
+
while delay>0 do
|
26
|
+
delay -= 1
|
27
|
+
print '.'
|
28
|
+
sleep 1
|
29
|
+
end
|
30
|
+
puts
|
31
|
+
end
|
32
|
+
|
33
|
+
def test_batch_put_attributes
|
34
|
+
@sdb.batch_put_attributes(@domain, @attributes)
|
35
|
+
wait(SDB_DELAY, "after putting attributes")
|
36
|
+
a = @sdb.get_attributes(@domain, 'a')[:attributes]
|
37
|
+
b = @sdb.get_attributes(@domain, 'b')[:attributes]
|
38
|
+
assert_equal( {'foo' => ['123']}, a)
|
39
|
+
assert_equal( {'bar' => ['456']}, b)
|
40
|
+
|
41
|
+
# Replace = false
|
42
|
+
@sdb.batch_put_attributes(@domain, { 'a' => {'foo' => ['789']}})
|
43
|
+
wait(SDB_DELAY, "after putting attributes")
|
44
|
+
a = @sdb.get_attributes(@domain, 'a')[:attributes]
|
45
|
+
assert_equal ['123', '789'], a['foo'].sort
|
46
|
+
|
47
|
+
# Replace = true
|
48
|
+
@sdb.batch_put_attributes(@domain, {'b' => {'bar' => ['789']}}, true)
|
49
|
+
wait(SDB_DELAY, "after putting attributes")
|
50
|
+
b = @sdb.get_attributes(@domain, 'b')[:attributes]
|
51
|
+
assert_equal ['789'], b['bar'].sort
|
52
|
+
|
53
|
+
end
|
54
|
+
end
|
@@ -0,0 +1,253 @@
|
|
1
|
+
require File.dirname(__FILE__) + '/test_helper.rb'
|
2
|
+
|
3
|
+
class TestSdb < Test::Unit::TestCase
|
4
|
+
|
5
|
+
def setup
|
6
|
+
STDOUT.sync = true
|
7
|
+
@domain = 'right_sdb_awesome_test_domain'
|
8
|
+
@item = 'toys'
|
9
|
+
@attr = { 'Jon' => %w{beer car} }
|
10
|
+
# Interface instance
|
11
|
+
@sdb = Rightscale::SdbInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
|
12
|
+
end
|
13
|
+
|
14
|
+
SDB_DELAY = 7
|
15
|
+
|
16
|
+
def wait(delay, msg='')
|
17
|
+
print "waiting #{delay} seconds #{msg}"
|
18
|
+
while delay>0 do
|
19
|
+
delay -= 1
|
20
|
+
print '.'
|
21
|
+
sleep 1
|
22
|
+
end
|
23
|
+
puts
|
24
|
+
end
|
25
|
+
|
26
|
+
#---------------------------
|
27
|
+
# Rightscale::SdbInterface
|
28
|
+
#---------------------------
|
29
|
+
|
30
|
+
def test_00_delete_domain
|
31
|
+
# delete the domain to reset all the things
|
32
|
+
assert @sdb.delete_domain(@domain), 'delete_domain fail'
|
33
|
+
wait SDB_DELAY, 'after domain deletion'
|
34
|
+
end
|
35
|
+
|
36
|
+
def test_01_create_domain
|
37
|
+
# check that domain does not exist
|
38
|
+
assert !@sdb.list_domains[:domains].include?(@domain)
|
39
|
+
# create domain
|
40
|
+
assert @sdb.create_domain(@domain), 'create_domain fail'
|
41
|
+
wait SDB_DELAY, 'after domain creation'
|
42
|
+
# check that we have received new domain from Amazin
|
43
|
+
assert @sdb.list_domains[:domains].include?(@domain)
|
44
|
+
end
|
45
|
+
|
46
|
+
def test_02_put_attributes
|
47
|
+
# put attributes
|
48
|
+
assert @sdb.put_attributes(@domain, @item, @attr)
|
49
|
+
wait SDB_DELAY, 'after putting attributes'
|
50
|
+
end
|
51
|
+
|
52
|
+
def test_03_get_attributes
|
53
|
+
# get attributes
|
54
|
+
values = Array(@sdb.get_attributes(@domain, @item)[:attributes]['Jon']).sort
|
55
|
+
# compare to original list
|
56
|
+
assert_equal values, @attr['Jon'].sort
|
57
|
+
end
|
58
|
+
|
59
|
+
def test_04_add_attributes
|
60
|
+
# add new attribute
|
61
|
+
new_value = 'girls'
|
62
|
+
@sdb.put_attributes @domain, @item, {'Jon' => new_value}
|
63
|
+
wait SDB_DELAY, 'after putting attributes'
|
64
|
+
# get attributes ('girls' must be added to already existent attributes)
|
65
|
+
values = Array(@sdb.get_attributes(@domain, @item)[:attributes]['Jon']).sort
|
66
|
+
assert_equal values, (@attr['Jon'] << new_value).sort
|
67
|
+
end
|
68
|
+
|
69
|
+
def test_05_replace_attributes
|
70
|
+
# replace attributes
|
71
|
+
@sdb.put_attributes @domain, @item, {'Jon' => 'pub'}, :replace
|
72
|
+
wait SDB_DELAY, 'after replacing attributes'
|
73
|
+
# get attributes (all must be removed except of 'pub')
|
74
|
+
values = @sdb.get_attributes(@domain, @item)[:attributes]['Jon']
|
75
|
+
assert_equal values, ['pub']
|
76
|
+
end
|
77
|
+
|
78
|
+
def test_06_delete_attribute
|
79
|
+
# add value 'girls' and 'vodka' to 'Jon'
|
80
|
+
@sdb.put_attributes @domain, @item, {'Jon' => ['girls','vodka']}
|
81
|
+
wait SDB_DELAY, 'after adding attributes'
|
82
|
+
# get attributes ('girls' and 'vodka' must be added 'pub')
|
83
|
+
values = Array(@sdb.get_attributes(@domain, @item)[:attributes]['Jon']).sort
|
84
|
+
assert_equal values, ['girls', 'pub', 'vodka']
|
85
|
+
# delete a single value 'girls' from attribute 'Jon'
|
86
|
+
@sdb.delete_attributes @domain, @item, 'Jon' => ['girls']
|
87
|
+
wait SDB_DELAY, 'after the deletion of attribute'
|
88
|
+
# get attributes ('girls' must be removed)
|
89
|
+
values = @sdb.get_attributes(@domain, @item)[:attributes]['Jon']
|
90
|
+
assert_equal values, ['pub', 'vodka']
|
91
|
+
# delete all values from attribute 'Jon'
|
92
|
+
@sdb.delete_attributes @domain, @item, ['Jon']
|
93
|
+
wait SDB_DELAY, 'after the deletion of attributes'
|
94
|
+
# get attributes (values must be empty)
|
95
|
+
values = @sdb.get_attributes(@domain, @item)[:attributes]['Jon']
|
96
|
+
assert_equal values, nil
|
97
|
+
end
|
98
|
+
|
99
|
+
def test_07_delete_item
|
100
|
+
@sdb.put_attributes @domain, @item, {'Volodya' => ['girls','vodka']}
|
101
|
+
wait SDB_DELAY, 'after adding attributes'
|
102
|
+
# get attributes ('girls' and 'vodka' must be there)
|
103
|
+
values = Array(@sdb.get_attributes(@domain, @item)[:attributes]['Volodya']).sort
|
104
|
+
assert_equal values, ['girls', 'vodka']
|
105
|
+
# delete an item
|
106
|
+
@sdb.delete_attributes @domain, @item
|
107
|
+
wait SDB_DELAY, 'after deleting attributes'
|
108
|
+
# get attributes (values must be empty)
|
109
|
+
values = @sdb.get_attributes(@domain, @item)[:attributes]['Volodya']
|
110
|
+
assert_equal values, nil
|
111
|
+
end
|
112
|
+
|
113
|
+
def test_08_query
|
114
|
+
# add some values for query
|
115
|
+
@sdb.put_attributes @domain, @item, {'Jon' => ['girls','vodka']}
|
116
|
+
wait SDB_DELAY, 'after adding attributes'
|
117
|
+
items = @sdb.query(@domain, ['[?=?]', 'Jon','vodka'])[:items]
|
118
|
+
assert_equal items.size, 1
|
119
|
+
assert_equal items.first, @item
|
120
|
+
end
|
121
|
+
|
122
|
+
def test_09_signature_version_0
|
123
|
+
sdb = Rightscale::SdbInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key, :signature_version => '0')
|
124
|
+
item = 'toys'
|
125
|
+
# TODO: need to change the below test. I think Juergen's intention was to include some umlauts in the values
|
126
|
+
# put attributes
|
127
|
+
# mhhh... Not sure how to translate this: hölzchehn klötzchen grÃŒnspan buÃe... Lets assume this is:
|
128
|
+
attributes = { 'Jurgen' => %w{kitten puppy chickabiddy piglet} }
|
129
|
+
assert sdb.put_attributes(@domain, item, attributes)
|
130
|
+
wait SDB_DELAY, 'after putting attributes'
|
131
|
+
# get attributes
|
132
|
+
values = Array(sdb.get_attributes(@domain, item)[:attributes]['Jurgen']).sort
|
133
|
+
# compare to original list
|
134
|
+
assert_equal values, attributes['Jurgen'].sort
|
135
|
+
# check that the request has correct signature version
|
136
|
+
assert sdb.last_request.path.include?('SignatureVersion=0')
|
137
|
+
end
|
138
|
+
|
139
|
+
def test_10_signature_version_1
|
140
|
+
sdb = Rightscale::SdbInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key, :signature_version => '1')
|
141
|
+
domains = nil
|
142
|
+
assert_nothing_thrown "Failed to use signature V1" do
|
143
|
+
domains = sdb.list_domains
|
144
|
+
end
|
145
|
+
assert domains
|
146
|
+
end
|
147
|
+
|
148
|
+
def test_11_signature_version_1
|
149
|
+
sdb = Rightscale::SdbInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key, :signature_version => '2')
|
150
|
+
domains = nil
|
151
|
+
assert_nothing_thrown "Failed to use signature V2" do
|
152
|
+
domains = sdb.list_domains
|
153
|
+
end
|
154
|
+
assert domains
|
155
|
+
end
|
156
|
+
|
157
|
+
def test_12_array_of_attrs
|
158
|
+
item = 'multiples'
|
159
|
+
assert_nothing_thrown "Failed to put multiple attrs" do
|
160
|
+
@sdb.put_attributes(@domain, item, {:one=>1, :two=>2, :three=>3})
|
161
|
+
end
|
162
|
+
end
|
163
|
+
|
164
|
+
def test_13_zero_len_attrs
|
165
|
+
item = 'zeroes'
|
166
|
+
assert_nothing_thrown "Failed to put zero-length attributes" do
|
167
|
+
@sdb.put_attributes(@domain, item, {:one=>"", :two=>"", :three=>""})
|
168
|
+
end
|
169
|
+
end
|
170
|
+
|
171
|
+
def test_14_nil_attrs
|
172
|
+
item = 'nils'
|
173
|
+
res = nil
|
174
|
+
assert_nothing_thrown do
|
175
|
+
@sdb.put_attributes(@domain, item, {:one=>nil, :two=>nil, :three=>'chunder'})
|
176
|
+
end
|
177
|
+
wait SDB_DELAY, 'after putting attributes'
|
178
|
+
assert_nothing_thrown do
|
179
|
+
res = @sdb.get_attributes(@domain, item)
|
180
|
+
end
|
181
|
+
assert_nil(res[:attributes]['one'][0])
|
182
|
+
assert_nil(res[:attributes]['two'][0])
|
183
|
+
assert_not_nil(res[:attributes]['three'][0])
|
184
|
+
end
|
185
|
+
|
186
|
+
def test_15_url_escape
|
187
|
+
item = 'urlescapes'
|
188
|
+
content = {:a=>"one & two & three",
|
189
|
+
:b=>"one ? two / three"}
|
190
|
+
@sdb.put_attributes(@domain, item, content)
|
191
|
+
wait SDB_DELAY, 'after putting attributes'
|
192
|
+
|
193
|
+
res = @sdb.get_attributes(@domain, item)
|
194
|
+
assert_equal(content[:a], res[:attributes]['a'][0])
|
195
|
+
assert_equal(content[:b], res[:attributes]['b'][0])
|
196
|
+
end
|
197
|
+
|
198
|
+
def test_16_put_attrs_by_post
|
199
|
+
item = 'reqgirth'
|
200
|
+
i = 0
|
201
|
+
sa = ""
|
202
|
+
while(i < 64) do
|
203
|
+
sa += "aaaaaaaa"
|
204
|
+
i += 1
|
205
|
+
end
|
206
|
+
@sdb.put_attributes(@domain, item, {:a => sa, :b => sa, :c => sa, :d => sa, :e => sa})
|
207
|
+
wait SDB_DELAY, 'after putting attributes'
|
208
|
+
end
|
209
|
+
|
210
|
+
def test_20_query_with_atributes
|
211
|
+
response = @sdb.query_with_attributes(@domain)
|
212
|
+
# convers response to a hash representation
|
213
|
+
items = {};
|
214
|
+
response[:items].each{ |item| items.merge!(item) }
|
215
|
+
# check we have receied all 5 items each full of attributes
|
216
|
+
assert_equal 6, items.keys.size
|
217
|
+
assert items['toys'].size > 0
|
218
|
+
assert items['nils'].size > 0
|
219
|
+
assert items['urlescapes'].size > 0
|
220
|
+
assert items['multiples'].size > 0
|
221
|
+
assert items['reqgirth'].size > 0
|
222
|
+
assert items['zeroes'].size > 0
|
223
|
+
# fetch only Jon's attributes from all items
|
224
|
+
response = @sdb.query_with_attributes(@domain,['Jon'])
|
225
|
+
items = {};
|
226
|
+
response[:items].each{ |item| items.merge!(item) }
|
227
|
+
# check we have receied all 5 items
|
228
|
+
# check we have receied all 5 items, but only 'toys' has attributes
|
229
|
+
puts items.inspect
|
230
|
+
assert_equal 2, items['toys']['Jon'].size
|
231
|
+
assert_equal 0, items['nils'].size
|
232
|
+
assert_equal 0, items['urlescapes'].size
|
233
|
+
assert_equal 0, items['multiples'].size
|
234
|
+
assert_equal 0, items['reqgirth'].size
|
235
|
+
assert_equal 0, items['zeroes'].size
|
236
|
+
# kust Jurgen's attriburs
|
237
|
+
response = @sdb.query_with_attributes(@domain,['Jurgen'], "['Jurgen'='piglet']")
|
238
|
+
items = {};
|
239
|
+
response[:items].each{ |item| items.merge!(item) }
|
240
|
+
# check we have receied an only item
|
241
|
+
assert_equal 1, items.keys.size
|
242
|
+
assert_equal ["chickabiddy", "kitten", "piglet", "puppy"], items['toys']['Jurgen'].sort
|
243
|
+
end
|
244
|
+
|
245
|
+
# Keep this test last, because it deletes the domain...
|
246
|
+
def test_21_delete_domain
|
247
|
+
assert @sdb.delete_domain(@domain), 'delete_domain fail'
|
248
|
+
wait SDB_DELAY, 'after domain deletion'
|
249
|
+
# check that domain does not exist
|
250
|
+
assert !@sdb.list_domains[:domains].include?(@domain)
|
251
|
+
end
|
252
|
+
|
253
|
+
end
|
@@ -0,0 +1,73 @@
|
|
1
|
+
require File.dirname(__FILE__) + '/test_helper.rb'
|
2
|
+
|
3
|
+
class TestSns < Test::Unit::TestCase
|
4
|
+
|
5
|
+
GRANTEE_EMAIL_ADDRESS = 'fester@example.com'
|
6
|
+
RIGHT_MESSAGE_TEXT = 'Right test message'
|
7
|
+
|
8
|
+
|
9
|
+
def setup
|
10
|
+
$stdout.sync = true
|
11
|
+
@grantee_aws_id = '100000000001'
|
12
|
+
@topic_name = 'right_sns_test_topic'
|
13
|
+
@sns = RightAws::Sns.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
|
14
|
+
end
|
15
|
+
|
16
|
+
#---------------------------
|
17
|
+
# RightAws::Sns
|
18
|
+
#---------------------------
|
19
|
+
|
20
|
+
def test_01_create_topic
|
21
|
+
topic = @sns.create_topic @topic_name
|
22
|
+
assert topic.kind_of?(RightAws::Sns::Topic), 'New topic wasn\'t initiated'
|
23
|
+
end
|
24
|
+
|
25
|
+
def test_02_list_topics
|
26
|
+
topics = @sns.topics
|
27
|
+
assert topics.kind_of?(Array), 'Must be an array'
|
28
|
+
assert topics.length > 0, 'Must be more than 0 topics in list'
|
29
|
+
end
|
30
|
+
|
31
|
+
def test_03_set_and_get_topic_attributes
|
32
|
+
topic = @sns.topics.last
|
33
|
+
assert topic.owner, 'Unable to get the Owner attribute - get_topic_attributes fail'
|
34
|
+
topic.display_name = 'Dave'
|
35
|
+
assert_equal 'Dave', topic.display_name, 'Unable to set the DisplayName attribute - set_topic_attributes fail'
|
36
|
+
end
|
37
|
+
|
38
|
+
def test_04_add_permissions
|
39
|
+
topic = @sns.topics.last
|
40
|
+
assert topic.add_permission('test', @grantee_aws_id, 'Publish')
|
41
|
+
end
|
42
|
+
|
43
|
+
def test_05_test_permissions
|
44
|
+
topic = @sns.topics.last
|
45
|
+
assert !topic.policy.blank?
|
46
|
+
end
|
47
|
+
|
48
|
+
def test_06_remove_permissions
|
49
|
+
topic = @sns.topics.last
|
50
|
+
assert topic.remove_permission('test')
|
51
|
+
end
|
52
|
+
|
53
|
+
def test_07_send_message
|
54
|
+
topic = @sns.topics.last
|
55
|
+
message = topic.send_message(RIGHT_MESSAGE_TEXT, 'A subject')
|
56
|
+
assert message, 'Message was not created'
|
57
|
+
end
|
58
|
+
|
59
|
+
#---------------------------
|
60
|
+
# RightAws::SnsInterface
|
61
|
+
#---------------------------
|
62
|
+
|
63
|
+
def test_08_set_amazon_problems
|
64
|
+
original_problems = RightAws::SnsInterface.amazon_problems
|
65
|
+
assert(original_problems.length > 0)
|
66
|
+
RightAws::SnsInterface.amazon_problems= original_problems << "A New Problem"
|
67
|
+
new_problems = RightAws::SnsInterface.amazon_problems
|
68
|
+
assert_equal(new_problems, original_problems)
|
69
|
+
|
70
|
+
RightAws::SnsInterface.amazon_problems= nil
|
71
|
+
assert_nil(RightAws::SnsInterface.amazon_problems)
|
72
|
+
end
|
73
|
+
end
|
@@ -0,0 +1,285 @@
|
|
1
|
+
require File.dirname(__FILE__) + '/test_helper.rb'
|
2
|
+
|
3
|
+
class TestSqs < Test::Unit::TestCase
|
4
|
+
|
5
|
+
GRANTEE_EMAIL_ADDRESS = 'madhur@amazon.com'
|
6
|
+
RIGHT_MESSAGE_TEXT = 'Right test message'
|
7
|
+
|
8
|
+
|
9
|
+
def setup
|
10
|
+
@sqs = Rightscale::SqsInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
|
11
|
+
@queue_name = 'right_sqs_test_awesome_queue'
|
12
|
+
# for classes
|
13
|
+
@s = Rightscale::Sqs.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key)
|
14
|
+
end
|
15
|
+
|
16
|
+
# Wait for the queue to appears in the queues list.
|
17
|
+
# Amazon needs some time to after the queue creation to place
|
18
|
+
# it to the accessible queues list. If we dont want to get
|
19
|
+
# the additional faults then wait a bit...
|
20
|
+
def wait_for_queue_url(queue_name)
|
21
|
+
queue_url = nil
|
22
|
+
until queue_url
|
23
|
+
queue_url = @sqs.queue_url_by_name(queue_name)
|
24
|
+
unless queue_url
|
25
|
+
print '-'
|
26
|
+
STDOUT.flush
|
27
|
+
sleep 1
|
28
|
+
end
|
29
|
+
end
|
30
|
+
queue_url
|
31
|
+
end
|
32
|
+
|
33
|
+
|
34
|
+
|
35
|
+
|
36
|
+
def assert_eventually_equal(value, timeout=30, failmsg="", &block)
|
37
|
+
start_time = Time.now.to_i
|
38
|
+
tries = 0
|
39
|
+
while(yield != value) do
|
40
|
+
tries += 1
|
41
|
+
print '-'
|
42
|
+
STDOUT.flush
|
43
|
+
s = Time.now.to_i - start_time
|
44
|
+
flunk("Timeout: #{failmsg}: did not equal \"#{value}\" after #{tries} tries in #{s}s.") if s > timeout
|
45
|
+
sleep(1)
|
46
|
+
setup if (tries % 10) == 0
|
47
|
+
end
|
48
|
+
end
|
49
|
+
|
50
|
+
|
51
|
+
#---------------------------
|
52
|
+
# Rightscale::SqsInterface
|
53
|
+
#---------------------------
|
54
|
+
|
55
|
+
def test_01_create_queue
|
56
|
+
queue_url = @sqs.create_queue @queue_name
|
57
|
+
assert queue_url[/http.*#{@queue_name}/], 'New queue creation fail'
|
58
|
+
end
|
59
|
+
|
60
|
+
def test_02_list_queues
|
61
|
+
wait_for_queue_url(@queue_name)
|
62
|
+
queues = @sqs.list_queues('right_')
|
63
|
+
assert queues.size>0, 'Must more that 0 queues in list'
|
64
|
+
end
|
65
|
+
|
66
|
+
def test_03_set_and_get_queue_attributes
|
67
|
+
queue_url = @sqs.queue_url_by_name(@queue_name)
|
68
|
+
assert queue_url[/http.*#{@queue_name}/], "#{@queue_name} must exist!"
|
69
|
+
assert @sqs.set_queue_attributes(queue_url, 'VisibilityTimeout', 111), 'Set_queue_attributes fail'
|
70
|
+
sleep 20 # Amazon needs some time to change attribute
|
71
|
+
assert_equal '111', @sqs.get_queue_attributes(queue_url)['VisibilityTimeout'], 'New VisibilityTimeout must be equal to 111'
|
72
|
+
end
|
73
|
+
|
74
|
+
def test_04_set_and_get_visibility_timeout
|
75
|
+
queue_url = @sqs.queue_url_by_name(@queue_name)
|
76
|
+
assert @sqs.set_visibility_timeout(queue_url, 222), 'Set_visibility_timeout fail'
|
77
|
+
sleep 20 # Amazon needs some time to change attribute
|
78
|
+
#assert_equal 222, @sqs.get_visibility_timeout(queue_url), 'Get_visibility_timeout must return to 222'
|
79
|
+
assert_eventually_equal(222, 60, 'Get_visibility_timeout must return to 222') do
|
80
|
+
@sqs.get_visibility_timeout(queue_url)
|
81
|
+
end
|
82
|
+
end
|
83
|
+
|
84
|
+
def test_05_add_test_remove_grant
|
85
|
+
queue_url = @sqs.queue_url_by_name(@queue_name)
|
86
|
+
assert @sqs.add_grant(queue_url, GRANTEE_EMAIL_ADDRESS, 'FULLCONTROL'), 'Add grant fail'
|
87
|
+
grants_list = @sqs.list_grants(queue_url, GRANTEE_EMAIL_ADDRESS)
|
88
|
+
assert grants_list.size>0, 'List_grants must return at least 1 record for user #{GRANTEE_EMAIL_ADDRESS}'
|
89
|
+
assert @sqs.remove_grant(queue_url, GRANTEE_EMAIL_ADDRESS, 'FULLCONTROL'), 'Remove_grant fail'
|
90
|
+
end
|
91
|
+
|
92
|
+
def test_06_send_message
|
93
|
+
queue_url = @sqs.queue_url_by_name(@queue_name)
|
94
|
+
# send 5 messages for the tests below
|
95
|
+
assert @sqs.send_message(queue_url, RIGHT_MESSAGE_TEXT)
|
96
|
+
assert @sqs.send_message(queue_url, RIGHT_MESSAGE_TEXT)
|
97
|
+
assert @sqs.send_message(queue_url, RIGHT_MESSAGE_TEXT)
|
98
|
+
assert @sqs.send_message(queue_url, RIGHT_MESSAGE_TEXT)
|
99
|
+
assert @sqs.send_message(queue_url, RIGHT_MESSAGE_TEXT)
|
100
|
+
end
|
101
|
+
|
102
|
+
def test_07_get_queue_length
|
103
|
+
queue_url = @sqs.queue_url_by_name(@queue_name)
|
104
|
+
assert_equal 5, @sqs.get_queue_length(queue_url), 'Queue must have 5 messages'
|
105
|
+
end
|
106
|
+
|
107
|
+
def test_08_receive_message
|
108
|
+
queue_url = @sqs.queue_url_by_name(@queue_name)
|
109
|
+
r_message = @sqs.receive_message(queue_url, 1)
|
110
|
+
assert_equal RIGHT_MESSAGE_TEXT, r_message[:body], 'Receive message get wron message text'
|
111
|
+
p_message = @sqs.peek_message(queue_url, r_message[:id])
|
112
|
+
assert_equal r_message[:body], p_message[:body], 'Received and Peeked messages must be equal'
|
113
|
+
assert @sqs.change_message_visibility(queue_url, r_message[:id], 0), 'Change_message_visibility fail'
|
114
|
+
end
|
115
|
+
|
116
|
+
def test_09_delete_message
|
117
|
+
queue_url = @sqs.queue_url_by_name(@queue_name)
|
118
|
+
message = @sqs.receive_message(queue_url)
|
119
|
+
assert @sqs.delete_message(queue_url, message[:id]), 'Delete_message fail'
|
120
|
+
assert @sqs.pop_message(queue_url), 'Pop_message fail'
|
121
|
+
end
|
122
|
+
|
123
|
+
def test_10_clear_and_delete_queue
|
124
|
+
queue_url = @sqs.queue_url_by_name(@queue_name)
|
125
|
+
assert_raise(Rightscale::AwsError) { @sqs.delete_queue(queue_url) }
|
126
|
+
## oops, force_clear_queue does not work any more - amazon expects for 60 secs timeout between
|
127
|
+
## queue deletion and recreation...
|
128
|
+
## assert @sqs.force_clear_queue(queue_url), 'Force_clear_queue fail'
|
129
|
+
assert @sqs.clear_queue(queue_url), 'Clear_queue fail'
|
130
|
+
assert @sqs.delete_queue(queue_url), 'Delete_queue fail'
|
131
|
+
end
|
132
|
+
|
133
|
+
#---------------------------
|
134
|
+
# Rightscale::Sqs classes
|
135
|
+
#---------------------------
|
136
|
+
|
137
|
+
def test_20_sqs_create_delete_queue
|
138
|
+
assert @s, 'Rightscale::Sqs must exist'
|
139
|
+
# get queues list
|
140
|
+
queues_size = @s.queues.size
|
141
|
+
# create new queue
|
142
|
+
queue = @s.queue("#{@queue_name}_20", true)
|
143
|
+
# check that it is created
|
144
|
+
assert queue.is_a?(Rightscale::Sqs::Queue)
|
145
|
+
wait_for_queue_url(@queue_name)
|
146
|
+
# check that amount of queues has increased
|
147
|
+
assert_eventually_equal(queues_size + 1, 60, "The number of queues did not increase by one") do
|
148
|
+
@s.queues.size
|
149
|
+
end
|
150
|
+
# delete queue
|
151
|
+
assert queue.delete
|
152
|
+
end
|
153
|
+
|
154
|
+
def test_21_queue_create
|
155
|
+
# create new queue
|
156
|
+
queue = Rightscale::Sqs::Queue.create(@s, "#{@queue_name}_21", true)
|
157
|
+
# check that it is created
|
158
|
+
assert queue.is_a?(Rightscale::Sqs::Queue)
|
159
|
+
wait_for_queue_url(@queue_name)
|
160
|
+
end
|
161
|
+
|
162
|
+
def test_22_queue_attributes
|
163
|
+
queue = Rightscale::Sqs::Queue.create(@s, "#{@queue_name}_21", false)
|
164
|
+
# get a list of attrinutes
|
165
|
+
attributes = queue.get_attribute
|
166
|
+
assert attributes.is_a?(Hash) && attributes.size>0
|
167
|
+
# get attribute value and increase it by 10
|
168
|
+
v = (queue.get_attribute('VisibilityTimeout').to_i + 10).to_s
|
169
|
+
# set attribute
|
170
|
+
assert queue.set_attribute('VisibilityTimeout', v)
|
171
|
+
# wait a bit
|
172
|
+
sleep 20
|
173
|
+
# check that attribute has changed
|
174
|
+
assert_equal v, queue.get_attribute('VisibilityTimeout')
|
175
|
+
# get queue visibility timeout
|
176
|
+
assert v.to_i, queue.visibility
|
177
|
+
# change it
|
178
|
+
queue.visibility += 10
|
179
|
+
# make sure that it is changed
|
180
|
+
assert v.to_i + 10, queue.visibility
|
181
|
+
end
|
182
|
+
|
183
|
+
def test_23_grantees
|
184
|
+
queue = Rightscale::Sqs::Queue.create(@s, "#{@queue_name}_21", false)
|
185
|
+
# get a list of grantees
|
186
|
+
grantees = queue.grantees
|
187
|
+
# well, queue must exist at least some seconds before we could add grantees to it....
|
188
|
+
# otherwise we get "Queue does not exists" message. Hence we use the queue
|
189
|
+
# has been created at previous step.
|
190
|
+
#
|
191
|
+
# create new grantee
|
192
|
+
grantee = Rightscale::Sqs::Grantee.new(queue, GRANTEE_EMAIL_ADDRESS)
|
193
|
+
assert grantee.perms.empty?
|
194
|
+
# grant perms
|
195
|
+
assert grantee.grant('FULLCONTROL')
|
196
|
+
assert grantee.grant('RECEIVEMESSAGE')
|
197
|
+
assert_equal 2, grantee.perms.size
|
198
|
+
# make sure that amount of grantees has increased
|
199
|
+
assert grantees.size < queue.grantees.size
|
200
|
+
# revoke perms
|
201
|
+
assert grantee.revoke('RECEIVEMESSAGE')
|
202
|
+
assert_equal 1, grantee.perms.size
|
203
|
+
# remove grantee
|
204
|
+
assert grantee.drop
|
205
|
+
# Don't test this - just for cleanup purposes
|
206
|
+
queue.delete
|
207
|
+
end
|
208
|
+
|
209
|
+
def test_24_send_size
|
210
|
+
queue_url = @sqs.queue_url_by_name("#{@queue_name}_24")
|
211
|
+
@sqs.delete_queue(queue_url)
|
212
|
+
queue = Rightscale::Sqs::Queue.create(@s, "#{@queue_name}_24", true)
|
213
|
+
# send 5 messages
|
214
|
+
assert queue.push('a1')
|
215
|
+
assert queue.push('a2')
|
216
|
+
assert queue.push('a3')
|
217
|
+
assert queue.push('a4')
|
218
|
+
assert queue.push('a5')
|
219
|
+
# check queue size
|
220
|
+
assert_equal 5, queue.size
|
221
|
+
# send one more
|
222
|
+
assert queue.push('a6')
|
223
|
+
# check queue size again
|
224
|
+
assert_equal 6, queue.size
|
225
|
+
end
|
226
|
+
|
227
|
+
def test_25_message_receive_pop_peek_delete
|
228
|
+
queue = Rightscale::Sqs::Queue.create(@s, "#{@queue_name}_24", false)
|
229
|
+
# get queue size
|
230
|
+
size = queue.size
|
231
|
+
# get first message
|
232
|
+
m1 = queue.receive(10)
|
233
|
+
assert m1.is_a?(Rightscale::Sqs::Message)
|
234
|
+
# pop second message
|
235
|
+
m2 = queue.pop
|
236
|
+
assert m2.is_a?(Rightscale::Sqs::Message)
|
237
|
+
# make sure that queue size has decreased
|
238
|
+
assert_equal size-1, queue.size
|
239
|
+
# peek message 1
|
240
|
+
m1p = queue.peek(m1.id)
|
241
|
+
assert m1p.is_a?(Rightscale::Sqs::Message)
|
242
|
+
assert_equal m1.id, m1p.id
|
243
|
+
assert_equal m1.body, m1p.body
|
244
|
+
# change message visibility
|
245
|
+
assert m1.visibility = 30
|
246
|
+
# delete messsage
|
247
|
+
assert m1.delete
|
248
|
+
# make sure that queue size has decreased again
|
249
|
+
assert_equal size-2, queue.size
|
250
|
+
end
|
251
|
+
|
252
|
+
def test_26
|
253
|
+
queue = Rightscale::Sqs::Queue.create(@s, "#{@queue_name}_24", false)
|
254
|
+
# lock message
|
255
|
+
queue.receive(100)
|
256
|
+
# clear queue
|
257
|
+
assert queue.clear
|
258
|
+
# queue size is greater than zero
|
259
|
+
assert queue.size>0
|
260
|
+
queue.push('123456')
|
261
|
+
assert_raise(Rightscale::AwsError) { queue.delete }
|
262
|
+
assert queue.delete(true)
|
263
|
+
end
|
264
|
+
|
265
|
+
def test_27_set_amazon_problems
|
266
|
+
original_problems = Rightscale::SqsInterface.amazon_problems
|
267
|
+
assert(original_problems.length > 0)
|
268
|
+
Rightscale::SqsInterface.amazon_problems= original_problems << "A New Problem"
|
269
|
+
new_problems = Rightscale::SqsInterface.amazon_problems
|
270
|
+
assert_equal(new_problems, original_problems)
|
271
|
+
|
272
|
+
Rightscale::SqsInterface.amazon_problems= nil
|
273
|
+
assert_nil(Rightscale::SqsInterface.amazon_problems)
|
274
|
+
end
|
275
|
+
|
276
|
+
def test_29_signature_version_0
|
277
|
+
sqs = Rightscale::SqsInterface.new(TestCredentials.aws_access_key_id, TestCredentials.aws_secret_access_key, :signature_version => '0')
|
278
|
+
assert_nothing_raised do
|
279
|
+
sqs.list_queues
|
280
|
+
end
|
281
|
+
# check that the request has correct signature version
|
282
|
+
assert sqs.last_request.path.include?('SignatureVersion=0')
|
283
|
+
end
|
284
|
+
|
285
|
+
end
|