Commit 450030e9 authored by Stan Hu's avatar Stan Hu

Add Accept header for JSON

parent d27ae9f5
...@@ -20,6 +20,7 @@ module BitbucketServer ...@@ -20,6 +20,7 @@ module BitbucketServer
def get(path, extra_query = {}) def get(path, extra_query = {})
response = Gitlab::HTTP.get(build_url(path), response = Gitlab::HTTP.get(build_url(path),
basic_auth: auth, basic_auth: auth,
headers: accept_headers,
query: extra_query) query: extra_query)
check_errors!(response) check_errors!(response)
...@@ -74,8 +75,12 @@ module BitbucketServer ...@@ -74,8 +75,12 @@ module BitbucketServer
@auth ||= { username: username, password: token } @auth ||= { username: username, password: token }
end end
def accept_headers
@accept_headers ||= { 'Accept' => 'application/json' }
end
def post_headers def post_headers
@post_headers ||= { 'Content-Type' => 'application/json' } @post_headers ||= accept_headers.merge({ 'Content-Type' => 'application/json' })
end end
def build_url(path) def build_url(path)
......
...@@ -10,46 +10,50 @@ describe BitbucketServer::Connection do ...@@ -10,46 +10,50 @@ describe BitbucketServer::Connection do
describe '#get' do describe '#get' do
it 'returns JSON body' do it 'returns JSON body' do
WebMock.stub_request(:get, url).to_return(body: payload.to_json, status: 200, headers: headers) WebMock.stub_request(:get, url).with(headers: { 'Accept' => 'application/json' }).to_return(body: payload.to_json, status: 200, headers: headers)
expect(subject.get(url, { something: 1 })).to eq(payload) expect(subject.get(url, { something: 1 })).to eq(payload)
end end
it 'throws an exception if the response is not 200' do it 'throws an exception if the response is not 200' do
WebMock.stub_request(:get, url).to_return(body: payload.to_json, status: 500, headers: headers) WebMock.stub_request(:get, url).with(headers: { 'Accept' => 'application/json' }).to_return(body: payload.to_json, status: 500, headers: headers)
expect { subject.get(url) }.to raise_error(described_class::ConnectionError) expect { subject.get(url) }.to raise_error(described_class::ConnectionError)
end end
end end
describe '#post' do describe '#post' do
let(:headers) { { 'Accept' => 'application/json', 'Content-Type' => 'application/json' } }
it 'returns JSON body' do it 'returns JSON body' do
WebMock.stub_request(:post, url).to_return(body: payload.to_json, status: 200, headers: headers) WebMock.stub_request(:post, url).with(headers: headers).to_return(body: payload.to_json, status: 200, headers: headers)
expect(subject.post(url, payload)).to eq(payload) expect(subject.post(url, payload)).to eq(payload)
end end
it 'throws an exception if the response is not 200' do it 'throws an exception if the response is not 200' do
WebMock.stub_request(:post, url).to_return(body: payload.to_json, status: 500, headers: headers) WebMock.stub_request(:post, url).with(headers: headers).to_return(body: payload.to_json, status: 500, headers: headers)
expect { subject.post(url, payload) }.to raise_error(described_class::ConnectionError) expect { subject.post(url, payload) }.to raise_error(described_class::ConnectionError)
end end
end end
describe '#delete' do describe '#delete' do
let(:headers) { { 'Accept' => 'application/json', 'Content-Type' => 'application/json' } }
context 'branch API' do context 'branch API' do
let(:branch_path) { '/projects/foo/repos/bar/branches' } let(:branch_path) { '/projects/foo/repos/bar/branches' }
let(:branch_url) { 'https://test:7990/rest/branch-utils/1.0/projects/foo/repos/bar/branches' } let(:branch_url) { 'https://test:7990/rest/branch-utils/1.0/projects/foo/repos/bar/branches' }
let(:path) { } let(:path) { }
it 'returns JSON body' do it 'returns JSON body' do
WebMock.stub_request(:delete, branch_url).to_return(body: payload.to_json, status: 200, headers: headers) WebMock.stub_request(:delete, branch_url).with(headers: headers).to_return(body: payload.to_json, status: 200, headers: headers)
expect(subject.delete(:branches, branch_path, payload)).to eq(payload) expect(subject.delete(:branches, branch_path, payload)).to eq(payload)
end end
it 'throws an exception if the response is not 200' do it 'throws an exception if the response is not 200' do
WebMock.stub_request(:delete, branch_url).to_return(body: payload.to_json, status: 500, headers: headers) WebMock.stub_request(:delete, branch_url).with(headers: headers).to_return(body: payload.to_json, status: 500, headers: headers)
expect { subject.delete(:branches, branch_path, payload) }.to raise_error(described_class::ConnectionError) expect { subject.delete(:branches, branch_path, payload) }.to raise_error(described_class::ConnectionError)
end end
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment