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
|
# frozen_string_literal: true
require "test_helper"
require "active_support/core_ext/securerandom"
module ActiveStorage::Service::SharedServiceTests
extend ActiveSupport::Concern
FIXTURE_DATA = (+"\211PNG\r\n\032\n\000\000\000\rIHDR\000\000\000\020\000\000\000\020\001\003\000\000\000%=m\"\000\000\000\006PLTE\000\000\000\377\377\377\245\331\237\335\000\000\0003IDATx\234c\370\377\237\341\377_\206\377\237\031\016\2603\334?\314p\1772\303\315\315\f7\215\031\356\024\203\320\275\317\f\367\201R\314\f\017\300\350\377\177\000Q\206\027(\316]\233P\000\000\000\000IEND\256B`\202").force_encoding(Encoding::BINARY)
included do
setup do
@key = SecureRandom.base58(24)
@service = self.class.const_get(:SERVICE)
@service.upload @key, StringIO.new(FIXTURE_DATA)
end
teardown do
@service.delete @key
end
test "uploading with integrity" do
begin
key = SecureRandom.base58(24)
data = "Something else entirely!"
@service.upload(key, StringIO.new(data), checksum: Digest::MD5.base64digest(data))
assert_equal data, @service.download(key)
ensure
@service.delete key
end
end
test "uploading without integrity" do
begin
key = SecureRandom.base58(24)
data = "Something else entirely!"
assert_raises(ActiveStorage::IntegrityError) do
@service.upload(key, StringIO.new(data), checksum: Digest::MD5.base64digest("bad data"))
end
assert_not @service.exist?(key)
ensure
@service.delete key
end
end
test "downloading" do
assert_equal FIXTURE_DATA, @service.download(@key)
end
test "downloading a nonexistent file" do
assert_raises(ActiveStorage::FileNotFoundError) do
@service.download(SecureRandom.base58(24))
end
end
test "downloading in chunks" do
key = SecureRandom.base58(24)
expected_chunks = [ "a" * 5.megabytes, "b" ]
actual_chunks = []
begin
@service.upload key, StringIO.new(expected_chunks.join)
@service.download key do |chunk|
actual_chunks << chunk
end
assert_equal expected_chunks, actual_chunks, "Downloaded chunks did not match uploaded data"
ensure
@service.delete key
end
end
test "downloading a nonexistent file in chunks" do
assert_raises(ActiveStorage::FileNotFoundError) do
@service.download(SecureRandom.base58(24)) { }
end
end
test "downloading partially" do
assert_equal "\x10\x00\x00", @service.download_chunk(@key, 19..21)
assert_equal "\x10\x00\x00", @service.download_chunk(@key, 19...22)
end
test "partially downloading a nonexistent file" do
assert_raises(ActiveStorage::FileNotFoundError) do
@service.download_chunk(SecureRandom.base58(24), 19..21)
end
end
test "existing" do
assert @service.exist?(@key)
assert_not @service.exist?(@key + "nonsense")
end
test "deleting" do
@service.delete @key
assert_not @service.exist?(@key)
end
test "deleting nonexistent key" do
assert_nothing_raised do
@service.delete SecureRandom.base58(24)
end
end
test "deleting by prefix" do
begin
@service.upload("a/a/a", StringIO.new(FIXTURE_DATA))
@service.upload("a/a/b", StringIO.new(FIXTURE_DATA))
@service.upload("a/b/a", StringIO.new(FIXTURE_DATA))
@service.delete_prefixed("a/a/")
assert_not @service.exist?("a/a/a")
assert_not @service.exist?("a/a/b")
assert @service.exist?("a/b/a")
ensure
@service.delete("a/a/a")
@service.delete("a/a/b")
@service.delete("a/b/a")
end
end
end
end
|
