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
|
# frozen_string_literal: true
module MultipartHelpers
include WorkhorseHelpers
def post_env(rewritten_fields:, params:, secret:, issuer:)
token = JWT.encode({ 'iss' => issuer, 'rewritten_fields' => rewritten_fields }, secret, 'HS256')
Rack::MockRequest.env_for(
'/',
method: 'post',
params: params,
described_class::RACK_ENV_KEY => token
)
end
# This function assumes a `mode` variable to be set
def upload_parameters_for(filepath: nil, key: nil, filename: 'filename', remote_id: 'remote_id')
result = {
"#{key}.name" => filename,
"#{key}.type" => "application/octet-stream",
"#{key}.sha256" => "1234567890"
}
case mode
when :local
result["#{key}.path"] = filepath
when :remote
result["#{key}.remote_id"] = remote_id
result["#{key}.size"] = 3.megabytes
else
raise ArgumentError, "can't handle #{mode} mode"
end
return result if ::Feature.disabled?(:upload_middleware_jwt_params_handler, default_enabled: true)
# the HandlerForJWTParams expects a jwt token with the upload parameters
# *without* the "#{key}." prefix
result.deep_transform_keys! { |k| k.remove("#{key}.") }
{
"#{key}.gitlab-workhorse-upload" => jwt_token('upload' => result)
}
end
# This function assumes a `mode` variable to be set
def rewritten_fields_hash(hash)
if mode == :remote
# For remote uploads, workhorse still submits rewritten_fields,
# but all the values are empty strings.
hash.keys.each { |k| hash[k] = '' }
end
hash
end
def expect_uploaded_files(uploaded_file_expectations)
expect(app).to receive(:call) do |env|
Array.wrap(uploaded_file_expectations).each do |expectation|
file = get_params(env).dig(*expectation[:params_path])
expect_uploaded_file(file, expectation)
end
end
end
# This function assumes a `mode` variable to be set
def expect_uploaded_file(file, expectation)
expect(file).to be_a(::UploadedFile)
expect(file.original_filename).to eq(expectation[:original_filename])
expect(file.sha256).to eq('1234567890')
case mode
when :local
expect(file.path).to eq(File.realpath(expectation[:filepath]))
expect(file.remote_id).to be_nil
expect(file.size).to eq(expectation[:size])
when :remote
expect(file.remote_id).to eq(expectation[:remote_id])
expect(file.path).to be_nil
expect(file.size).to eq(3.megabytes)
else
raise ArgumentError, "can't handle #{mode} mode"
end
end
# Rails doesn't combine the GET/POST parameters in
# ActionDispatch::HTTP::Parameters if action_dispatch.request.parameters is set:
# https://github.com/rails/rails/blob/aea6423f013ca48f7704c70deadf2cd6ac7d70a1/actionpack/lib/action_dispatch/http/parameters.rb#L41
def get_params(env)
req = ActionDispatch::Request.new(env)
req.GET.merge(req.POST)
end
end
|
