-
Notifications
You must be signed in to change notification settings - Fork 0
/
bucket_test.rb
215 lines (174 loc) · 8.74 KB
/
bucket_test.rb
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
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
require "test_helper"
class BucketTest < Test::Unit::TestCase
def setup
@bucket_vhost = S3::Bucket.send(:new, nil, "Data-Bucket")
@bucket_path = S3::Bucket.send(:new, nil, "Data_Bucket")
@bucket = @bucket_vhost
@bucket_location = "EU"
@bucket_location_body = <<-EOLocation
<?xml version="1.0" encoding="UTF-8"?>\n<LocationConstraint xmlns="http://s3.amazonaws.com/doc/2006-03-01/">EU</LocationConstraint>
EOLocation
@response_location = Net::HTTPOK.new("1.1", "200", "OK")
@response_location.stubs(:body).returns(@bucket_location_body)
@bucket_owned_by_you_body = <<-EOOwnedByYou
<?xml version="1.0" encoding="UTF-8"?>\n<Error> <Code>BucketAlreadyOwnedByYou</Code> <Message>Your previous request to create the named bucket succeeded and you already own it.</Message> <BucketName>bucket</BucketName> <RequestId>117D08EA0EC6E860</RequestId> <HostId>4VpMSvmJ+G5+DLtVox6O5cZNgdPlYcjCu3l0n4HjDe01vPxxuk5eTAtcAkUynRyV</HostId> </Error>
EOOwnedByYou
@reponse_owned_by_you = Net::HTTPConflict.new("1.1", "409", "Conflict")
@reponse_owned_by_you.stubs(:body).returns(@bucket_owned_by_you_body)
@bucket_already_exists_body = <<-EOAlreadyExists
<?xml version="1.0" encoding="UTF-8"?>\n<Error> <Code>BucketAlreadyExists</Code> <Message>The requested bucket name is not available. The bucket namespace is shared by all users of the system. Please select a different name and try again.</Message> <BucketName>bucket</BucketName> <RequestId>4C154D32807C92BD</RequestId> <HostId>/xyHQgXcUXTZQhoO+NUBzbaxbFrIhKlyuaRHFnmcId0bMePvY9Zwg+dyk2LYE4g5</HostId> </Error>
EOAlreadyExists
@reponse_already_exists = Net::HTTPConflict.new("1.1", "409", "Conflict")
@response_already_exists.stubs(:body).returns(@bucket_already_exists_body)
@objects_list_empty = []
@objects_list = [
S3::Object.send(:new, @bucket, :key => "obj1"),
S3::Object.send(:new, @bucket, :key => "obj2")
]
@response_objects_list_empty_body = <<-EOEmpty
<?xml version="1.0" encoding="UTF-8"?>\n<ListBucketResult xmlns="http://s3.amazonaws.com/doc/2006-03-01/"> <Name>bucket</Name> <Prefix></Prefix> <Marker></Marker> <MaxKeys>1000</MaxKeys> <IsTruncated>false</IsTruncated> </ListBucketResult>
EOEmpty
@response_objects_list_empty = Net::HTTPOK.new("1.1", "200", "OK")
@response_objects_list_empty.stubs(:body).returns(@response_objects_list_empty_body)
@response_objects_list_body = <<-EOObjects
<?xml version="1.0" encoding="UTF-8"?>\n<ListBucketResult xmlns="http://s3.amazonaws.com/doc/2006-03-01/"> <Name>bucket</Name> <Prefix></Prefix> <Marker></Marker> <MaxKeys>1000</MaxKeys> <IsTruncated>false</IsTruncated> <Contents> <Key>obj1</Key> <LastModified>2009-07-03T10:17:33.000Z</LastModified> <ETag>"99519cdf14c255e580e1b7bca85a458c"</ETag> <Size>1729</Size> <Owner> <ID>df864aeb6f42be43f1d9e60aaabe3f15e245b035a4b79d1cfe36c4deaec67205</ID> <DisplayName>owner</DisplayName> </Owner> <StorageClass>STANDARD</StorageClass> </Contents> <Contents> <Key>obj2</Key> <LastModified>2009-07-03T11:17:33.000Z</LastModified> <ETag>"99519cdf14c255e586e1b12bca85a458c"</ETag> <Size>179</Size> <Owner> <ID>df864aeb6f42be43f1d9e60aaabe3f17e247b037a4b79d1cfe36c4deaec67205</ID> <DisplayName>owner</DisplayName> </Owner> <StorageClass>STANDARD</StorageClass> </Contents> </ListBucketResult>
EOObjects
@response_objects_list = Net::HTTPOK.new("1.1", "200", "OK")
@response_objects_list.stubs(:body).returns(@response_objects_list_body)
end
test "name valid" do
assert_raise ArgumentError do S3::Bucket.send(:new, nil, "") end # should not be valid with empty name
assert_raise ArgumentError do S3::Bucket.send(:new, nil, "10.0.0.1") end # should not be valid with IP as name
assert_raise ArgumentError do S3::Bucket.send(:new, nil, "as") end # should not be valid with name shorter than 3 characters
assert_raise ArgumentError do S3::Bucket.send(:new, nil, "a" * 256) end # should not be valid with name longer than 255 characters
assert_raise ArgumentError do S3::Bucket.send(:new, nil, ".asdf") end # should not allow special characters as first character
assert_raise ArgumentError do S3::Bucket.send(:new, nil, "-asdf") end # should not allow special characters as first character
assert_raise ArgumentError do S3::Bucket.send(:new, nil, "_asdf") end # should not allow special characters as first character
assert_nothing_raised do
S3::Bucket.send(:new, nil, "a-a-")
S3::Bucket.send(:new, nil, "a.a.")
S3::Bucket.send(:new, nil, "a_a_")
end
end
test "path prefix" do
expected = ""
actual = @bucket_vhost.path_prefix
assert_equal expected, actual
expected = "Data_Bucket/"
actual = @bucket_path.path_prefix
assert_equal expected, actual
end
test "host" do
expected = "Data-Bucket.s3.amazonaws.com"
actual = @bucket_vhost.host
assert_equal expected, actual
expected = "s3.amazonaws.com"
actual = @bucket_path.host
assert_equal expected, actual
end
test "vhost" do
assert @bucket_vhost.vhost?
assert ! @bucket_path.vhost?
end
test "exists" do
@bucket.expects(:retrieve).returns(@bucket_vhost)
assert @bucket.exists?
@bucket.expects(:retrieve).raises(S3::Error::NoSuchBucket.new(nil, nil))
assert ! @bucket.exists?
end
test "location and parse location" do
@bucket.expects(:bucket_request).with(:get, { :params => { :location => nil } }).returns(@response_location)
expected = @bucket_location
actual = @bucket.location
assert_equal expected, actual
@bucket.stubs(:bucket_request).with(:get, { :params => { :location => nil } })
actual = @bucket.location
assert_equal expected, actual
end
test "save" do
@bucket.expects(:bucket_request).with(:put, { :headers => {} })
assert @bucket.save
# mock ensures that bucket_request was called
end
test "save failure owned by you" do
@bucket.expects(:bucket_request).with(:put, { :headers => {} }).raises(S3::Error::BucketAlreadyOwnedByYou.new(409, @response_owned_by_you))
assert_raise S3::Error::BucketAlreadyOwnedByYou do
@bucket.save
end
@bucket.expects(:bucket_request).with(:put, { :headers => {} }).raises(S3::Error::BucketAlreadyExists.new(409, @response_already_exists))
assert_raise S3::Error::BucketAlreadyExists do
@bucket.save
end
end
test "objects" do
@bucket.expects(:list_bucket).returns(@objects_list_empty)
expected = @objects_list_empty
actual = @bucket.objects
assert_equal expected, actual
@bucket.stubs(:list_bucket).returns(@objects_list_empty)
actual = @bucket.objects
assert_equal expected, actual
@bucket.stubs(:list_bucket).returns(@objects_list)
expected = @objects_list
actual = @bucket.objects
assert_equal expected, actual
end
test "list bucket and parse objects" do
@bucket.expects(:bucket_request).with(:get, :params => { :test=>true }).returns(@response_objects_list_empty)
expected = @objects_list_empty
actual = @bucket.objects.find_all(:test => true)
assert_equal expected, actual
@bucket.expects(:bucket_request).with(:get, :params => { :test => true }).returns(@response_objects_list)
expected = @objects_list
actual = @bucket.objects.find_all(:test => true)
assert_equal expected, actual
end
test "destroy" do
@bucket.expects(:bucket_request).with(:delete)
assert @bucket.destroy
end
test "objects build" do
@bucket.stubs(:bucket_request)
expected = "object_name"
actual = @bucket.objects.build("object_name")
assert_kind_of S3::Object, actual
assert_equal expected, actual.key
end
test "objects find first" do
assert_nothing_raised do
S3::Object.any_instance.stubs(:retrieve).returns(S3::Object.send(:new, nil, :key => "obj2"))
expected = "obj2"
actual = @bucket.objects.find_first("obj2")
assert_equal "obj2", actual.key
end
end
test "objects find first fail" do
assert_raise S3::Error::NoSuchKey do
S3::Object.any_instance.stubs(:retrieve).raises(S3::Error::NoSuchKey.new(404, nil))
@bucket.objects.find_first("obj3")
end
end
test "objects find all on empty list" do
@bucket.stubs(:list_bucket).returns(@objects_list_empty)
assert_nothing_raised do
expected = @objects_list_empty
actual = @bucket.objects.find_all
assert_equal expected, actual
end
end
test "objects find all" do
@bucket.stubs(:list_bucket).returns(@objects_list)
assert_nothing_raised do
expected = @objects_list
actual = @bucket.objects.find_all
assert_equal expected, actual
end
end
test "objects destroy all" do
@bucket.stubs(:list_bucket).returns(@objects_list)
@bucket.objects.each do |obj|
obj.expects(:destroy)
end
@bucket.objects.destroy_all
end
end