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
|
require File.expand_path '../../test_helper', __dir__
# Storage Blob Class
class TestGetBlobHttpsUrl < Minitest::Test
HOUR = 3600
DAY = 24 * HOUR
WEEK = 7 * DAY
# This class posesses the test cases for the requests of Blob service.
def setup
Fog.mock!
@mock_service = Fog::AzureRM::Storage.new(storage_account_credentials)
Fog.unmock!
@service = Fog::AzureRM::Storage.new(storage_account_credentials)
@blob_client = @service.instance_variable_get(:@blob_client)
@signature_client = @service.send(:signature_client, Minitest::Mock.new)
@url = ApiStub::Requests::Storage::File.blob_https_url
@token = ApiStub::Requests::Storage::File.blob_url_token
end
def test_get_blob_https_url_success
mock_generate_uri = Minitest::Mock.new
2.times do
mock_generate_uri.expect(:call, URI.parse(@url), ['test_container/test_blob', {}, { encode: true }])
end
@blob_client.stub :generate_uri, mock_generate_uri do
@signature_client.stub :generate_service_sas_token, @token do
assert_equal "#{@url}?#{@token}", @service.get_blob_https_url('test_container', 'test_blob', Time.now.utc + 3600)
assert_equal "#{@url}?#{@token}", @service.get_object_url('test_container', 'test_blob', Time.now.utc + 3600)
end
end
end
def test_get_blob_https_url_with_content_disposition
mock_generate_uri = Minitest::Mock.new
mock_generate_service_token = Minitest::Mock.new
url_params = { content_disposition: 'attachment' }
2.times do
mock_generate_uri.expect(:call, URI.parse(@url), ['test_container/test_blob', {}, { encode: true }])
mock_generate_service_token.expect(:call, @token) do |_relative_path, params|
params[:service] == 'b' &&
params[:resource] == 'b' &&
params[:permissions] == 'r' &&
params[:protocol] == 'https' &&
params[:content_disposition] == url_params[:content_disposition]
end
end
@blob_client.stub :generate_uri, mock_generate_uri do
@signature_client.stub :generate_service_sas_token, mock_generate_service_token do
assert_equal "#{@url}?#{@token}", @service.get_blob_https_url('test_container', 'test_blob', Time.now.utc + 3600, url_params)
assert_equal "#{@url}?#{@token}", @service.get_object_url('test_container', 'test_blob', Time.now.utc + 3600, url_params)
end
end
end
def test_get_url_remove_trailing_periods_from_path_segments
mock_generate_uri = Minitest::Mock.new
mock_generate_service_token = Minitest::Mock.new
2.times do
mock_generate_uri.expect(:call, URI.parse(@url), ['.test0/..test1/...test2', {}, { encode: true }])
mock_generate_service_token.expect(:call, @token) do |relative_path, _|
relative_path == '.test0/..test1/...test2'
end
end
@blob_client.stub :generate_uri, mock_generate_uri do
@signature_client.stub :generate_service_sas_token, mock_generate_service_token do
assert_equal "#{@url}?#{@token}", @service.get_blob_https_url('.test0.', '..test1../...test2...', Time.now.utc + 3600)
assert_equal "#{@url}?#{@token}", @service.get_object_url('.test0.', '..test1../...test2...', Time.now.utc + 3600)
end
end
end
def test_get_blob_https_url_with_endpoint_success
service = Fog::AzureRM::Storage.new(storage_account_credentials_with_endpoint)
signature_client = service.send(:signature_client, Minitest::Mock.new)
url = 'http://localhost:10000/mockaccount/test_container/test_blob'
token = ApiStub::Requests::Storage::File.blob_url_token
signature_client.stub :generate_service_sas_token, token do
assert_equal "#{url}?#{token}", service.get_blob_https_url('test_container', 'test_blob', Time.now.utc + 3600)
assert_equal "#{url}?#{token}", service.get_object_url('test_container', 'test_blob', Time.now.utc + 3600)
end
end
def test_get_blob_https_url_with_domain_success
service = Fog::AzureRM::Storage.new(storage_account_credentials_with_domain)
signature_client = service.send(:signature_client, Minitest::Mock.new)
url = 'https://mockaccount.test.example.com/test_container/test_blob'
token = ApiStub::Requests::Storage::File.blob_url_token
signature_client.stub :generate_service_sas_token, token do
assert_equal "#{url}?#{token}", service.get_blob_https_url('test_container', 'test_blob', Time.now.utc + 3600)
assert_equal "#{url}?#{token}", service.get_object_url('test_container', 'test_blob', Time.now.utc + 3600)
end
end
def test_get_blob_https_url_with_token_signer_success
service = Fog::AzureRM::Storage.new(storage_account_credentials_with_token_signer)
blob_client = service.instance_variable_get(:@blob_client)
ref_time = Time.now
stubbed_times = []
requested_expiries = []
expected_user_delegation_key_starts = []
# initial request
stubbed_times << ref_time
requested_expiries << stubbed_times.last + 1 * HOUR
expected_user_delegation_key_starts << stubbed_times.last
# second request during expiry window
stubbed_times << ref_time + 5.5 * DAY
requested_expiries << stubbed_times.last + 1 * DAY
# no additonal expected_user_delegation_key_starts
# request extending past current expiry
stubbed_times << ref_time + 6.5 * DAY
requested_expiries << stubbed_times.last + 1 * DAY
expected_user_delegation_key_starts << stubbed_times.last
# second request within new expiry
stubbed_times << ref_time + 10.5 * DAY
requested_expiries << stubbed_times.last + 1 * DAY
# no additonal expected_user_delegation_key_starts
user_delegation_key_starts = []
mock_user_delegation_key = lambda do |start, expiry|
user_delegation_key_starts << start
assert_equal 1 * WEEK, expiry - start
key = Azure::Storage::Common::Service::UserDelegationKey.new
key.signed_start = "start-#{start.to_i}"
key.signed_expiry = 'test-expiry'
key.value = 'delegation-key'
key
end
mock_new_signer = lambda do |token|
assert_equal 'delegation-key', token
mock_token_signer
end
requested_expiries.each do
mock_token_signer.expect(:sign, 'test-signature', [/\Ar\n.+test_blob\n.+\nstart-\d+\ntest-expiry/m])
end
Time.stub :now, -> { stubbed_times.first } do
blob_client.stub :get_user_delegation_key, mock_user_delegation_key do
Azure::Core::Auth::Signer.stub :new, mock_new_signer do
while (requested_expiry = requested_expiries.shift)
url = service.get_blob_https_url('test_container', 'test_blob', requested_expiry)
assert_match(/^#{Regexp.escape @url}.+&skt=start-#{user_delegation_key_starts.last.to_i}&.+&sig=test-signature/, url)
stubbed_times.shift
end
end
end
end
assert_equal expected_user_delegation_key_starts, user_delegation_key_starts
end
def test_get_blob_https_url_mock
assert_equal "#{@url}?#{@token}", @mock_service.get_blob_https_url('test_container', 'test_blob', Time.now.utc + 3600)
assert_equal "#{@url}?#{@token}", @mock_service.get_object_url('test_container', 'test_blob', Time.now.utc + 3600)
end
end
|
